diff options
author | razrab <razrab@PimoLin> | 2023-10-23 07:25:46 +0200 |
---|---|---|
committer | razrab <razrab@PimoLin> | 2023-10-23 07:25:46 +0200 |
commit | 2b3bc749b39b0c6f3e8921b9e3b2fc52263c568c (patch) | |
tree | 7de51129865bd786bdf7b21fa658725a474a8cd8 /g4f/Provider/Geekgpt.py | |
parent | Added code to ignore certain providers in the API (diff) | |
parent | ~ | g4f `v-0.1.7.5` (diff) | |
download | gpt4free-2b3bc749b39b0c6f3e8921b9e3b2fc52263c568c.tar gpt4free-2b3bc749b39b0c6f3e8921b9e3b2fc52263c568c.tar.gz gpt4free-2b3bc749b39b0c6f3e8921b9e3b2fc52263c568c.tar.bz2 gpt4free-2b3bc749b39b0c6f3e8921b9e3b2fc52263c568c.tar.lz gpt4free-2b3bc749b39b0c6f3e8921b9e3b2fc52263c568c.tar.xz gpt4free-2b3bc749b39b0c6f3e8921b9e3b2fc52263c568c.tar.zst gpt4free-2b3bc749b39b0c6f3e8921b9e3b2fc52263c568c.zip |
Diffstat (limited to 'g4f/Provider/Geekgpt.py')
-rw-r--r-- | g4f/Provider/Geekgpt.py | 27 |
1 files changed, 15 insertions, 12 deletions
diff --git a/g4f/Provider/Geekgpt.py b/g4f/Provider/Geekgpt.py index 1a82757c..3c577cf8 100644 --- a/g4f/Provider/Geekgpt.py +++ b/g4f/Provider/Geekgpt.py @@ -14,19 +14,23 @@ class GeekGpt(BaseProvider): supports_gpt_4 = True @classmethod - def create_completion(cls, - model: str, - messages: Messages, - stream: bool, **kwargs) -> CreateResult: - + def create_completion( + cls, + model: str, + messages: Messages, + stream: bool, + **kwargs + ) -> CreateResult: + if not model: + model = "gpt-3.5-turbo" json_data = { 'messages': messages, - 'model': model, - 'temperature': kwargs.get('temperature', 0.9), - 'presence_penalty': kwargs.get('presence_penalty', 0), - 'top_p': kwargs.get('top_p', 1), - 'frequency_penalty': kwargs.get('frequency_penalty', 0), - 'stream': True + 'model': model, + 'temperature': kwargs.get('temperature', 0.9), + 'presence_penalty': kwargs.get('presence_penalty', 0), + 'top_p': kwargs.get('top_p', 1), + 'frequency_penalty': kwargs.get('frequency_penalty', 0), + 'stream': True } data = dumps(json_data, separators=(',', ':')) @@ -61,7 +65,6 @@ class GeekGpt(BaseProvider): try: content = json.loads(json_data)["choices"][0]["delta"].get("content") - except Exception as e: raise RuntimeError(f'error | {e} :', json_data) |