diff options
-rw-r--r-- | gpt4free/hpgptai/README.md | 39 | ||||
-rw-r--r-- | gpt4free/hpgptai/__init__.py | 83 | ||||
-rw-r--r-- | gpt4free/oraai/README.md | 20 | ||||
-rw-r--r-- | gpt4free/oraai/__init__.py | 102 | ||||
-rw-r--r-- | testing/hpgptai_test.py | 41 |
5 files changed, 285 insertions, 0 deletions
diff --git a/gpt4free/hpgptai/README.md b/gpt4free/hpgptai/README.md new file mode 100644 index 00000000..2735902f --- /dev/null +++ b/gpt4free/hpgptai/README.md @@ -0,0 +1,39 @@ +# HpgptAI +Written by [hp_mzx](https://github.com/hpsj). + +## Examples: +### Completion: +```python +res = hpgptai.Completion.create("你是谁","127.0.0.1:7890") +print(res["reply"]) +``` + +### Chat Completion: +Support context +```python +messages = [ + { + "content": "你是谁", + "html": "你是谁", + "id": hpgptai.ChatCompletion.randomStr(), + "role": "user", + "who": "User: ", + }, + { + "content": "我是一位AI助手,专门为您提供各种服务和支持。我可以回答您的问题,帮助您解决问题,提供相关信息,并执行一些任务。请随时告诉我您需要什么帮助。", + "html": "我是一位AI助手,专门为您提供各种服务和支持。我可以回答您的问题,帮助您解决问题,提供相关信息,并执行一些任务。请随时告诉我您需要什么帮助。", + "id": hpgptai.ChatCompletion.randomStr(), + "role": "assistant", + "who": "AI: ", + }, + { + "content": "我上一句问的是什么?", + "html": "我上一句问的是什么?", + "id": hpgptai.ChatCompletion.randomStr(), + "role": "user", + "who": "User: ", + }, +] +res = hpgptai.ChatCompletion.create(messages,proxy="127.0.0.1:7890") +print(res["reply"]) +```
\ No newline at end of file diff --git a/gpt4free/hpgptai/__init__.py b/gpt4free/hpgptai/__init__.py new file mode 100644 index 00000000..66841a87 --- /dev/null +++ b/gpt4free/hpgptai/__init__.py @@ -0,0 +1,83 @@ +# -*- coding: utf-8 -*- +""" +@Time : 2023/5/22 14:04 +@Auth : Hp_mzx +@File :__init__.py.py +@IDE :PyCharm +""" +import json +import requests +import random +import string + +class ChatCompletion: + @staticmethod + def create( + messages: list, + context: str="Converse as if you were an AI assistant. Be friendly, creative.", + proxy:str=None + ): + url = "https://chatgptlogin.ac/wp-json/ai-chatbot/v1/chat" + headers = { + "Content-Type": "application/json", + "X-Wp-Nonce": "02244d73c2" + } + proxies = {'http': 'http://' + proxy, 'https': 'http://' + proxy} if proxy else None + data = { + "env": "chatbot", + "session": "N/A", + "prompt": ChatCompletion.__build_prompt(context,messages), + "context": context, + "messages": messages, + "newMessage": messages[-1]["content"], + "userName": "<div class=\"mwai-name-text\">User:</div>", + "aiName": "<div class=\"mwai-name-text\">AI:</div>", + "model": "gpt-3.5-turbo", + "temperature": 0.8, + "maxTokens": 1024, + "maxResults": 1, + "apiKey": "", + "service": "openai", + "embeddingsIndex": "", + "stop": "", + "clientId": ChatCompletion.randomStr(), + } + res = requests.post(url=url, data=json.dumps(data), headers=headers, proxies=proxies) + if res.status_code == 200: + return res.json() + return res.text + + + @staticmethod + def randomStr(): + return ''.join(random.choices(string.ascii_lowercase + string.digits, k=34))[:11] + + @classmethod + def __build_prompt(cls, context: str, message: list, isCasuallyFineTuned=False, last=15): + prompt = context + '\n\n' if context else '' + message = message[-last:] + if isCasuallyFineTuned: + lastLine = message[-1] + prompt = lastLine.content + "" + return prompt + conversation = [x["who"] + x["content"] for x in message] + prompt += '\n'.join(conversation) + prompt += '\n' + "AI: " + return prompt + + + + +class Completion: + @staticmethod + def create(prompt: str,proxy:str): + messages = [ + { + "content": prompt, + "html": prompt, + "id": ChatCompletion.randomStr(), + "role": "user", + "who": "User: ", + }, + ] + return ChatCompletion.create(messages=messages,proxy=proxy)
\ No newline at end of file diff --git a/gpt4free/oraai/README.md b/gpt4free/oraai/README.md new file mode 100644 index 00000000..6f538711 --- /dev/null +++ b/gpt4free/oraai/README.md @@ -0,0 +1,20 @@ +# OraAI +Written by [hp_mzx](https://github.com/hpsj). + +## Examples: +### Completion: +```python +chunk = oraai.Completion.create("who are you") +print(chunk) +``` + +### Chat Completion: +Support context +```python +obj = oraai.Completion() +whilt True: + prompt = input("Please enter a question:") + chunk = obj.create(prompt) + print(chunk) +print() +```
\ No newline at end of file diff --git a/gpt4free/oraai/__init__.py b/gpt4free/oraai/__init__.py new file mode 100644 index 00000000..99a70098 --- /dev/null +++ b/gpt4free/oraai/__init__.py @@ -0,0 +1,102 @@ +import time +import json +import uuid +import random +import requests +from fake_useragent import UserAgent + + +class ChatCompletion: + def __init__(self,proxy=None,chatbotId="5111b690-edd3-403f-b02a-607332d059f9"): + self.userId = "auto:"+str(uuid.uuid4()) + self.chatbotId = chatbotId + self.proxies = {'http': 'http://' + proxy, 'https': 'http://' + proxy} if proxy else None + self.conversationId = None + self.headers = { + "Content-Type": "application/json", + "Origin": "https://ora.ai", + "Referer": "https://ora.ai/", + 'user-agent': UserAgent().random, + } + + def create(self,prompt: str): + url = "https://ora.ai/api/conversation" + data = { + "chatbotId": self.chatbotId, + "config": False, + "includeHistory": True, + "input": prompt, + "provider": "OPEN_AI", + "userId": self.userId, + } + + if self.conversationId: + data["conversationId"] = self.conversationId + response = requests.post( + url, + data=json.dumps(data), + proxies=self.proxies, + headers=self.headers + ) + if response.status_code == 200: + response_json = response.json() + self.conversationId = response_json["conversationId"] + return response_json["response"] + + raise ValueError(response.text) + + + def generate_image(self,prompt:str): + url = "https://ora.ai/api/images/request" + data = { + "prompt":prompt, + "seed":random.randint(0, 4294967295) + } + response = requests.post( + url, + data=json.dumps(data), + proxies=self.proxies, + headers=self.headers + ) + if response.status_code == 200: + inferenceId = response.json()["id"] + else: + raise ValueError(response.text) + + data = { + "chatbotId":self.chatbotId, + "inferenceId":inferenceId, + "userId":self.userId, + "userInput":"/generate " + prompt + } + print(data) + if self.conversationId: + data["conversationId"] = self.conversationId + while True: + response = requests.post( + "https://ora.ai/api/images/check", + data=json.dumps(data), + proxies=self.proxies, + headers=self.headers + ) + if response.status_code == 200: + response_json = response.json() + if response_json.get("conversationId"): + self.conversationId = response_json["conversationId"] + return response_json["inference"]["images"][0]["uri"] + else: + time.sleep(0.5) + else: + raise ValueError(response.text) + + +class Completion: + @classmethod + def create(self, prompt, proxy): + return ChatCompletion(proxy).create(prompt) + + + + + + diff --git a/testing/hpgptai_test.py b/testing/hpgptai_test.py new file mode 100644 index 00000000..cdd146dd --- /dev/null +++ b/testing/hpgptai_test.py @@ -0,0 +1,41 @@ +import hpgptai + +#single completion +res = hpgptai.Completion.create("你是谁","127.0.0.1:7890") +print(res["reply"]) + + +#chat completion +messages = [ + { + "content": "你是谁", + "html": "你是谁", + "id": hpgptai.ChatCompletion.randomStr(), + "role": "user", + "who": "User: ", + }, + { + "content": "我是一位AI助手,专门为您提供各种服务和支持。我可以回答您的问题,帮助您解决问题,提供相关信息,并执行一些任务。请随时告诉我您需要什么帮助。", + "html": "我是一位AI助手,专门为您提供各种服务和支持。我可以回答您的问题,帮助您解决问题,提供相关信息,并执行一些任务。请随时告诉我您需要什么帮助。", + "id": hpgptai.ChatCompletion.randomStr(), + "role": "assistant", + "who": "AI: ", + }, + { + "content": "我上一句问的是什么?", + "html": "我上一句问的是什么?", + "id": hpgptai.ChatCompletion.randomStr(), + "role": "user", + "who": "User: ", + }, +] +res = hpgptai.ChatCompletion.create(messages,proxy="127.0.0.1:7890") +print(res["reply"]) + + + + + + + + |