Konversationen eingefügt
This commit is contained in:
24
assistant.py
24
assistant.py
@@ -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))
|
||||
Reference in New Issue
Block a user