yoursdvniel commited on
Commit
9d3473d
·
verified ·
1 Parent(s): 4704c97

Update openai_client.py

Browse files
Files changed (1) hide show
  1. openai_client.py +7 -7
openai_client.py CHANGED
@@ -1,12 +1,12 @@
1
  import os
2
- import openai
3
 
4
- openai.api_key = os.environ["OPENAI_API_KEY"]
5
 
6
- def ask_gpt(messages, model="gpt-4"):
7
- response = openai.ChatCompletion.create(
8
- model=model,
9
  messages=messages,
10
- temperature=0.3,
11
  )
12
- return response.choices[0].message['content']
 
1
  import os
2
+ from openai import OpenAI
3
 
4
+ client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
5
 
6
+ def ask_gpt(messages):
7
+ response = client.chat.completions.create(
8
+ model="gpt-3.5-turbo",
9
  messages=messages,
10
+ temperature=0.6
11
  )
12
+ return response.choices[0].message.content