chat agent abstract class

Former-commit-id: 1d36cd9c45dd5875458a3e4d01e846cb6b4a687d
pull/160/head
Kye 2 years ago
parent eb595dccf3
commit 03a1331cf7

@ -9,7 +9,7 @@ from langchain.chat_models.base import BaseChatModel
from langchain.schema import BaseOutputParser
from langchain.callbacks.base import BaseCallbackManager
from .ConversationalChatAgent import ConversationalChatAgent
from .ConversationalChatAgent import ChatAgent
# from .ChatOpenAI import ChatOpenAI
from langchain.chat_models import ChatOpenAI
from .EvalOutputParser import EvalOutputParser
@ -79,7 +79,7 @@ class AgentBuilder:
if self.global_tools is None:
raise ValueError("Global tools must be initialized before agent")
return ConversationalChatAgent.from_llm_and_tools(
return ChatAgent.from_llm_and_tools(
llm=self.llm,
tools=[
*self.global_tools,

@ -140,4 +140,9 @@ class ConversationalChatAgent(Agent):
)
except Exception as e:
logging.error(f"Error while creating agent from LLM and tools: {str(e)}")
raise e
raise e
class ChatAgent(ConversationalChatAgent):
def _get_default_output_parser(self):
"""Get default output parser for this class."""

Loading…
Cancel
Save