gpt4free/g4f/Provider/Opchatgpts.py

63 lines
1.7 KiB
Python
Raw Normal View History

2023-07-28 13:07:17 +03:00
import requests
from ..typing import Any, CreateResult
from .base_provider import BaseProvider
class Opchatgpts(BaseProvider):
url = "https://opchatgpts.net"
working = True
supports_gpt_35_turbo = True
@staticmethod
def create_completion(
model: str,
messages: list[dict[str, str]],
stream: bool,
**kwargs: Any,
) -> CreateResult:
temperature = kwargs.get("temperature", 0.8)
max_tokens = kwargs.get("max_tokens", 1024)
system_prompt = kwargs.get(
"system_prompt",
"Converse as if you were an AI assistant. Be friendly, creative.",
)
payload = _create_payload(
messages=messages,
temperature=temperature,
max_tokens=max_tokens,
system_prompt=system_prompt,
)
response = requests.post(
"https://opchatgpts.net/wp-json/ai-chatbot/v1/chat", json=payload
)
response.raise_for_status()
yield response.json()["reply"]
def _create_payload(
messages: list[dict[str, str]],
temperature: float,
max_tokens: int,
system_prompt: str,
):
return {
"env": "chatbot",
"session": "N/A",
"prompt": "\n",
"context": system_prompt,
"messages": messages,
"newMessage": messages[::-1][0]["content"],
"userName": '<div class="mwai-name-text">User:</div>',
"aiName": '<div class="mwai-name-text">AI:</div>',
"model": "gpt-3.5-turbo",
"temperature": temperature,
"maxTokens": max_tokens,
"maxResults": 1,
"apiKey": "",
"service": "openai",
"embeddingsIndex": "",
"stop": "",
}