summaryrefslogtreecommitdiffstats
path: root/g4f/api/__init__.py
diff options
context:
space:
mode:
authoralvarosoares01@gmail.com <alvarosoares01@gmail.com>2023-12-14 03:16:35 +0100
committeralvarosoares01@gmail.com <alvarosoares01@gmail.com>2023-12-14 03:16:35 +0100
commit0ea720c1db32ce0c7d31e1a0b96a4d7d3d130cc7 (patch)
tree585adedaeebaf4caa6013fb7a7894739fff5ae1c /g4f/api/__init__.py
parentMerge pull request #1343 from nullstreak/main (diff)
downloadgpt4free-0ea720c1db32ce0c7d31e1a0b96a4d7d3d130cc7.tar
gpt4free-0ea720c1db32ce0c7d31e1a0b96a4d7d3d130cc7.tar.gz
gpt4free-0ea720c1db32ce0c7d31e1a0b96a4d7d3d130cc7.tar.bz2
gpt4free-0ea720c1db32ce0c7d31e1a0b96a4d7d3d130cc7.tar.lz
gpt4free-0ea720c1db32ce0c7d31e1a0b96a4d7d3d130cc7.tar.xz
gpt4free-0ea720c1db32ce0c7d31e1a0b96a4d7d3d130cc7.tar.zst
gpt4free-0ea720c1db32ce0c7d31e1a0b96a4d7d3d130cc7.zip
Diffstat (limited to '')
-rw-r--r--g4f/api/__init__.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/g4f/api/__init__.py b/g4f/api/__init__.py
index 410fcc5f..a79da7b0 100644
--- a/g4f/api/__init__.py
+++ b/g4f/api/__init__.py
@@ -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)