Missing self from error handler

This commit is contained in:
Megham Garg
2023-04-21 15:16:08 +05:30
committed by GitHub
parent 6119e89560
commit 85f149c975
+2 -2
View File
@@ -44,7 +44,7 @@ class ChatGPT:
)
answer = r.choices[0].text
else:
raise ValueError(f"Unknown model: {model}")
raise ValueError(f"Unknown model: {self.model}")
answer = self._postprocess_answer(answer)
n_input_tokens, n_output_tokens = r.usage.prompt_tokens, r.usage.completion_tokens
@@ -193,4 +193,4 @@ async def generate_images(prompt, n_images=4):
async def is_content_acceptable(prompt):
r = await openai.Moderation.acreate(input=prompt)
return not all(r.results[0].categories.values())
return not all(r.results[0].categories.values())