summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/Raycast.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--g4f/Provider/Raycast.py59
1 files changed, 36 insertions, 23 deletions
diff --git a/g4f/Provider/Raycast.py b/g4f/Provider/Raycast.py
index 49481202..1f13c9fa 100644
--- a/g4f/Provider/Raycast.py
+++ b/g4f/Provider/Raycast.py
@@ -1,18 +1,17 @@
import json
-
import requests
-
-from ..Provider.base_provider import BaseProvider
from ..typing import Any, CreateResult
+from .base_provider import BaseProvider
class Raycast(BaseProvider):
- url = "https://backend.raycast.com/api/v1/ai/chat_completions"
- working = True
- supports_stream = True
+ url = "https://raycast.com"
+ # model = ['gpt-3.5-turbo', 'gpt-4']
supports_gpt_35_turbo = True
supports_gpt_4 = True
+ supports_stream = True
needs_auth = True
+ working = True
@staticmethod
def create_completion(
@@ -21,20 +20,20 @@ class Raycast(BaseProvider):
stream: bool,
**kwargs: Any,
) -> CreateResult:
- auth = kwargs.get("auth")
-
+ 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",
+ '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]] = []
+ parsed_messages = []
for message in messages:
- parsed_messages.append(
- {"author": message["role"], "content": {"text": message["content"]}}
- )
+ parsed_messages.append({
+ 'author': message['role'],
+ 'content': {'text': message['content']}
+ })
data = {
"debug": False,
"locale": "en-CN",
@@ -43,14 +42,28 @@ class Raycast(BaseProvider):
"provider": "openai",
"source": "ai_chat",
"system_instruction": "markdown",
- "temperature": 0.5,
+ "temperature": 0.5
}
- url = "https://backend.raycast.com/api/v1/ai/chat_completions"
- response = requests.post(url, headers=headers, json=data, stream=True)
+ response = requests.post("https://backend.raycast.com/api/v1/ai/chat_completions", headers=headers, json=data, stream=True)
for token in response.iter_lines():
- if b"data: " not in token:
+ if b'data: ' not in token:
continue
- completion_chunk = json.loads(token.decode().replace("data: ", ""))
- token = completion_chunk["text"]
+ completion_chunk = json.loads(token.decode().replace('data: ', ''))
+ token = completion_chunk['text']
if token != None:
yield token
+
+ @classmethod
+ @property
+ def params(cls):
+ params = [
+ ("model", "str"),
+ ("messages", "list[dict[str, str]]"),
+ ("stream", "bool"),
+ ("temperature", "float"),
+ ("top_p", "int"),
+ ("model", "str"),
+ ("auth", "str"),
+ ]
+ param = ", ".join([": ".join(p) for p in params])
+ return f"g4f.provider.{cls.__name__} supports: ({param})"