diff options
Diffstat (limited to 'g4f/.v1/gpt4free/aiassist/__init__.py')
-rw-r--r-- | g4f/.v1/gpt4free/aiassist/__init__.py | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/g4f/.v1/gpt4free/aiassist/__init__.py b/g4f/.v1/gpt4free/aiassist/__init__.py new file mode 100644 index 00000000..95a9f08b --- /dev/null +++ b/g4f/.v1/gpt4free/aiassist/__init__.py @@ -0,0 +1,36 @@ +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 |