fixed some hard-coded values

yusufk · Oct 3, 2023 · 9885cc6 · 9885cc6
1 parent 9e42b69
commit 9885cc6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/conversation.py b/conversation.py
@@ -53,9 +53,9 @@ def __init__(self, user_id=None):
 
     def get_answer(self, tg_user=None):
         # Initialise OpenAI
-        openai.api_type = "azure"
-        openai.api_base = "https://jarvis-openai.openai.azure.com/"
-        openai.api_version = "2022-12-01"
+        openai.api_type = os.getenv("OPENAI_API_TYPE","azure")
+        openai.api_base = os.getenv("OPENAI_API_BASE")
+        openai.api_version = os.getenv("OPENAI_API_VERSION","2022-12-01")
         openai.api_key = os.getenv("OPENAI_API_KEY")
         try:
             response = openai.Completion.create(