From c5e904ade989066b79656453ac4b73d353c6d973 Mon Sep 17 00:00:00 2001
From: Martin Rattensberger <martin@rattensberger.com>
Date: Thu, 5 Dec 2024 23:08:46 +0100
Subject: [PATCH] API-Keys aus Umgebungsvariable ziehen

---
 .env            | 2 ++
 assistant_v1.py | 6 ++++--
 2 files changed, 6 insertions(+), 2 deletions(-)
 create mode 100644 .env

diff --git a/.env b/.env
new file mode 100644
index 0000000..7f02c95
--- /dev/null
+++ b/.env
@@ -0,0 +1,2 @@
+OPENAI_API_KEY = "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"
+OPENAI_ASSISTANT_ID = "asst_0LyJEWQS90Sx0oEIYBVWE47C"
\ No newline at end of file
diff --git a/assistant_v1.py b/assistant_v1.py
index 4904e11..abc3f3d 100644
--- a/assistant_v1.py
+++ b/assistant_v1.py
@@ -1,10 +1,12 @@
+import os
 import sys
 import json
 from openai import OpenAI
 import time
 
 def get_assistant_response(user_message):
-    client = OpenAI(api_key='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')
+    api_key = os.environ.get("OPENAI_API_KEY")
+    client = OpenAI(api_key)
     
     # Thread erstellen
     thread = client.beta.threads.create()
@@ -19,7 +21,7 @@ def get_assistant_response(user_message):
     # Run mit dem Assistant starten
     run = client.beta.threads.runs.create(
         thread_id=thread.id,
-        assistant_id="asst_0LyJEWQS90Sx0oEIYBVWE47C"
+        assistant_id=os.environ.get("OPENAI_ASSISTANT_ID")
     )
     
     # Auf Antwort warten