from posthog import Posthog
from posthog.ai import Prompts
from posthog.ai.openai import OpenAI

posthog = Posthog(
    "phc_xxx",
    host="https://us.i.posthog.com",
    personal_api_key="phx_xxx",
)

prompts = Prompts(posthog)
openai = OpenAI(api_key="sk-xxx", posthog=posthog)

template = prompts.get(
    "support-system-prompt",
    fallback="You are a helpful assistant.",
)

system_prompt = prompts.compile(template, {
    "company": "Acme Corp",
    "tier": "premium",
})

response = openai.chat.completions.create(
    model="gpt-4",
    messages=[
        {"role": "system", "content": system_prompt},
        {"role": "user", "content": "How do I reset my password?"},
    ],
    posthog_distinct_id="user-123",
)

print(response.choices[0].message.content)