1. Installation

Install the traceAI and Portkey packages.

pip install portkey_ai traceAI-portkey 

2. Set Environment Variables

Set up your environment variables to authenticate with both FutureAGI and Portkey.

import os

os.environ["FI_API_KEY"] = "your-futureagi-api-key"
os.environ["FI_SECRET_KEY"] = "your-futureagi-secret-key"
os.environ["PORTKEY_VIRTUAL_KEY"] = "your-portkey-virtual-key"

3. Initialize Trace Provider

Set up the trace provider to create a new project in FutureAGI, establish telemetry data pipelines .

from fi_instrumentation import register
from fi_instrumentation.fi_types import ProjectType

trace_provider = register(
    project_type=ProjectType.OBSERVE,
    project_name="portkey_project",
)

4. Instrument your Project

Instrument your project to enable automatic tracing.

from traceai_portkey import PortkeyInstrumentor

PortkeyInstrumentor().instrument(tracer_provider=tracer_provider)

5. Interact with Portkey

Interact with Portkey as you normally would. Our Instrumentor will automatically trace and send the telemetry data to our platform.

client = Portkey(virtual_key=os.environ["PORTKEY_VIRTUAL_KEY"])

completion = client.chat.completions.create(
    model="gpt-4o",
    messages=[{"role": "user", "content": "Write a 6-word story about a robot who discovers music."}]
)

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