summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/Raycast.py
diff options
context:
space:
mode:
authorTekky <98614666+xtekky@users.noreply.github.com>2023-08-17 14:50:35 +0200
committerGitHub <noreply@github.com>2023-08-17 14:50:35 +0200
commite8066c1df55c429a72e0770ee0e30a0ffc03f019 (patch)
treec8bc44917ea03909cf586140f984ff0814bc30ea /g4f/Provider/Raycast.py
parent~ | small fixes & new pypi version | v-0.0.1.9 (diff)
parentrefactor: refactor provider (diff)
downloadgpt4free-e8066c1df55c429a72e0770ee0e30a0ffc03f019.tar
gpt4free-e8066c1df55c429a72e0770ee0e30a0ffc03f019.tar.gz
gpt4free-e8066c1df55c429a72e0770ee0e30a0ffc03f019.tar.bz2
gpt4free-e8066c1df55c429a72e0770ee0e30a0ffc03f019.tar.lz
gpt4free-e8066c1df55c429a72e0770ee0e30a0ffc03f019.tar.xz
gpt4free-e8066c1df55c429a72e0770ee0e30a0ffc03f019.tar.zst
gpt4free-e8066c1df55c429a72e0770ee0e30a0ffc03f019.zip
Diffstat (limited to 'g4f/Provider/Raycast.py')
-rw-r--r--g4f/Provider/Raycast.py56
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