mirror of
https://github.com/hanzoai/python-sdk.git
synced 2026-10-23 20:23:00 +00:00
zeekay
adc3ca0872
Converts hanzo-agent from git submodule to regular directory. This enables simpler cross-package refactoring and unified CI/CD. Source: github.com/hanzoai/agent.git (a8d7ec9)
53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Quick test of Agent SDK
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
sys.path.insert(0, 'src')
|
|
|
|
# For synchronous execution
|
|
from agents import Agent, Runner
|
|
|
|
# Create a simple agent
|
|
agent = Agent(
|
|
name="MathHelper",
|
|
instructions="You are a helpful assistant. Answer concisely.",
|
|
)
|
|
|
|
# Make sure we have an API key
|
|
if not os.getenv("OPENAI_API_KEY") and not os.getenv("ANTHROPIC_API_KEY"):
|
|
# Set a dummy key for now
|
|
os.environ["OPENAI_API_KEY"] = "sk-dummy"
|
|
|
|
print("Testing Agent SDK...")
|
|
print("Question: What's 1+2?")
|
|
print("-" * 40)
|
|
|
|
try:
|
|
# Use the synchronous runner
|
|
result = Runner.run_sync(agent, "What's 1+2?")
|
|
print(f"Answer: {result.final_output}")
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
|
|
# Try with explicit provider
|
|
print("\nTrying with Hanzo Router...")
|
|
from agents.models.openai_provider import OpenAIProvider
|
|
|
|
provider = OpenAIProvider(
|
|
base_url="http://localhost:4000/v1",
|
|
api_key="sk-1234"
|
|
)
|
|
|
|
try:
|
|
from agents import RunConfig
|
|
result = Runner.run_sync(
|
|
agent,
|
|
"What's 1+2?",
|
|
run_config=RunConfig(model_provider=provider, model="gpt-3.5-turbo")
|
|
)
|
|
print(f"Answer: {result.final_output}")
|
|
except Exception as e2:
|
|
print(f"Router Error: {e2}") |