Spaces:
Running
Running
Commit ·
e07ea06
1
Parent(s): 71db74e
update
Browse files
.env
CHANGED
|
@@ -16,4 +16,4 @@ DEPLOY_VERCEL_URL="https://humanoid-ai-robotics-book-1.vercel.app" # /sitema
|
|
| 16 |
# AGENT_MODEL="gpt-4o-mini"
|
| 17 |
|
| 18 |
|
| 19 |
-
|
|
|
|
| 16 |
# AGENT_MODEL="gpt-4o-mini"
|
| 17 |
|
| 18 |
|
| 19 |
+
OPENROUTER_API_KEY="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"
|
agent.py
CHANGED
|
@@ -18,16 +18,27 @@ from fastapi.middleware.cors import CORSMiddleware
|
|
| 18 |
from pydantic import BaseModel, Field, validator
|
| 19 |
from dotenv import load_dotenv
|
| 20 |
|
|
|
|
|
|
|
|
|
|
| 21 |
from agents import OpenAIChatCompletionsModel
|
| 22 |
from openai import AsyncOpenAI
|
| 23 |
|
| 24 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
|
|
|
|
| 26 |
client = AsyncOpenAI(
|
| 27 |
api_key=OPENROUTER_API_KEY,
|
| 28 |
base_url="https://openrouter.ai/api/v1",
|
| 29 |
)
|
| 30 |
|
|
|
|
| 31 |
third_party_model = OpenAIChatCompletionsModel(
|
| 32 |
openai_client=client, model="tencent/hy3-preview:free"
|
| 33 |
)
|
|
@@ -50,9 +61,6 @@ except ImportError:
|
|
| 50 |
"openai-agents package required. Install: pip install openai-agents"
|
| 51 |
)
|
| 52 |
|
| 53 |
-
# Load environment
|
| 54 |
-
load_dotenv()
|
| 55 |
-
|
| 56 |
# Setup logging
|
| 57 |
logger = setup_logging("agent")
|
| 58 |
|
|
|
|
| 18 |
from pydantic import BaseModel, Field, validator
|
| 19 |
from dotenv import load_dotenv
|
| 20 |
|
| 21 |
+
# Load environment first
|
| 22 |
+
load_dotenv()
|
| 23 |
+
|
| 24 |
from agents import OpenAIChatCompletionsModel
|
| 25 |
from openai import AsyncOpenAI
|
| 26 |
|
| 27 |
+
# Get OpenRouter API key from environment
|
| 28 |
+
OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
|
| 29 |
+
if not OPENROUTER_API_KEY:
|
| 30 |
+
raise ValueError(
|
| 31 |
+
"OPENROUTER_API_KEY environment variable must be set. "
|
| 32 |
+
"Get a free key from https://openrouter.ai/"
|
| 33 |
+
)
|
| 34 |
|
| 35 |
+
# Configure AsyncOpenAI client for OpenRouter
|
| 36 |
client = AsyncOpenAI(
|
| 37 |
api_key=OPENROUTER_API_KEY,
|
| 38 |
base_url="https://openrouter.ai/api/v1",
|
| 39 |
)
|
| 40 |
|
| 41 |
+
# Use OpenRouter's free model: tencent/hy3-preview:free
|
| 42 |
third_party_model = OpenAIChatCompletionsModel(
|
| 43 |
openai_client=client, model="tencent/hy3-preview:free"
|
| 44 |
)
|
|
|
|
| 61 |
"openai-agents package required. Install: pip install openai-agents"
|
| 62 |
)
|
| 63 |
|
|
|
|
|
|
|
|
|
|
| 64 |
# Setup logging
|
| 65 |
logger = setup_logging("agent")
|
| 66 |
|
api.py
CHANGED
|
@@ -31,8 +31,8 @@ load_dotenv()
|
|
| 31 |
|
| 32 |
# Import agent components
|
| 33 |
try:
|
| 34 |
-
from agent import get_agent,
|
| 35 |
-
from
|
| 36 |
except ImportError as e:
|
| 37 |
raise ImportError(f"Failed to import agent module: {e}")
|
| 38 |
|
|
|
|
| 31 |
|
| 32 |
# Import agent components
|
| 33 |
try:
|
| 34 |
+
from agent import get_agent, Source as AgentSource
|
| 35 |
+
from agents import Runner, ToolCallOutputItem
|
| 36 |
except ImportError as e:
|
| 37 |
raise ImportError(f"Failed to import agent module: {e}")
|
| 38 |
|