Subclass chat messages from Document (#86)

2023-11-27 10:38:19 +07:00
parent 3ac277cc0b
commit 0dede9c82d
7 changed files with 36 additions and 10 deletions
--- a/knowledgehub/llms/chats/base.py
+++ b/knowledgehub/llms/chats/base.py
@@ -4,9 +4,10 @@ import logging
 from typing import Type

 from langchain.chat_models.base import BaseChatModel
-from langchain.schema.messages import BaseMessage, HumanMessage
 from theflow.base import Param

+from kotaemon.base.schema import BaseMessage, HumanMessage
+
 from ...base import BaseComponent
 from ...base.schema import LLMInterface