mirror of https://github.com/xtekky/gpt4free.git
add hpgptapi
This commit is contained in:
parent
c7d26ed867
commit
3faf3630ce
|
@ -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"])
|
||||
```
|
|
@ -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)
|
|
@ -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"])
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue