File size: 1,398 Bytes
a4b70d9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
from __future__ import annotations
import os
from ...typing import Messages, AsyncResult
from ...errors import MissingAuthError
from ..template import OpenaiTemplate
class Claude(OpenaiTemplate):
label = "Claude 💥"
url = "https://claude.ai"
api_base = "https://g4f.dev/api/claude"
working = True
active_by_default = True
login_url = "https://discord.gg/qXA4Wf4Fsm"
organization_id = None
@classmethod
async def create_async_generator(
cls,
model: str,
messages: Messages,
api_key: str = None,
api_base: str = api_base,
**kwargs
) -> AsyncResult:
api_key = os.environ.get("CLAUDE_COOKIE", api_key)
if not api_key:
raise MissingAuthError("Claude cookie not found. Please set the 'CLAUDE_COOKIE' environment variable.")
if not cls.organization_id:
cls.organization_id = os.environ.get("CLAUDE_ORGANIZATION_ID")
if not cls.organization_id:
raise MissingAuthError("Claude organization ID not found. Please set the 'CLAUDE_ORGANIZATION_ID' environment variable.")
async for chunk in super().create_async_generator(
model=model,
messages=messages,
api_base=f"{api_base}/{cls.organization_id}",
headers={"cookie": api_key},
**kwargs
):
yield chunk |