Quickstart

import os
os.environ["FI_API_KEY"] = "YOUR_API_KEY"
os.environ["FI_SECRET_KEY"] = "YOUR_SECRET_KEY"

process.env.FI_API_KEY = "YOUR_API_KEY";
process.env.FI_SECRET_KEY = "YOUR_SECRET_KEY";

from fi_instrumentation import register, Transport
from fi_instrumentation.fi_types import ProjectType, EvalName, EvalTag, EvalTagType, EvalSpanKind, ModelChoices

# Setup OTel via our register function
trace_provider = register(
    project_type=ProjectType.EXPERIMENT,  
    project_name="FUTURE_AGI",            # Your project name
    project_version_name="openai-exp",    # Version identifier for this prototype
    transport=Transport.HTTP,             # Transport mechanism for your traces
    eval_tags = [
        EvalTag(
            eval_name=EvalName.TONE,
            value=EvalSpanKind.LLM,
            type=EvalTagType.OBSERVATION_SPAN,
            model=ModelChoices.TURING_LARGE,
            mapping={
                'input' : 'llm.input_messages'
            },
            custom_eval_name="<custom_eval_name2>",
        ),
    ]
)

import { register, Transport, ProjectType, EvalName, EvalTag, EvalTagType, EvalSpanKind, ModelChoices } from "@traceai/fi-core";

const tracerProvider = await register({
    projectName: "FUTURE_AGI",
    projectType: ProjectType.EXPERIMENT,
    transport: Transport.HTTP,
    projectVersionName: "openai-exp", // Version identifier for this prototype
    evalTags: [
      await EvalTag.create({
        type: EvalTagType.OBSERVATION_SPAN,
        value: EvalSpanKind.LLM,
        eval_name: EvalName.CHUNK_ATTRIBUTION,
        custom_eval_name: "Chunk_Attribution",
        mapping: {
          "context": "raw.input",
          "output": "raw.output"
        },
        model: ModelChoices.TURING_SMALL
      })
    ]
});

pip install traceAI-openai

npm install @traceai/openai

from traceai_openai import OpenAIInstrumentor

OpenAIInstrumentor().instrument(tracer_provider=trace_provider)

import { OpenAIInstrumentation } from "@traceai/openai";
import { registerInstrumentations } from "@opentelemetry/instrumentation";

const openaiInstrumentation = new OpenAIInstrumentation({});

registerInstrumentations({
  instrumentations: [openaiInstrumentation],
  tracerProvider: tracerProvider,
});

from openai import OpenAI

os.environ["OPENAI_API_KEY"] = "your-openai-api-key"

client = OpenAI()

completion = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {
            "role": "user",
            "content": "Write a one-sentence bedtime story about a unicorn."
        }
    ]
)

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

import { OpenAI } from "openai";

const client = new OpenAI({
    apiKey: process.env.OPENAI_API_KEY,
});

const completion = await client.chat.completions.create({
    model: "gpt-4o",
    messages: [
        {
            "role": "user",
            "content": "Write a one-sentence bedtime story about a unicorn."
        }
    ]
});

console.log(completion.choices[0].message.content);