<?php

namespace app\common\service\ai;

class AiAssistantRetriever
{
    public static function retrieve(AiAssistantTrace $trace, int $userId, int $sessionId): AiAssistantTrace
    {
        $context = AiAssistantContextService::build($trace->originalMessage, $userId, $sessionId, $trace->plan);

        $trace->retrievals = [
            'knowledge_hits' => $context['knowledge_hits'] ?? [],
            'match_items' => $context['match_items'] ?? [],
            'crypto_items' => $context['crypto_items'] ?? [],
        ];
        $trace->hasContext = !empty($context['has_context']);

        return $trace;
    }
}