jackculpan commited on
Commit
030a269
·
1 Parent(s): dda849d
Files changed (1) hide show
  1. conversation.py +7 -3
conversation.py CHANGED
@@ -108,9 +108,13 @@ class Conversation:
108
  # Replace `messages` with `self.messages`
109
  if input:
110
  self.messages.append({"role": "user", "content": input})
111
- chat = openai.ChatCompletion.create(
112
- model="gpt-3.5-turbo", messages=self.messages
113
- )
 
 
 
 
114
  reply = chat.choices[0].message.content
115
  if not reply:
116
  raise ValueError("Please try again")
 
108
  # Replace `messages` with `self.messages`
109
  if input:
110
  self.messages.append({"role": "user", "content": input})
111
+ try:
112
+ chat = openai.ChatCompletion.create(
113
+ model="gpt-3.5-turbo", messages=self.messages
114
+ )
115
+ except openai.error.InvalidRequestError:
116
+ raise ValueError("The website is too large to understand. Please try again.")
117
+
118
  reply = chat.choices[0].message.content
119
  if not reply:
120
  raise ValueError("Please try again")