rodrigomasini commited on
Commit
3307c04
·
1 Parent(s): 46b4328

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -3
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
- gr.Error("OpenAI Error")
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