diff options
| author | H Lohaus <hlohaus@users.noreply.github.com> | 2024-02-14 19:43:07 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-14 19:43:07 +0100 |
| commit | d1abe370cd3180b07354adf41fff6e7f1cbabaea (patch) | |
| tree | a6db83f110ddaf566ff0e784b625424df61f2858 | |
| parent | Delete image.html (diff) | |
| parent | Basic support for the temperature parameter (diff) | |
| download | gpt4free-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.py | 1 | ||||
| -rw-r--r-- | g4f/api/__init__.py | 2 |
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 ) |