Update app.py

app.py

@@ -78,7 +78,7 @@ class SpyAgent(BasicAgent):
             base_url=os.getenv('BASE_URL')
         )
         completion = client.chat.completions.create(
-            model=os.getenv('MODEL_NAME'),
+            model=self.model_name,
             messages=[
                 {'role': 'system', 'content': 'You are a helpful assistant.'},
                 {'role': 'user', 'content': prompt}
@@ -94,5 +94,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()