Quickstart

pip install pikarc

from pikarc import AsyncPikarc, PikarcBlockedError
from openai import AsyncOpenAI

openai = AsyncOpenAI()
guard = AsyncPikarc(
    api_key="lg_xxxx_your_api_key",
    base_url="http://localhost:8000",
)

async def chat(user_id: str, message: str) -> str:
    try:
        async with guard.run(user_id=user_id) as run:
            response = await run.model_call(
                fn=lambda: openai.chat.completions.create(
                    model="gpt-4o",
                    messages=[{"role": "user", "content": message}],
                ),
                model="gpt-4o",
                input_data={"messages": [{"role": "user", "content": message}]},
                token_extractor=lambda r: (
                    r.usage.prompt_tokens,
                    r.usage.completion_tokens,
                ),
            )
            return response.choices[0].message.content or ""

    except PikarcBlockedError as e:
        return f"Blocked: {e.reason}"

    finally:
        await guard.close()