import json
import requests
from ..provider.base_provider import BaseProvider
from ..typing import Any, CreateResult
class Raycast(BaseProvider):
url = "https://backend.raycast.com/api/v1/ai/chat_completions"
working = True
supports_stream = True
supports_gpt_35_turbo = True
supports_gpt_4 = True
needs_auth = True
@staticmethod
def create_completion(
model: str,
messages: list[dict[str, str]],
stream: bool,
**kwargs: Any,
) -> CreateResult:
auth = kwargs.get("auth")
headers = {
"Accept": "application/json",
"Accept-Language": "en-US,en;q=0.9",
"Authorization": f"Bearer {auth}",
"Content-Type": "application/json",
"User-Agent": "Raycast/0 CFNetwork/1410.0.3 Darwin/22.6.0",
}
parsed_messages: list[dict[str, Any]] = []
for message in messages:
parsed_messages.append(
{"author": message["role"], "content": {"text": message["content"]}}
)
data = {
"debug": False,
"locale": "en-CN",
"messages": parsed_messages,
"model": model,
"provider": "openai",
"source": "ai_chat",
"system_instruction": "markdown",
"temperature": 0.5,
}
url = "https://backend.raycast.com/api/v1/ai/chat_completions"
response = requests.post(url, headers=headers, json=data, stream=True)
for token in response.iter_lines():
if b"data: " not in token:
continue
completion_chunk = json.loads(token.decode().replace("data: ", ""))
token = completion_chunk["text"]
if token != None:
yield token
