diff options
Diffstat (limited to 'g4f/Provider/Providers/Lockchat.py')
| -rw-r--r-- | g4f/Provider/Providers/Lockchat.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/g4f/Provider/Providers/Lockchat.py b/g4f/Provider/Providers/Lockchat.py new file mode 100644 index 00000000..d97bc67b --- /dev/null +++ b/g4f/Provider/Providers/Lockchat.py @@ -0,0 +1,32 @@ +import requests +import os +import json +from ...typing import sha256, Dict, get_type_hints +url = 'http://super.lockchat.app' +model = ['gpt-4', 'gpt-3.5-turbo'] +supports_stream = True +needs_auth = False + +def _create_completion(model: str, messages: list, stream: bool, temperature: float = 0.7, **kwargs): + + payload = { + "temperature": 0.7, + "messages": messages, + "model": model, + "stream": True, + } + headers = { + "user-agent": "ChatX/39 CFNetwork/1408.0.4 Darwin/22.5.0", + } + response = requests.post("http://super.lockchat.app/v1/chat/completions?auth=FnMNPlwZEnGFqvEc9470Vw==", + json=payload, headers=headers, stream=True) + for token in response.iter_lines(): + if b'The model: `gpt-4` does not exist' in token: + print('error, retrying...') + _create_completion(model=model, messages=messages, stream=stream, temperature=temperature, **kwargs) + if b"content" in token: + token = json.loads(token.decode('utf-8').split('data: ')[1])['choices'][0]['delta'].get('content') + if token: yield (token) + +params = f'g4f.Providers.{os.path.basename(__file__)[:-3]} supports: ' + \ + '(%s)' % ', '.join([f"{name}: {get_type_hints(_create_completion)[name].__name__}" for name in _create_completion.__code__.co_varnames[:_create_completion.__code__.co_argcount]])
\ No newline at end of file |