fix: arrange output parser classes (#59)

* fix: arrange output parser classes Put all the output parser classes in one place. Remove redundant output parser classes. Reuse output parser classes with the same functionality. * fix: arrange output parser classes Put all the output parser classes in one place. Remove redundant output parser classes. Reuse output parser classes with the same functionality. * fix: output parser error * fix: output parser error --------- Co-authored-by: chenweize1998 <[email protected]>
OpenBMB · Oct 11, 2023 · 88744f1 · 88744f1
1 parent 0208699
commit 88744f1
Show file tree

Hide file tree

Showing 38 changed files with 646 additions and 1,323 deletions.
diff --git a/agentverse/__init__.py b/agentverse/__init__.py
@@ -1,7 +1,4 @@
-from .tasks import *
-
-
-# from .agents import Agent
+from .output_parser import output_parser_registry
 from .environments import env_registry
 from .environments.simulation_env.rules.order import order_registry
 from .environments.simulation_env.rules.describer import describer_registry

diff --git a/agentverse/agents/base.py b/agentverse/agents/base.py
@@ -8,7 +8,7 @@
 from agentverse.llms import BaseLLM
 from agentverse.memory import BaseMemory, ChatHistoryMemory
 from agentverse.message import Message
-from agentverse.parser import OutputParser
+from agentverse.output_parser import OutputParser
 from agentverse.memory_manipulator import BaseMemoryManipulator
 
 

diff --git a/agentverse/agents/simulation_agent/reflection.py b/agentverse/agents/simulation_agent/reflection.py
@@ -14,7 +14,7 @@
 from agentverse.llms import BaseLLM
 from agentverse.memory import BaseMemory, ChatHistoryMemory
 from agentverse.message import Message
-from agentverse.parser import OutputParser
+from agentverse.output_parser import OutputParser
 
 from agentverse.message import Message
 from agentverse.agents.base import BaseAgent

diff --git a/agentverse/initialization.py b/agentverse/initialization.py
@@ -19,7 +19,7 @@
 from agentverse.memory import memory_registry
 from agentverse.memory_manipulator import memory_manipulator_registry
 
-from agentverse.parser import output_parser_registry
+from agentverse.output_parser import output_parser_registry
 
 if TYPE_CHECKING:
     from agentverse.agents import BaseAgent

diff --git a/agentverse/output_parser/__init__.py b/agentverse/output_parser/__init__.py
@@ -0,0 +1,5 @@
+from agentverse.registry import Registry
+
+output_parser_registry = Registry(name="OutputParserRegistry")
+
+from .output_parser import *