fix: export agentInput and also remove self referring process

PropsAI · Oct 25, 2024 · a62a3c8 · a62a3c8
1 parent b2b31c4
commit a62a3c8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/agentserve/__init__.py b/agentserve/__init__.py
@@ -1,5 +1,5 @@
 # agentserve/__init__.py
 
-from .agent import Agent
+from .agent import Agent, AgentInput
 from .agent_server import AgentServer
 from .cli import main as cli_main
diff --git a/agentserve/agent.py b/agentserve/agent.py
@@ -10,7 +10,7 @@ class Agent:
     def _process(self, task_data: Dict[str, Any]) -> Dict[str, Any]:
         # Validate task_data against input_schema
         validated_data = self.input_schema(**task_data).dict()
-        return self._process(validated_data)
+        return self.process(validated_data)
 
     def process(self, task_data: Dict[str, Any]) -> Dict[str, Any]:
         """