3 files changed, 146 insertions, 0 deletions

diff --git a/gpt4free/gptworldAi/README.md b/gpt4free/gptworldAi/README.md
new file mode 100644
index 00000000..a6b07f86
--- /dev/null
+++ b/gpt4free/gptworldAi/README.md
@@ -0,0 +1,25 @@
+# gptworldAi
+Written by [hp_mzx](https://github.com/hpsj).
+
+## Examples:
+### Completion:
+```python
+for chunk in gptworldAi.Completion.create("你是谁", "127.0.0.1:7890"):
+    print(chunk, end="", flush=True)
+    print()
+```
+
+### Chat Completion:
+Support context
+```python
+message = []
+while True:
+    prompt = input("请输入问题：")
+    message.append({"role": "user","content": prompt})
+    text = ""
+    for chunk in gptworldAi.ChatCompletion.create(message,'127.0.0.1:7890'):
+        text = text+chunk
+        print(chunk, end="", flush=True)
+        print()
+        message.append({"role": "assistant", "content": text})
+```
\ No newline at end of file
diff --git a/gpt4free/gptworldAi/__init__.py b/gpt4free/gptworldAi/__init__.py
new file mode 100644
index 00000000..a729fdf8
--- /dev/null
+++ b/gpt4free/gptworldAi/__init__.py
@@ -0,0 +1,103 @@
+# -*- coding: utf-8 -*-
+"""
+@Time ： 2023/5/23 13:37
+@Auth ： Hp_mzx
+@File ：__init__.py.py
+@IDE ：PyCharm
+"""
+import json
+import random
+import binascii
+import requests
+import Crypto.Cipher.AES as AES
+from fake_useragent import UserAgent
+
+class ChatCompletion:
+    @staticmethod
+    def create(messages:[],proxy: str = None):
+        url = "https://chat.getgpt.world/api/chat/stream"
+        headers = {
+            "Content-Type": "application/json",
+            "Referer": "https://chat.getgpt.world/",
+               'user-agent': UserAgent().random,
+        }
+        proxies = {'http': 'http://' + proxy, 'https': 'http://' + proxy} if proxy else None
+        data = json.dumps({
+            "messages": messages,
+            "frequency_penalty": 0,
+            "max_tokens": 4000,
+            "model": "gpt-3.5-turbo",
+            "presence_penalty": 0,
+            "temperature": 1,
+            "top_p": 1,
+            "stream": True
+        })
+        signature = ChatCompletion.encrypt(data)
+        res = requests.post(url, headers=headers, data=json.dumps({"signature": signature}), proxies=proxies,stream=True)
+        for chunk in res.iter_content(chunk_size=None):
+            res.raise_for_status()
+            datas = chunk.decode('utf-8').split('data: ')
+            for data in datas:
+                if not data or "[DONE]" in data:
+                    continue
+                data_json = json.loads(data)
+                content = data_json['choices'][0]['delta'].get('content')
+                if content:
+                    yield content
+
+
+    @staticmethod
+    def random_token(e):
+        token = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
+        n = len(token)
+        return "".join([token[random.randint(0, n - 1)] for i in range(e)])
+
+    @staticmethod
+    def encrypt(e):
+        t = ChatCompletion.random_token(16).encode('utf-8')
+        n = ChatCompletion.random_token(16).encode('utf-8')
+        r = e.encode('utf-8')
+        cipher = AES.new(t, AES.MODE_CBC, n)
+        ciphertext = cipher.encrypt(ChatCompletion.__pad_data(r))
+        return binascii.hexlify(ciphertext).decode('utf-8') + t.decode('utf-8') + n.decode('utf-8')
+
+    @staticmethod
+    def __pad_data(data: bytes) -> bytes:
+        block_size = AES.block_size
+        padding_size = block_size - len(data) % block_size
+        padding = bytes([padding_size] * padding_size)
+        return data + padding
+
+
+class Completion:
+    @staticmethod
+    def create(prompt:str,proxy:str=None):
+        return ChatCompletion.create([
+            {
+                "content": "You are ChatGPT, a large language model trained by OpenAI.\nCarefully heed the user's instructions. \nRespond using Markdown.",
+                "role": "system"
+            },
+            {"role": "user", "content": prompt}
+        ], proxy)
+
+
+if __name__ == '__main__':
+    # single completion
+    text = ""
+    for chunk in Completion.create("你是谁", "127.0.0.1:7890"):
+        text = text + chunk
+        print(chunk, end="", flush=True)
+    print()
+
+
+    #chat completion
+    message = []
+    while True:
+        prompt = input("请输入问题：")
+        message.append({"role": "user","content": prompt})
+        text = ""
+        for chunk in ChatCompletion.create(message,'127.0.0.1:7890'):
+            text = text+chunk
+            print(chunk, end="", flush=True)
+        print()
+        message.append({"role": "assistant", "content": text})
\ No newline at end of file
diff --git a/testing/gptworldai_test.py b/testing/gptworldai_test.py
new file mode 100644
index 00000000..3dfb32ce
--- /dev/null
+++ b/testing/gptworldai_test.py
@@ -0,0 +1,18 @@
+import gptworldAi
+
+# single completion
+for chunk in gptworldAi.Completion.create("你是谁", "127.0.0.1:7890"):
+    print(chunk, end="", flush=True)
+print()
+
+# chat completion
+message = []
+while True:
+    prompt = input("请输入问题：")
+    message.append({"role": "user", "content": prompt})
+    text = ""
+    for chunk in gptworldAi.ChatCompletion.create(message, '127.0.0.1:7890'):
+        text = text + chunk
+        print(chunk, end="", flush=True)
+    print()
+    message.append({"role": "assistant", "content": text})