add g4f support - Support for passing model class.

lss233 · Sep 7, 2023 · 145282e · 145282e
1 parent d9de8fa
commit 145282e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/adapter/gpt4free/g4f_helper.py b/adapter/gpt4free/g4f_helper.py
@@ -9,7 +9,7 @@ def g4f_check_account(account: G4fModels):
 
     try:
         response = g4f.ChatCompletion.create(
-            model=account.model,
+            model=eval(account.model) if account.model.startswith("g4f.models.") else account.model,
             provider=eval(account.provider),
             messages=[vars(ChatMessage(ROLE_USER, "hello"))],
         )

diff --git a/adapter/gpt4free/gpt4free.py b/adapter/gpt4free/gpt4free.py
@@ -29,7 +29,7 @@ async def on_reset(self):
     async def ask(self, prompt: str) -> Generator[str, None, None]:
         self.conversation_history.append(vars(ChatMessage(ROLE_USER, prompt)))
         response = g4f.ChatCompletion.create(
-            model=self.model.model,
+            model=eval(self.model.model) if self.model.model.startswith("g4f.models.") else self.model.model,
             provider=eval(self.model.provider),
             messages=self.conversation_history,
         )