Spaces:
Running
Running
File size: 1,049 Bytes
d4adf88 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import urllib.request
import json
class Completion:
@staticmethod
def create(
systemMessage: str = "You are a helpful assistant",
prompt: str = "",
parentMessageId: str = "",
temperature: float = 0.8,
top_p: float = 1,
):
json_data = {
"prompt": prompt,
"options": {"parentMessageId": parentMessageId},
"systemMessage": systemMessage,
"temperature": temperature,
"top_p": top_p,
}
url = "http://43.153.7.56:8080/api/chat-process"
headers = {"Content-type": "application/json"}
data = json.dumps(json_data).encode("utf-8")
req = urllib.request.Request(url, data=data, headers=headers)
response = urllib.request.urlopen(req)
content = response.read().decode()
return Completion.__load_json(content)
@classmethod
def __load_json(cls, content) -> dict:
split = content.rsplit("\n", 1)[1]
to_json = json.loads(split)
return to_json
|