Haystack

pip install zespan haystack-ai

import zespan
from zespan import ZespanHaystackTracer
import haystack.tracing
from haystack import Pipeline
from haystack.components.generators import OpenAIGenerator
from haystack.components.builders import PromptBuilder

zespan.init(api_key="zsp_your_api_key_here")

template = """
Given the following question, provide a concise answer.
Question: {{ question }}
"""

pipeline = Pipeline()
pipeline.add_component("prompt_builder", PromptBuilder(template=template))
pipeline.add_component("llm", OpenAIGenerator(model="gpt-4o"))
pipeline.connect("prompt_builder", "llm")

tracer = ZespanHaystackTracer(client=zespan.get_client())
haystack.tracing.enable_tracing(tracer)

result = pipeline.run(
    {"prompt_builder": {"question": "What is retrieval-augmented generation?"}},
    include_outputs_from={"llm"},
)
print(result["llm"]["replies"][0])