SmartAssistant/files/custom/Espo/Modules/SmartAssistant/Services/AssistantPromptContextProvider.php

<?php

namespace Espo\Modules\SmartAssistant\Services;

use Espo\ORM\EntityManager;
use Espo\Core\Utils\Log;

/**
 * Loads active prompt sections keyed by their `key` field, filtered by mode.
 *
 * Returned shape: `["tool_rules" => "...", "writing_style" => "...", ...]`.
 * The Python prompt_builder treats a missing key (or empty string) as
 * "use the in-code default" — so this provider is purely additive.
 */
class AssistantPromptContextProvider
{
    private EntityManager $entityManager;
    private Log $log;

    public function __construct(EntityManager $entityManager, Log $log)
    {
        $this->entityManager = $entityManager;
        $this->log = $log;
    }

    /**
     * @return array<string, string>  key => content
     */
    public function getPromptsForMode(string $mode): array
    {
        $entries = $this->entityManager->getRDBRepository('AssistantPrompt')
            ->where([
                'isActive' => true,
                'deleted' => false,
                'mode' => ['both', $mode],
            ])
            ->order('displayOrder', 'ASC')
            ->find();

        $out = [];
        foreach ($entries as $entry) {
            $key = (string) $entry->get('key');
            $content = (string) ($entry->get('content') ?? '');
            if ($key === '' || $content === '') {
                continue;
            }
            $out[$key] = $content;
        }

        return $out;
    }
}