Observability for Watsonx with Highlight

Step 1

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

from ibm_watson_machine_learning.foundation_models import Model

from traceloop.sdk import Traceloop

Traceloop.init(app_name="watsonx_example")


def get_credentials(api_key):
    return {
        "url": "https://us-south.ml.cloud.ibm.com",
        "apikey": api_key,
    }


iam_api_key = os.environ["IAM_API_KEY"]
project_id = os.environ["PROJECT_ID"]

prompt_input = """Calculate result

Input:
what is the capital of China.

Output:
"""

model_id = "ibm/granite-3.0-8b-instruct"
parameters = {
    "decoding_method": "sample",
    "max_new_tokens": 60,
    "min_new_tokens": 10,
    "random_seed": 111,
    "temperature": 0.9,
    "top_k": 50,
    "top_p": 1,
    "repetition_penalty": 2
}

model = Model(
    model_id=model_id,
    params=parameters,
    credentials=get_credentials(iam_api_key),
    project_id=project_id
    )

prompt_input = "What is the the captical of China"
generated_response = model.generate(prompt=prompt_input)

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

Discover use cases

Trace prompts and completions in Watsonx

Call Watsonx and see prompts, completions, and token usage for your call.

See Code Example

Trace your RAG retrieval pipeline

Build a RAG pipeline with Chroma and OpenAI. See vectors returned from Chroma, full prompt in OpenAI and responses

See Code Example

Observability for Watsonx with Highlight

Step 1

Step 2

Discover use cases

Trace prompts and completions in Watsonx

Trace your RAG retrieval pipeline

More things you can connect to {platform}

Chroma

OpenAI

Cohere

Ollama

Gemini

Bedrock

Anthropic

LangChain

LlamaIndex

Mistral AI

together.ai