v2

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