neue Version von Perplexity mit Beispielcode von OpenAI

assistant.py

@@ -1,28 +1,45 @@
 import sys
-import openai
 import json
+from openai import OpenAI
+import time
 
-def main(user_message):
-    # Set your OpenAI API key from the environment, or replace with your key here
-    openai.api_key = "sk-proj-CMVUSsmXIr-Da3a8bpAByG0v2FD1hxEahGs7CqTz7tcegAWGP03OitAAD5LQVf_z5ZUucDWZ10pSHXJVzoWZeGCHueskkC5IMLccUldlvTlsfUA"
+def get_assistant_response(user_message):
+    client = OpenAI(api_key='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')
     
-    try:
-        # Call the OpenAI ChatCompletion API
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages=[{"role": "user", "content": user_message}],
-            assistant_id="asst_0LyJEWQS90Sx0oEIYBVWE47C"  # Assistant ID
+    # Thread erstellen
+    thread = client.beta.threads.create()
+    
+    # Nachricht zum Thread hinzufügen
+    message = client.beta.threads.messages.create(
+        thread_id=thread.id,
+        role="user",
+        content=user_message
+    )
+    
+    # Run mit dem Assistant starten
+    run = client.beta.threads.runs.create(
+        thread_id=thread.id,
+        assistant_id="asst_0LyJEWQS90Sx0oEIYBVWE47C"
+    )
+    
+    # Auf Antwort warten
+    while True:
+        run_status = client.beta.threads.runs.retrieve(
+            thread_id=thread.id,
+            run_id=run.id
         )
-
-        assistant_message = response['choices'][0]['message']['content']
-        print(json.dumps({"response": assistant_message}))
-
-    except Exception as e:
-        print(json.dumps({"response": f"An error occurred: {str(e)}"}))
+        if run_status.status == 'completed':
+            break
+        time.sleep(1)
+    
+    # Antwort abrufen
+    messages = client.beta.threads.messages.list(thread_id=thread.id)
+    assistant_message = messages.data[0].content[0].text.value
+    
+    return {"response": assistant_message}
 
 if __name__ == "__main__":
     if len(sys.argv) > 1:
         user_message = sys.argv[1]
-        main(user_message)
-    else:
-        print(json.dumps({"response": "No message provided"}))
+        response = get_assistant_response(user_message)
+        print(json.dumps(response))