summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/Lockchat.py
diff options
context:
space:
mode:
authorabc <98614666+xtekky@users.noreply.github.com>2023-08-27 17:37:44 +0200
committerabc <98614666+xtekky@users.noreply.github.com>2023-08-27 17:37:44 +0200
commitefd75a11b871d61ac31b0e274acdfb33daba361d (patch)
tree22bd158324444cdbb9dbed46f25f8dffa60f3756 /g4f/Provider/Lockchat.py
parent ~ | v-0.0.2.5 (diff)
downloadgpt4free-efd75a11b871d61ac31b0e274acdfb33daba361d.tar
gpt4free-efd75a11b871d61ac31b0e274acdfb33daba361d.tar.gz
gpt4free-efd75a11b871d61ac31b0e274acdfb33daba361d.tar.bz2
gpt4free-efd75a11b871d61ac31b0e274acdfb33daba361d.tar.lz
gpt4free-efd75a11b871d61ac31b0e274acdfb33daba361d.tar.xz
gpt4free-efd75a11b871d61ac31b0e274acdfb33daba361d.tar.zst
gpt4free-efd75a11b871d61ac31b0e274acdfb33daba361d.zip
Diffstat (limited to '')
-rw-r--r--g4f/Provider/Lockchat.py44
1 files changed, 19 insertions, 25 deletions
diff --git a/g4f/Provider/Lockchat.py b/g4f/Provider/Lockchat.py
index 974d1331..3c1ce0d1 100644
--- a/g4f/Provider/Lockchat.py
+++ b/g4f/Provider/Lockchat.py
@@ -1,52 +1,46 @@
-import json
+import json, requests
-import requests
-
-from ..typing import Any, CreateResult
+from ..typing import Any, CreateResult
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")