1 files changed, 56 insertions, 0 deletions
diff --git a/g4f/Provider/Raycast.py b/g4f/Provider/Raycast.py
new file mode 100644
index 00000000..9441f629
--- /dev/null
+++ b/g4f/Provider/Raycast.py
@@ -0,0 +1,56 @@
+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