summaryrefslogtreecommitdiffstats
path: root/g4f/.v1/gpt4free/aiassist/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'g4f/.v1/gpt4free/aiassist/__init__.py')
-rw-r--r--g4f/.v1/gpt4free/aiassist/__init__.py36
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