xtekky/gpt4free
1
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-11-23 17:23:57 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1344 from alvarosoaress/api-providers

Browse Source 
Recieve providers in api
This commit is contained in:
H Lohaus 2023-12-14 07:50:43 +01:00 committed by GitHub
commit 36592e4952
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
g4f/api/__init__.py
View File

@ -84,6 +84,10 @@ class Api:
stream = True if item_data.get("stream") == "True" else False
messages = item_data.get('messages')
conversation = item_data.get('conversation') if item_data.get('conversation') != None else None
provider = item_data.get('provider').replace('g4f.Provider.', '')
provider = provider if provider and provider != "Auto" else None
if provider != None:
provider = g4f.Provider.ProviderUtils.convert.get(provider)
try:
if model == 'pi':
@ -92,12 +96,14 @@ class Api:
stream=stream,
messages=messages,
conversation=conversation,
provider = provider,
ignored=self.list_ignored_providers)
else:
response = g4f.ChatCompletion.create(
model=model,
stream=stream,
messages=messages,
provider = provider,
ignored=self.list_ignored_providers)
except Exception as e:
logging.exception(e)