[Refactor] Improve logging package wide (#1315)

This commit is contained in:
Deshraj Yadav
2024-03-13 17:13:30 -07:00
committed by GitHub
parent ef69c91b60
commit 3616eaadb4
54 changed files with 263 additions and 231 deletions

View File

@@ -3,6 +3,8 @@ from typing import Any, Optional
from embedchain.helpers.json_serializable import JSONSerializable
logger = logging.getLogger(__name__)
class BaseMessage(JSONSerializable):
"""
@@ -52,7 +54,7 @@ class ChatMessage(JSONSerializable):
def add_user_message(self, message: str, metadata: Optional[dict] = None):
if self.human_message:
logging.info(
logger.info(
"Human message already exists in the chat message,\
overwriting it with new message."
)
@@ -61,7 +63,7 @@ class ChatMessage(JSONSerializable):
def add_ai_message(self, message: str, metadata: Optional[dict] = None):
if self.ai_message:
logging.info(
logger.info(
"AI message already exists in the chat message,\
overwriting it with new message."
)