Konversationen eingefügt

This commit is contained in:
2024-11-19 10:25:02 +01:00
parent a41ebc2e98
commit 2a31ef1c19
5 changed files with 53 additions and 81 deletions

View File

@@ -1,20 +1,21 @@
import sys
import json
from openai import OpenAI
import time
from openai import OpenAI
def get_assistant_response(user_message):
def get_assistant_response(conversation):
client = OpenAI(api_key='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')
# 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
)
# Alle Nachrichten zum Thread hinzufügen
for message in conversation:
client.beta.threads.messages.create(
thread_id=thread.id,
role=message['role'],
content=message['content']
)
# Run mit dem Assistant starten
run = client.beta.threads.runs.create(
@@ -39,7 +40,6 @@ def get_assistant_response(user_message):
return {"response": assistant_message}
if __name__ == "__main__":
if len(sys.argv) > 1:
user_message = sys.argv[1]
response = get_assistant_response(user_message)
print(json.dumps(response))
conversation = json.loads(sys.argv[1])
response = get_assistant_response(conversation)
print(json.dumps(response))