Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Subclass chat messages from Document (#86)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
knowledgehub/llms/__init__.py
View File

@@ -1,7 +1,7 @@
from langchain.schema.messages import AIMessage, SystemMessage
from kotaemon.base.schema import AIMessage, BaseMessage, HumanMessage, SystemMessage
from .branching import GatedBranchingPipeline, SimpleBranchingPipeline
from .chats import AzureChatOpenAI, BaseMessage, ChatLLM, HumanMessage
from .chats import AzureChatOpenAI, ChatLLM
from .completions import LLM, AzureOpenAI, OpenAI
from .linear import GatedLinearPipeline, SimpleLinearPipeline
from .prompts import BasePromptComponent, PromptTemplate

3
knowledgehub/llms/chats/base.py
View File

@@ -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