Subclass chat messages from Document (#86)

This commit is contained in:
ian_Cin
2023-11-27 10:38:19 +07:00
committed by GitHub
parent 3ac277cc0b
commit 0dede9c82d
7 changed files with 36 additions and 10 deletions

View File

@@ -1,10 +1,14 @@
from unittest.mock import patch
from langchain.chat_models import AzureChatOpenAI as AzureChatOpenAILC
from langchain.schema.messages import AIMessage, HumanMessage, SystemMessage
from openai.types.chat.chat_completion import ChatCompletion
from kotaemon.base.schema import LLMInterface
from kotaemon.base.schema import (
AIMessage,
HumanMessage,
LLMInterface,
SystemMessage,
)
from kotaemon.llms.chats.openai import AzureChatOpenAI
_openai_chat_completion_response = ChatCompletion.parse_obj(