blob: 10082c22638bc7166a9b927db97effb1a22ee9ae (
plain) (
blame)
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
|
import json
import requests
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"
request = requests.post(url, json=json_data)
content = request.content
response = Completion.__load_json(content)
return response
@classmethod
def __load_json(cls, content) -> dict:
decode_content = str(content.decode("utf-8"))
split = decode_content.rsplit("\n", 1)[1]
to_json = json.loads(split)
return to_json
|