helpers.php

<?php

declare(strict_types=1);

namespace Cortex\Support;

use Closure;
use Cortex\Cortex;
use Cortex\Prompts\Prompt;
use Cortex\LLM\Contracts\LLM;
use Cortex\Tools\ClosureTool;
use Cortex\Tasks\Enums\TaskType;
use Cortex\Tasks\Builders\TextTaskBuilder;
use Cortex\Prompts\Contracts\PromptBuilder;
use Cortex\LLM\Data\Messages\MessageCollection;
use Cortex\Tasks\Builders\StructuredTaskBuilder;

/**
 * Helper function to create a chat prompt builder.
 *
 * @param MessageCollection|array<int, \Cortex\LLM\Contracts\Message>|string|null $messages
 *
 * @return ($messages is null ? \Cortex\Prompts\Prompt : ($messages is string ? \Cortex\Prompts\Builders\TextPromptBuilder : \Cortex\Prompts\Builders\ChatPromptBuilder))
 */
function prompt(MessageCollection|array|string|null $messages): Prompt|PromptBuilder
{
    return Cortex::prompt($messages);
}

/**
 * Helper function to create an LLM instance.
 */
function llm(?string $provider = null, ?string $model = null): LLM
{
    return Cortex::llm($provider, $model);
}

/**
 * Helper function to build tasks.
 */
function task(?string $name = null, TaskType $type = TaskType::Text): TextTaskBuilder|StructuredTaskBuilder
{
    return Cortex::task($name, $type);
}

/**
 * Helper function to create a closure tool with a name and optional description.
 */
function tool(string $name, string $description, Closure $closure): ClosureTool
{
    return new ClosureTool($closure, $name, $description);
}