DanielDJ1987 commited on
Commit
533a87f
·
verified ·
1 Parent(s): 3a7b9fd

Create utils_openai.py

Browse files
Files changed (1) hide show
  1. utils_openai.py +16 -0
utils_openai.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import openai
2
+
3
+ # API OPENAI ================================================
4
+ def retorna_resposta_modelo(mensagens,
5
+ openai_key,
6
+ modelo='gpt-3.5-turbo',
7
+ temperatura=0,
8
+ stream=False):
9
+ openai.api_key = openai_key
10
+ response = openai.ChatCompletion.create(
11
+ model=modelo,
12
+ messages=mensagens,
13
+ temperature=temperatura,
14
+ stream=stream
15
+ )
16
+ return response