from trainly import TrainlyClient
from openai import OpenAI

client = TrainlyClient(api_key="tk_...", project_id="proj_...")
openai_client = OpenAI()

@client.observe(model="gpt-4o", tags=["production"])
def ask(question: str) -> str:
    response = openai_client.chat.completions.create(
        model="gpt-4o",
        messages=[{"role": "user", "content": question}]
    )
    return response.choices[0].message.content

# Every call is now traced automatically
result = ask("What is AI observability?")

import { TrainlyProvider, useTrainlyObserve } from '@trainly/react';

function App() {
  return (
    <TrainlyProvider projectId="proj_..." apiKey="tk_...">
      <MyComponent />
    </TrainlyProvider>
  );
}

function MyComponent() {
  const { trace } = useTrainlyObserve({ model: 'gpt-4o' });

  async function handleQuery() {
    const output = await callMyAI("Hello");
    await trace("Hello", output, { tags: ['demo'] });
  }

  return <button onClick={handleQuery}>Ask AI</button>;
}