Update app.py

app.py

@@ -10,25 +10,6 @@ from agent_build_sdk.utils.logger import logger
 from openai import OpenAI
 import os
 
-def get_aliyun_response(prompt, model_name="qwen-turbo"):
-    client = OpenAI(
-        api_key=os.getenv('API_KEY'), #  Replace with your own API
-        base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",  
-    )
-    completion = client.chat.completions.create(
-        model=model_name,
-        messages=[
-            {'role': 'system', 'content': 'You are a helpful assistant.'},
-            {'role': 'user', 'content': prompt}],
-        temperature=0.8,
-        top_p=0.8
-        )
-    try:
-        return completion.choices[0].message.content
-    except Exception as e:
-        print(e)
-        return None
-
 
 class SpyAgent(BasicAgent):
 
@@ -96,7 +77,22 @@ class SpyAgent(BasicAgent):
             raise NotImplementedError
 
     def llm_caller(self, prompt):
-        return get_aliyun_response(prompt)
+        client = OpenAI(
+            api_key=os.getenv('API_KEY')
+        )
+        completion = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {'role': 'system', 'content': 'You are a helpful assistant.'},
+                {'role': 'user', 'content': prompt}
+            ],
+            temperature=0
+        )
+        try:
+            return completion.choices[0].message.content
+        except Exception as e:
+            print(e)
+            return None
 
 
 if __name__ == '__main__':