@@ -114,7 +114,7 @@ class TestHelloModel(StaticTestModel):
114
114
class HelloWorldAgent :
115
115
@workflow .run
116
116
async def run (self , prompt : str ) -> str :
117
- agent = Agent (
117
+ agent = Agent [ None ] (
118
118
name = "Assistant" ,
119
119
instructions = "You only respond in haikus." ,
120
120
)
@@ -358,7 +358,7 @@ class TestNexusWeatherModel(StaticTestModel):
358
358
class ToolsWorkflow :
359
359
@workflow .run
360
360
async def run (self , question : str ) -> str :
361
- agent : Agent = Agent (
361
+ agent = Agent [ str ] (
362
362
name = "Tools Workflow" ,
363
363
instructions = "You are a helpful agent." ,
364
364
tools = [
@@ -390,7 +390,7 @@ async def run(self, question: str) -> str:
390
390
class NexusToolsWorkflow :
391
391
@workflow .run
392
392
async def run (self , question : str ) -> str :
393
- agent = Agent (
393
+ agent = Agent [ str ] (
394
394
name = "Nexus Tools Workflow" ,
395
395
instructions = "You are a helpful agent." ,
396
396
tools = [
@@ -747,25 +747,25 @@ async def test_research_workflow(client: Client, use_local_model: bool):
747
747
748
748
749
749
def orchestrator_agent () -> Agent :
750
- spanish_agent = Agent (
750
+ spanish_agent = Agent [ None ] (
751
751
name = "spanish_agent" ,
752
752
instructions = "You translate the user's message to Spanish" ,
753
753
handoff_description = "An english to spanish translator" ,
754
754
)
755
755
756
- french_agent = Agent (
756
+ french_agent = Agent [ None ] (
757
757
name = "french_agent" ,
758
758
instructions = "You translate the user's message to French" ,
759
759
handoff_description = "An english to french translator" ,
760
760
)
761
761
762
- italian_agent = Agent (
762
+ italian_agent = Agent [ None ] (
763
763
name = "italian_agent" ,
764
764
instructions = "You translate the user's message to Italian" ,
765
765
handoff_description = "An english to italian translator" ,
766
766
)
767
767
768
- orchestrator_agent = Agent (
768
+ orchestrator_agent = Agent [ None ] (
769
769
name = "orchestrator_agent" ,
770
770
instructions = (
771
771
"You are a translation agent. You use the tools given to you to translate."
0 commit comments