v2

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