chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,64 @@
|
||||
import asyncio
|
||||
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from agents import Agent, Runner
|
||||
|
||||
"""
|
||||
This example shows structured input for agent-as-tool calls.
|
||||
"""
|
||||
|
||||
|
||||
class TranslationInput(BaseModel):
|
||||
text: str = Field(description="Text to translate.")
|
||||
source: str = Field(description="Source language code or name.")
|
||||
target: str = Field(description="Target language code or name.")
|
||||
|
||||
|
||||
translator = Agent(
|
||||
name="translator",
|
||||
instructions=(
|
||||
"Translate the input text into the target language. "
|
||||
"If the target is not clear, ask the user for clarification."
|
||||
),
|
||||
)
|
||||
|
||||
orchestrator = Agent(
|
||||
name="orchestrator",
|
||||
instructions=(
|
||||
"You are a task dispatcher. Always call the tool with sufficient input. "
|
||||
"Do not handle the translation yourself."
|
||||
),
|
||||
tools=[
|
||||
translator.as_tool(
|
||||
tool_name="translate_text",
|
||||
tool_description=(
|
||||
"Translate text between languages. Provide text, source language, "
|
||||
"and target language."
|
||||
),
|
||||
parameters=TranslationInput,
|
||||
# By default, the input schema will be included in a simpler format.
|
||||
# Set include_input_schema to true to include the full JSON Schema:
|
||||
# include_input_schema=True,
|
||||
# Build a custom prompt from structured input data:
|
||||
# input_builder=lambda options: (
|
||||
# f'Translate the text "{options["params"]["text"]}" '
|
||||
# f'from {options["params"]["source"]} to {options["params"]["target"]}.'
|
||||
# ),
|
||||
)
|
||||
],
|
||||
)
|
||||
|
||||
|
||||
async def main() -> None:
|
||||
query = 'Translate "Hola" from Spanish to French.'
|
||||
|
||||
response1 = await Runner.run(translator, query)
|
||||
print(f"Translator agent direct run: {response1.final_output}")
|
||||
|
||||
response2 = await Runner.run(orchestrator, query)
|
||||
print(f"Translator agent as tool: {response2.final_output}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
Reference in New Issue
Block a user