Observability for Cohere with Highlight

Step 1

Install the Traceloop SDK and initialize it. It will automatically log all calls to Cohere, with prompts and completions as separate spans.

import boto3
import json

from traceloop.sdk import Traceloop
from traceloop.sdk.decorators import task, workflow

Traceloop.init(app_name="joke_generation_service")
co = cohere.Client(os.environ.get("COHERE_API_KEY"))

import * as traceloop from "@traceloop/node-server-sdk";
import { CohereClient } from "cohere-ai";

traceloop.initialize({
  appName: "sample_cohere",
  apiKey: process.env.TRACELOOP_API_KEY,
  disableBatch: true,
});

const cohere = new CohereClient({
  token: process.env.COHERE_API_KEY ?? "",
});

const sampleGenerate = async () => {
  return await traceloop.withWorkflow({ name: "sample_generate" }, async () => {
    const response = await cohere.generate({
      model: "command-light",
      prompt: "What happened to Pluto?",
      k: 1,
      temperature: 2,
    });

    return response.generations?.[0].text;
  });
};

traceloop.withAssociationProperties({}, async () => {
  console.log(await sampleGenerate());
});

Step 2

Route traces to Highlights’s OTLP endpoint and set the Highlight project ID in the headers.

TRACELOOP_BASE_URL=https://otel.highlight.io:4318
TRACELOOP_HEADERS="x-highlight-project=<YOUR_HIGHLIGHT_PROJECT_ID>"

Docs

Observability for Cohere with Highlight

Step 1

Step 2

Discover use cases

Trace prompts and completions in Cohere

More things you can connect to {platform}

Chroma

OpenAI

Ollama

Gemini

Watsonx

Bedrock

Anthropic

LangChain

LlamaIndex

Mistral AI

together.ai