Nassiraaa commited on
Commit
6a14771
·
verified ·
1 Parent(s): 4f79931

Create openai_utils.py

Browse files
Files changed (1) hide show
  1. openai_utils.py +22 -0
openai_utils.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from openai import OpenAI
3
+
4
+ # OpenAI configuration
5
+ client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
6
+ OPENAI_MODEL = "gpt-3.5-turbo"
7
+
8
+ def get_ai_response(messages):
9
+ """
10
+ Get a response from the AI model using the OpenAI client.
11
+ :param messages: List of message dictionaries as expected by OpenAI API
12
+ :return: The content of the AI's response
13
+ """
14
+ try:
15
+ response = client.chat.completions.create(
16
+ model=OPENAI_MODEL,
17
+ messages=messages
18
+ )
19
+ return response.choices[0].message.content
20
+ except Exception as e:
21
+ print(f"Error getting AI response: {str(e)}")
22
+ return None