<?php

namespace app\common\service\ai;

class AiAssistantTrace
{
    public string $originalMessage = '';
    public array $historyMessages = [];
    public array $plan = [];
    public array $retrievals = [
        'knowledge_hits' => [],
        'match_items' => [],
        'crypto_items' => [],
        'web_news_items' => [],
    ];
    public array $rankedHits = [];
    public array $finalSources = [];
    public string $contextText = '{}';
    public string $finalAnswer = '';
    public array $responseResult = [];
    public bool $hasContext = false;

    public function __construct(string $originalMessage, array $historyMessages = [])
    {
        $this->originalMessage = $originalMessage;
        $this->historyMessages = $historyMessages;
    }

    public function toArray(): array
    {
        return [
            'original_message' => $this->originalMessage,
            'history_messages' => $this->historyMessages,
            'plan' => $this->plan,
            'retrievals' => $this->retrievals,
            'ranked_hits' => $this->rankedHits,
            'final_sources' => $this->finalSources,
            'context_text' => $this->contextText,
            'final_answer' => $this->finalAnswer,
            'response_result' => $this->responseResult,
            'has_context' => $this->hasContext,
        ];
    }
}