| from __future__ import annotations |
|
|
| import requests |
|
|
| from ..typing import Any, CreateResult |
| from .base_provider import BaseProvider |
|
|
|
|
| class AiService(BaseProvider): |
| url = "https://aiservice.vercel.app/" |
| working = False |
| supports_gpt_35_turbo = True |
|
|
| @staticmethod |
| def create_completion( |
| model: str, |
| messages: list[dict[str, str]], |
| stream: bool, |
| **kwargs: Any, |
| ) -> CreateResult: |
| base = "\n".join(f"{message['role']}: {message['content']}" for message in messages) |
| base += "\nassistant: " |
|
|
| headers = { |
| "accept": "*/*", |
| "content-type": "text/plain;charset=UTF-8", |
| "sec-fetch-dest": "empty", |
| "sec-fetch-mode": "cors", |
| "sec-fetch-site": "same-origin", |
| "Referer": "https://aiservice.vercel.app/chat", |
| } |
| data = {"input": base} |
| url = "https://aiservice.vercel.app/api/chat/answer" |
| response = requests.post(url, headers=headers, json=data) |
| response.raise_for_status() |
| yield response.json()["data"] |
|
|
