adapt to pydantic v2

agentverse/agents/simulation_agent/tool.py

@@ -1,8 +1,8 @@
 import logging
 from string import Template
-from typing import List, NamedTuple, Optional, Union
+from typing import List, NamedTuple, Optional, Union, Any
 
-from langchain.tools import BaseTool
+# from langchain.tools import BaseTool
 from pydantic import Field
 
 
@@ -28,7 +28,7 @@ class ToolNotExistError(BaseException):
 
 @agent_registry.register("tool")
 class ToolAgent(BaseAgent):
-    tools: List[BaseTool] = Field(default=[])
+    tools: List[Any] = Field(default=[])
     tool_memory: BaseMemory = Field(default_factory=ChatHistoryMemory)
     verbose: bool = Field(default=False)
 

agentverse/environments/simulation_env/rules/describer/prisoner.py

@@ -11,7 +11,7 @@ if TYPE_CHECKING:
 
 @DescriberRegistry.register("prisoner")
 class PrisonerDescriber(BaseDescriber):
-    switch_func = {
+    switch_func: dict = {
         "Both Suspects": "Suspect2",
         "Suspect1": "Suspect2",
         "Suspect2": "Suspect1",

requirements.txt

@@ -8,7 +8,7 @@ opencv-python==4.8.0.76
 gradio==3.50.2
 httpx[socks]
 astunparse
-langchain==0.0.157
+# langchain==0.0.157
 scikit-learn
 pydantic==1.10.7
 typing-extensions==4.5.0
@@ -17,3 +17,4 @@ colorlog
 rapidfuzz
 spacy
 colorama==0.4.6
+tenacity