from adaline_api.models.log_span_retrieval_content import LogSpanRetrievalContent

from adaline_api.models.log_span_content import LogSpanContent
from adaline_api.models.log_span_retrieval_content import LogSpanRetrievalContent

content = LogSpanContent(
    actual_instance=LogSpanRetrievalContent(
        type="Retrieval",
        input=json.dumps({"query": "How does auth work?", "top_k": 5}),
        output=json.dumps({"documents": [{"id": "doc1", "score": 0.95}]}),
    )
)

import json
from adaline_api.models.log_span_content import LogSpanContent
from adaline_api.models.log_span_retrieval_content import LogSpanRetrievalContent

retrieval_input = {"query": "refund policy", "top_k": 3, "namespace": "support"}
retrieval_output = {
    "documents": [
        {"id": "doc_42", "score": 0.97, "text": "Refunds are processed within 5 days."},
        {"id": "doc_18", "score": 0.91, "text": "Contact support for refund requests."},
    ]
}

span.update({
    "status": "success",
    "content": LogSpanContent(
        actual_instance=LogSpanRetrievalContent(
            type="Retrieval",
            input=json.dumps(retrieval_input),
            output=json.dumps(retrieval_output),
        )
    ),
})