Spaces:
Sleeping
Sleeping
Update openai_utils.py
Browse files- openai_utils.py +10 -2
openai_utils.py
CHANGED
@@ -1,8 +1,16 @@
|
|
1 |
import os
|
2 |
from openai import OpenAI
|
|
|
|
|
|
|
|
|
3 |
|
4 |
# OpenAI configuration
|
5 |
-
|
|
|
|
|
|
|
|
|
6 |
OPENAI_MODEL = "gpt-3.5-turbo"
|
7 |
|
8 |
def get_ai_response(messages):
|
@@ -19,4 +27,4 @@ def get_ai_response(messages):
|
|
19 |
return response.choices[0].message.content
|
20 |
except Exception as e:
|
21 |
print(f"Error getting AI response: {str(e)}")
|
22 |
-
return None
|
|
|
1 |
import os
|
2 |
from openai import OpenAI
|
3 |
+
from dotenv import load_dotenv
|
4 |
+
|
5 |
+
# Load environment variables
|
6 |
+
load_dotenv()
|
7 |
|
8 |
# OpenAI configuration
|
9 |
+
api_key = os.environ.get("OPENAI_API_KEY")
|
10 |
+
if not api_key:
|
11 |
+
raise ValueError("OPENAI_API_KEY environment variable is not set. Please set it and try again.")
|
12 |
+
|
13 |
+
client = OpenAI(api_key=api_key)
|
14 |
OPENAI_MODEL = "gpt-3.5-turbo"
|
15 |
|
16 |
def get_ai_response(messages):
|
|
|
27 |
return response.choices[0].message.content
|
28 |
except Exception as e:
|
29 |
print(f"Error getting AI response: {str(e)}")
|
30 |
+
return None
|