Merge pull request #638 from noes14155/patch-1

Update __init__.py
This commit is contained in:
xtekky 2023-06-04 00:28:50 +02:00 committed by GitHub
commit 9430f20fcf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,5 +1,5 @@
import urllib.request
import json import json
import requests
class Completion: class Completion:
@ -20,16 +20,17 @@ class Completion:
} }
url = "http://43.153.7.56:8080/api/chat-process" url = "http://43.153.7.56:8080/api/chat-process"
request = requests.post(url, json=json_data) headers = {"Content-type": "application/json"}
request.encoding = request.apparent_encoding
content = request.content
response = Completion.__load_json(content) data = json.dumps(json_data).encode("utf-8")
return response req = urllib.request.Request(url, data=data, headers=headers)
response = urllib.request.urlopen(req)
content = response.read().decode()
return Completion.__load_json(content)
@classmethod @classmethod
def __load_json(cls, content) -> dict: def __load_json(cls, content) -> dict:
decode_content = str(content.decode("utf-8")) split = content.rsplit("\n", 1)[1]
split = decode_content.rsplit("\n", 1)[1]
to_json = json.loads(split) to_json = json.loads(split)
return to_json return to_json