Update app.py

app.py

@@ -17,8 +17,12 @@ from my_tools import my_tool_list  # 你需要新建/补全这个模块和列表
 
 class BasicAgent:
     def __init__(self):
-        model = OpenAIChat(model="gpt-4.1")
-        self.agent = CodeAgent(model=model, tools=my_tool_list, max_steps=6)
+        api_key = os.getenv("OPENAI_API_KEY")   # ← 这里自动读取环境变量
+        if not api_key:
+            raise ValueError("OPENAI_API_KEY not set in environment variables!")
+        self.model = OpenAIChat(model="gpt-4o", api_key=api_key)
+        self.agent = CodeAgent(model=self.model, tools=my_tool_list, max_steps=6)
+
     
     def __call__(self, question: str) -> str:
         print(f"Agent received question: {question[:80]}...")