kotaemon/knowledgehub/chatbot/simple_respondent.py
Nguyen Trung Duc (john) 6e7905cbc0 [AUR-411] Adopt to Example2 project (#28)
Add the chatbot from Example2. Create the UI for chat.
2023-10-12 15:13:25 +07:00

14 lines
317 B
Python

from theflow import Node
from ..llms import ChatLLM
from .base import BaseChatBot
class SimpleRespondentChatbot(BaseChatBot):
"""Simple text respondent chatbot that essentially wraps around a chat LLM"""
llm: Node[ChatLLM]
def _get_message(self) -> str:
return self.llm(self.history).text[0]