2023-08-27 11:37:44 -04:00
|
|
|
import json, uuid, requests
|
2023-07-28 06:07:17 -04:00
|
|
|
|
2023-08-27 11:37:44 -04:00
|
|
|
from ..typing import Any, CreateResult
|
2023-07-28 06:07:17 -04:00
|
|
|
from .base_provider import BaseProvider
|
|
|
|
|
|
|
|
|
|
|
|
class H2o(BaseProvider):
|
2023-08-27 11:37:44 -04:00
|
|
|
url = "https://gpt-gm.h2o.ai"
|
|
|
|
working = True
|
2023-07-28 06:07:17 -04:00
|
|
|
supports_stream = True
|
2023-08-27 11:37:44 -04:00
|
|
|
model = "h2oai/h2ogpt-gm-oasst1-en-2048-falcon-40b-v1"
|
2023-07-28 06:07:17 -04:00
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
def create_completion(
|
|
|
|
model: str,
|
|
|
|
messages: list[dict[str, str]],
|
2023-08-27 11:37:44 -04:00
|
|
|
stream: bool, **kwargs: Any) -> CreateResult:
|
|
|
|
|
2023-07-28 06:07:17 -04:00
|
|
|
conversation = ""
|
|
|
|
for message in messages:
|
|
|
|
conversation += "%s: %s\n" % (message["role"], message["content"])
|
|
|
|
conversation += "assistant: "
|
|
|
|
|
|
|
|
session = requests.Session()
|
|
|
|
|
|
|
|
headers = {"Referer": "https://gpt-gm.h2o.ai/r/jGfKSwU"}
|
|
|
|
data = {
|
2023-08-27 11:37:44 -04:00
|
|
|
"ethicsModalAccepted" : "true",
|
2023-07-28 06:07:17 -04:00
|
|
|
"shareConversationsWithModelAuthors": "true",
|
2023-08-27 11:37:44 -04:00
|
|
|
"ethicsModalAcceptedAt" : "",
|
|
|
|
"activeModel" : model,
|
|
|
|
"searchEnabled" : "true",
|
2023-07-28 06:07:17 -04:00
|
|
|
}
|
2023-08-27 11:37:44 -04:00
|
|
|
|
|
|
|
session.post("https://gpt-gm.h2o.ai/settings",
|
|
|
|
headers=headers, data=data)
|
2023-07-28 06:07:17 -04:00
|
|
|
|
|
|
|
headers = {"Referer": "https://gpt-gm.h2o.ai/"}
|
2023-08-27 11:37:44 -04:00
|
|
|
data = {"model": model}
|
2023-07-28 06:07:17 -04:00
|
|
|
|
2023-08-27 11:37:44 -04:00
|
|
|
response = session.post("https://gpt-gm.h2o.ai/conversation",
|
|
|
|
headers=headers, json=data).json()
|
|
|
|
|
2023-08-22 17:27:34 -04:00
|
|
|
if "conversationId" not in response:
|
|
|
|
return
|
2023-07-28 06:07:17 -04:00
|
|
|
|
|
|
|
data = {
|
|
|
|
"inputs": conversation,
|
|
|
|
"parameters": {
|
2023-08-27 11:37:44 -04:00
|
|
|
"temperature" : kwargs.get("temperature", 0.4),
|
|
|
|
"truncate" : kwargs.get("truncate", 2048),
|
|
|
|
"max_new_tokens" : kwargs.get("max_new_tokens", 1024),
|
|
|
|
"do_sample" : kwargs.get("do_sample", True),
|
2023-07-28 06:07:17 -04:00
|
|
|
"repetition_penalty": kwargs.get("repetition_penalty", 1.2),
|
2023-08-27 11:37:44 -04:00
|
|
|
"return_full_text" : kwargs.get("return_full_text", False),
|
2023-07-28 06:07:17 -04:00
|
|
|
},
|
2023-08-27 11:37:44 -04:00
|
|
|
"stream" : True,
|
2023-07-28 06:07:17 -04:00
|
|
|
"options": {
|
2023-08-27 11:37:44 -04:00
|
|
|
"id" : kwargs.get("id", str(uuid.uuid4())),
|
|
|
|
"response_id" : kwargs.get("response_id", str(uuid.uuid4())),
|
|
|
|
"is_retry" : False,
|
|
|
|
"use_cache" : False,
|
2023-07-28 06:07:17 -04:00
|
|
|
"web_search_id": "",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2023-08-27 11:37:44 -04:00
|
|
|
response = session.post(f"https://gpt-gm.h2o.ai/conversation/{response['conversationId']}",
|
|
|
|
headers=headers, json=data)
|
|
|
|
|
2023-08-22 02:59:58 -04:00
|
|
|
response.raise_for_status()
|
|
|
|
response.encoding = "utf-8"
|
2023-08-27 11:37:44 -04:00
|
|
|
generated_text = response.text.replace("\n", "").split("data:")
|
|
|
|
generated_text = json.loads(generated_text[-1])
|
2023-07-28 06:07:17 -04:00
|
|
|
|
|
|
|
yield generated_text["generated_text"]
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
@property
|
|
|
|
def params(cls):
|
|
|
|
params = [
|
|
|
|
("model", "str"),
|
|
|
|
("messages", "list[dict[str, str]]"),
|
|
|
|
("stream", "bool"),
|
|
|
|
("temperature", "float"),
|
|
|
|
("truncate", "int"),
|
|
|
|
("max_new_tokens", "int"),
|
|
|
|
("do_sample", "bool"),
|
|
|
|
("repetition_penalty", "float"),
|
|
|
|
("return_full_text", "bool"),
|
|
|
|
]
|
|
|
|
param = ", ".join([": ".join(p) for p in params])
|
|
|
|
return f"g4f.provider.{cls.__name__} supports: ({param})"
|