Update app.py

app.py

@@ -78,10 +78,11 @@ class SpyAgent(BasicAgent):
 
     def llm_caller(self, prompt):
         client = OpenAI(
-            api_key=os.getenv('API_KEY')
+            api_key=os.getenv('API_KEY'),
+            base_url=os.getenv('BASE_URL')
         )
         completion = client.chat.completions.create(
-            model="gpt-4o-mini",
+            model=self.model_name,
             messages=[
                 {'role': 'system', 'content': 'You are a helpful assistant.'},
                 {'role': 'user', 'content': prompt}
@@ -97,5 +98,5 @@ class SpyAgent(BasicAgent):
 
 if __name__ == '__main__':
     name = 'spy'
-    agent_builder = AgentBuilder(name, agent=SpyAgent(name, model_name='gpt-4o-mini'))
+    agent_builder = AgentBuilder(name, agent=SpyAgent(name, model_name=os.getenv('MODEL_NAME')))
     agent_builder.start()