rename langchain BaseMessage

Author: amanjaiswal73892

src/agentlab/llm/llm_utils.py

@@ -14,7 +14,8 @@
 import numpy as np
 import tiktoken
 import yaml
-from langchain.schema import BaseMessage
+# from langchain.schema import BaseMessage 
+from langchain.schema import BaseMessage as LangchainBaseMessage
 from langchain_community.adapters.openai import convert_message_to_dict
 from PIL import Image
 from transformers import AutoModel, AutoTokenizer
@@ -23,14 +24,14 @@
     from agentlab.llm.chat_api import ChatModel
 
 
-def messages_to_dict(messages: list[dict] | list[BaseMessage]) -> dict:
+def messages_to_dict(messages: list[dict] | list[LangchainBaseMessage]) -> dict:
     new_messages = Discussion()
     for m in messages:
         if isinstance(m, dict):
             new_messages.add_message(m)
         elif isinstance(m, str):
             new_messages.add_message({"role": "<unknown role>", "content": m})
-        elif isinstance(m, BaseMessage):
+        elif isinstance(m, LangchainBaseMessage):
             new_messages.add_message(convert_message_to_dict(m))
         else:
             raise ValueError(f"Unknown message type: {type(m)}")