🔧 pydantic v2 fix typing

Author: shroominic

codeinterpreterapi/agents/functions_agent.py

@@ -6,14 +6,15 @@
 
 from langchain.agents import BaseSingleActionAgent
 from langchain.callbacks.base import BaseCallbackManager
-from langchain.callbacks.manager import Callbacks
+from langchain.callbacks.manager import Callbacks  # type: ignore
 from langchain.chat_models.openai import ChatOpenAI
 from langchain.prompts.chat import (
     BaseMessagePromptTemplate,
     ChatPromptTemplate,
     HumanMessagePromptTemplate,
     MessagesPlaceholder,
 )
+from langchain.pydantic_v1 import root_validator
 from langchain.schema import (
     AgentAction,
     AgentFinish,
@@ -29,7 +30,6 @@
 )
 from langchain.tools import BaseTool
 from langchain.tools.convert_to_openai import format_tool_to_openai_function
-from pydantic import root_validator
 
 
 @dataclass

codeinterpreterapi/session.py

@@ -240,7 +240,7 @@ def _run_handler(self, code: str):
                         "got installed now. Please try again."
                     )
             else:
-                # TODO: preanalyze error to optimize next code generation
+                # TODO: pre-analyze error to optimize next code generation
                 pass
             if self.verbose:
                 print("Error:", output.content)
@@ -287,7 +287,7 @@ async def _arun_handler(self, code: str):
                         "got installed now. Please try again."
                     )
             else:
-                # TODO: preanalyze error to optimize next code generation
+                # TODO: pre-analyze error to optimize next code generation
                 pass
             if self.verbose:
                 print("Error:", output.content)