Govern AI Agents in Real-Time

from pikarc import AsyncPikarc, PikarcBlockedError

guard = AsyncPikarc(
    api_key="lg_xxxx_yyyy",
    base_url="https://api.pikarc.dev",
)

try:
    async with guard.run(user_id="alice") as run:
        response = await run.model_call(
            fn=lambda: openai.chat.completions.create(
                model="gpt-4o",
                messages=[{"role": "user", "content": query}],
            ),
            model="gpt-4o",
            token_extractor=lambda r: (
                r.usage.prompt_tokens,
                r.usage.completion_tokens,
            ),
        )

        result = await run.tool_call(
            fn=lambda: search_database(query),
            tool_name="search_database",
        )

except PikarcBlockedError as e:
    print(f"Blocked: {e.reason}")