from adaline_api.models.log_span_model_stream_content import LogSpanModelStreamContent

from adaline_api.models.log_span_content import LogSpanContent
from adaline_api.models.log_span_model_stream_content import LogSpanModelStreamContent

content = LogSpanContent(
    actual_instance=LogSpanModelStreamContent(
        type="ModelStream",
        provider="anthropic",
        model="claude-sonnet-4-20250514",
        input=json.dumps(params),
        output=raw_chunks,
        aggregate_output=json.dumps(aggregated),
    )
)

import json
from adaline_api.models.log_span_content import LogSpanContent
from adaline_api.models.log_span_model_stream_content import LogSpanModelStreamContent

params = {
    "model": "gpt-4o",
    "messages": [{"role": "user", "content": "Write a haiku."}],
    "stream": True,
}

span.update({
    "status": "success",
    "content": LogSpanContent(
        actual_instance=LogSpanModelStreamContent(
            type="ModelStream",
            provider="openai",
            model="gpt-4o",
            input=json.dumps(params),
            output=collected_chunks,
            aggregate_output=json.dumps({"role": "assistant", "content": full_text}),
            cost=0.003,
        )
    ),
})