audrius/gpt4free
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added validation for model parameter

Browse Source
This commit is contained in:
Raju Komati
2023-04-24 14:34:14 +05:30
parent 94282777ea
commit 2c720f83a8
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
quora/__init__.py
View File

@@ -380,8 +380,13 @@ class Completion:
class Poe: class Poe:
def __init__(self, model: str = "ChatGPT"): def __init__(self, model: str = "ChatGPT"):
self.cookie = self.__load_cookie() # validating the model
if model and model not in MODELS:
raise RuntimeError(
"Sorry, the model you provided does not exist. Please check and try again."
)
self.model = MODELS[model] self.model = MODELS[model]
self.cookie = self.__load_cookie()
self.client = PoeClient(self.cookie) self.client = PoeClient(self.cookie)
def __load_cookie(self) -> str: def __load_cookie(self) -> str:
@@ -443,7 +448,11 @@ class Poe:
return cookie return cookie
def chat(self, message: str, model: Optional[str] = None) -> str: def chat(self, message: str, model: Optional[str] = None) -> str:
model = MODELS[model] or self.model if model and model not in MODELS:
raise RuntimeError(
"Sorry, the model you provided does not exist. Please check and try again."
)
model = MODELS[model] if model else self.model
response = None response = None
for chunk in self.client.send_message(model, message): for chunk in self.client.send_message(model, message):
response = chunk["text"] response = chunk["text"]