LLM traces for debugging

import os
from orq_ai_sdk import Orq
from orq_ai_sdk.traced import traced

@traced(name="process_document", type="agent")
def process_document(doc_id: str):
    result = {"status": "processed", "doc_id": doc_id}
    return result

@traced(name="fetch_context", type="retrieval")
def fetch_relevant_context(query: str):
    return {"context": "relevant information", "sources": ["doc1", "doc2"]}

@traced(name="format_response", type="function")
def format_response(raw_output: str, metadata: dict):
    return {"formatted": raw_output.strip(), "metadata": metadata}

@traced(name="document_pipeline", type="agent")
def document_pipeline(orq, doc_id: str):
    doc = process_document(doc_id)
    context = fetch_relevant_context("user query")

    response = orq.agents.responses.create(
        agent_key="simple-agent",
        background=False,
        message={
            "role": "user",
            "parts": [
                {
                    "kind": "text",
                    "text": f"Summarize this document: {doc['status']}. Context: {context['context']}"
                }
            ]
        }
    )

    return format_response(response.output[0].parts[0].text, {"doc_id": doc_id})

with Orq(api_key=os.getenv("ORQ_API_KEY")) as orq:
    result = document_pipeline(orq, "doc-123")
    print(result)