summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorH Lohaus <hlohaus@users.noreply.github.com>2024-02-14 19:43:07 +0100
committerGitHub <noreply@github.com>2024-02-14 19:43:07 +0100
commitd1abe370cd3180b07354adf41fff6e7f1cbabaea (patch)
treea6db83f110ddaf566ff0e784b625424df61f2858
parentDelete image.html (diff)
parentBasic support for the temperature parameter (diff)
downloadgpt4free-0.2.1.4.tar
gpt4free-0.2.1.4.tar.gz
gpt4free-0.2.1.4.tar.bz2
gpt4free-0.2.1.4.tar.lz
gpt4free-0.2.1.4.tar.xz
gpt4free-0.2.1.4.tar.zst
gpt4free-0.2.1.4.zip
-rw-r--r--g4f/Provider/DeepInfra.py1
-rw-r--r--g4f/api/__init__.py2
2 files changed, 3 insertions, 0 deletions
diff --git a/g4f/Provider/DeepInfra.py b/g4f/Provider/DeepInfra.py
index 8e36128d..183f00ea 100644
--- a/g4f/Provider/DeepInfra.py
+++ b/g4f/Provider/DeepInfra.py
@@ -60,6 +60,7 @@ class DeepInfra(AsyncGeneratorProvider, ProviderModelMixin):
json_data = {
'model' : cls.get_model(model),
'messages': messages,
+ 'temperature': kwargs.get("temperature", 0.7),
'stream' : True
}
async with session.post('https://api.deepinfra.com/v1/openai/chat/completions',
diff --git a/g4f/api/__init__.py b/g4f/api/__init__.py
index 15e8b3a5..3f0778a1 100644
--- a/g4f/api/__init__.py
+++ b/g4f/api/__init__.py
@@ -90,12 +90,14 @@ class Api:
messages = item_data.get('messages')
provider = item_data.get('provider', '').replace('g4f.Provider.', '')
provider = provider if provider and provider != "Auto" else None
+ temperature = item_data.get('temperature')
try:
response = g4f.ChatCompletion.create(
model=model,
stream=stream,
messages=messages,
+ temperature = temperature,
provider = provider,
ignored=self.list_ignored_providers
)