Update app.py

app.py

@@ -7,11 +7,48 @@ from agent_build_sdk.sdk.agent import format_prompt
 from prompts import DESC_PROMPT, VOTE_PROMPT
 from agent_build_sdk.utils.logger import logger
 
-from openai import OpenAI
+import openai
 import os
 
 
 class SpyAgent(BasicAgent):
+    def __init__(self, api_key: str):
+        super().__init__()
+        self.api_key = api_key
+        openai.api_key = self.api_key
+
+    def generate_response(self, prompt: str) -> str:
+        """
+        Use GPT-3.5 to generate a response based on the provided prompt.
+        """
+        try:
+            logger.info(f"Sending prompt to GPT-3.5: {prompt}")
+            response = openai.ChatCompletion.create(
+                model="gpt-3.5-turbo",
+                messages=[
+                    {"role": "system", "content": "You are a helpful AI assistant."},
+                    {"role": "user", "content": prompt}
+                ],
+                max_tokens=1000,
+                temperature=0.7
+            )
+            # Extract and return the content of the response
+            return response['choices'][0]['message']['content'].strip()
+        except Exception as e:
+            logger.error(f"Error generating response: {e}")
+            return "An error occurred while generating a response."
+
+# Example usage
+if __name__ == "__main__":
+    # Assuming you have set your OpenAI API key in an environment variable
+    api_key = os.getenv("OPENAI_API_KEY")
+    if not api_key:
+        raise ValueError("OPENAI_API_KEY environment variable not set.")
+    
+    agent = SpyAgent(api_key=api_key)
+    prompt = "Describe the role of AI in modern society."
+    response = agent.generate_response(prompt)
+    print(f"GPT-3.5 Response: {response}")
 
     def perceive(self, req=AgentReq):
         logger.info("spy perceive: {}".format(req))