Spaces:
Sleeping
Sleeping
Commit
·
3307c04
1
Parent(s):
46b4328
Update utils.py
Browse files
utils.py
CHANGED
@@ -6,12 +6,11 @@ from langchain.schema import (
|
|
6 |
SystemMessage
|
7 |
)
|
8 |
import os
|
9 |
-
import gradio as gr
|
10 |
|
11 |
OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
|
12 |
|
13 |
def get_api_response(content: str, max_tokens=None):
|
14 |
-
|
15 |
chat = ChatOpenAI(
|
16 |
openai_api_key=OPENAI_API_KEY,
|
17 |
#model='gpt-3',
|
@@ -46,7 +45,7 @@ def get_api_response(content: str, max_tokens=None):
|
|
46 |
try:
|
47 |
response = chat(messages)
|
48 |
except:
|
49 |
-
|
50 |
|
51 |
if response is not None:
|
52 |
return response.content
|
|
|
6 |
SystemMessage
|
7 |
)
|
8 |
import os
|
|
|
9 |
|
10 |
OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
|
11 |
|
12 |
def get_api_response(content: str, max_tokens=None):
|
13 |
+
response = None
|
14 |
chat = ChatOpenAI(
|
15 |
openai_api_key=OPENAI_API_KEY,
|
16 |
#model='gpt-3',
|
|
|
45 |
try:
|
46 |
response = chat(messages)
|
47 |
except:
|
48 |
+
raise Exception("OpenAI Error")
|
49 |
|
50 |
if response is not None:
|
51 |
return response.content
|