🔥 Remove unused parsers

shroominic · Dec 12, 2023 · c8898d0 · c8898d0
1 parent fc2e537
commit c8898d0
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/codeinterpreterapi/session.py b/src/codeinterpreterapi/session.py
@@ -37,7 +37,6 @@
 )
 from codeinterpreterapi.chat_history import CodeBoxChatMessageHistory
 from codeinterpreterapi.config import settings
-from codeinterpreterapi.parser import CodeAgentOutputParser, CodeChatAgentOutputParser
 from codeinterpreterapi.schema import (
     CodeInput,
     CodeInterpreterResponse,
@@ -178,14 +177,12 @@ def _choose_agent(self) -> BaseSingleActionAgent:
                 llm=self.llm,
                 tools=self.tools,
                 system_message=settings.SYSTEM_MESSAGE.content.__str__(),
-                output_parser=CodeChatAgentOutputParser(self.llm),
             )
             if isinstance(self.llm, BaseChatModel)
             else ConversationalAgent.from_llm_and_tools(
                 llm=self.llm,
                 tools=self.tools,
                 prefix=settings.SYSTEM_MESSAGE.content.__str__(),
-                output_parser=CodeAgentOutputParser(),
             )
         )