summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/Lockchat.py
diff options
context:
space:
mode:
authorRyan Jordan <ryjordan@gmail.com>2023-09-06 02:39:57 +0200
committerGitHub <noreply@github.com>2023-09-06 02:39:57 +0200
commitf81e618958318a092ca4c70a1b3ea15260bda97c (patch)
tree3ce022a8d719011268da7f1a0befc97bf32a60d8 /g4f/Provider/Lockchat.py
parentfeat(docker): add Docker and Docker Compose support (diff)
parent~ | Merge pull request #869 from ahobsonsayers/add-console-script (diff)
downloadgpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.tar
gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.tar.gz
gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.tar.bz2
gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.tar.lz
gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.tar.xz
gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.tar.zst
gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.zip
Diffstat (limited to 'g4f/Provider/Lockchat.py')
-rw-r--r--g4f/Provider/Lockchat.py40
1 files changed, 19 insertions, 21 deletions
diff --git a/g4f/Provider/Lockchat.py b/g4f/Provider/Lockchat.py
index 974d1331..c15eec8d 100644
--- a/g4f/Provider/Lockchat.py
+++ b/g4f/Provider/Lockchat.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
import json
import requests
@@ -7,46 +9,42 @@ from .base_provider import BaseProvider
class Lockchat(BaseProvider):
- url = "http://supertest.lockchat.app"
- supports_stream = True
+ url: str = "http://supertest.lockchat.app"
+ supports_stream = True
supports_gpt_35_turbo = True
- supports_gpt_4 = True
+ supports_gpt_4 = True
@staticmethod
def create_completion(
model: str,
messages: list[dict[str, str]],
- stream: bool,
- **kwargs: Any,
- ) -> CreateResult:
+ stream: bool, **kwargs: Any) -> CreateResult:
+
temperature = float(kwargs.get("temperature", 0.7))
payload = {
"temperature": temperature,
- "messages": messages,
- "model": model,
- "stream": True,
+ "messages" : messages,
+ "model" : model,
+ "stream" : True,
}
headers = {
"user-agent": "ChatX/39 CFNetwork/1408.0.4 Darwin/22.5.0",
}
- response = requests.post(
- "http://supertest.lockchat.app/v1/chat/completions",
- json=payload,
- headers=headers,
- stream=True,
- )
+ response = requests.post("http://supertest.lockchat.app/v1/chat/completions",
+ json=payload, headers=headers, stream=True)
+
response.raise_for_status()
for token in response.iter_lines():
if b"The model: `gpt-4` does not exist" in token:
print("error, retrying...")
Lockchat.create_completion(
- model=model,
- messages=messages,
- stream=stream,
- temperature=temperature,
- **kwargs,
- )
+ model = model,
+ messages = messages,
+ stream = stream,
+ temperature = temperature,
+ **kwargs)
+
if b"content" in token:
token = json.loads(token.decode("utf-8").split("data: ")[1])
token = token["choices"][0]["delta"].get("content")