update

agent.py

@@ -21,18 +21,31 @@ from dotenv import load_dotenv
 from agents import OpenAIChatCompletionsModel
 from openai import AsyncOpenAI
 
-OPENROUTER_API_KEY = (
-    "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"
-)
+# Load environment first
+load_dotenv()
 
-client = AsyncOpenAI(
-    api_key=OPENROUTER_API_KEY,
-    base_url="https://openrouter.ai/api/v1",
-)
+# Get API key from environment
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
+
+if not OPENAI_API_KEY and not OPENROUTER_API_KEY:
+    raise ValueError(
+        "Neither OPENAI_API_KEY nor OPENROUTER_API_KEY environment variable is set"
+    )
+
+# Initialize client based on available keys
+if OPENAI_API_KEY:
+    client = AsyncOpenAI(api_key=OPENAI_API_KEY)
+    third_party_model = OpenAIChatCompletionsModel(openai_client=client, model="gpt-4")
+elif OPENROUTER_API_KEY:
+    client = AsyncOpenAI(
+        api_key=OPENROUTER_API_KEY,
+        base_url="https://openrouter.ai/api/v1",
+    )
+    third_party_model = OpenAIChatCompletionsModel(
+        openai_client=client, model="openai/gpt-4-turbo-preview"
+    )
 
-third_party_model = OpenAIChatCompletionsModel(
-    openai_client=client, model="tencent/hy3-preview:free"
-)
 
 # Make backend package importable
 current_dir = os.path.dirname(os.path.abspath(__file__))
@@ -52,9 +65,6 @@ except ImportError:
         "openai-agents package required. Install: pip install openai-agents"
     )
 
-# Load environment
-load_dotenv()
-
 # Setup logging
 logger = setup_logging("agent")