Skip to content

Commit a62a3c8

Browse files
committed
fix: export agentInput and also remove self referring process
1 parent b2b31c4 commit a62a3c8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

agentserve/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# agentserve/__init__.py
22

3-
from .agent import Agent
3+
from .agent import Agent, AgentInput
44
from .agent_server import AgentServer
55
from .cli import main as cli_main

agentserve/agent.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ class Agent:
1010
def _process(self, task_data: Dict[str, Any]) -> Dict[str, Any]:
1111
# Validate task_data against input_schema
1212
validated_data = self.input_schema(**task_data).dict()
13-
return self._process(validated_data)
13+
return self.process(validated_data)
1414

1515
def process(self, task_data: Dict[str, Any]) -> Dict[str, Any]:
1616
"""

0 commit comments

Comments
 (0)