Cortex.php

<?php

declare(strict_types=1);

namespace Cortex;

use Cortex\Facades\LLM;
use Cortex\Prompts\Prompt;
use Cortex\Facades\Embeddings;
use Cortex\Tasks\Enums\TaskType;
use Cortex\Tasks\Builders\TextTaskBuilder;
use Cortex\Prompts\Contracts\PromptBuilder;
use Cortex\LLM\Contracts\LLM as LLMContract;
use Cortex\LLM\Data\Messages\MessageCollection;
use Cortex\Tasks\Builders\StructuredTaskBuilder;
use Cortex\Embeddings\Contracts\Embeddings as EmbeddingsContract;

class Cortex
{
    /**
     * Create a prompt builder or factory.
     *
     * @param \Cortex\LLM\Data\Messages\MessageCollection|array<int, \Cortex\LLM\Contracts\Message|\Cortex\LLM\Data\Messages\MessagePlaceholder>|string|null $messages
     *
     * @return ($messages is null ? \Cortex\Prompts\Prompt : ($messages is string ? \Cortex\Prompts\Builders\TextPromptBuilder : \Cortex\Prompts\Builders\ChatPromptBuilder))
     */
    public static function prompt(
        MessageCollection|array|string|null $messages = null,
    ): Prompt|PromptBuilder {
        if (func_num_args() === 0) {
            return new Prompt();
        }

        return is_string($messages)
            ? Prompt::builder('text')->text($messages)
            : Prompt::builder('chat')->messages($messages);
    }

    /**
     * Create an LLM instance.
     */
    public static function llm(?string $provider = null, ?string $model = null): LLMContract
    {
        $llm = LLM::provider($provider);

        if ($model !== null) {
            $llm->withModel($model);
        }

        return $llm;
    }

    public static function embeddings(?string $driver = null, ?string $model = null): EmbeddingsContract
    {
        $embeddings = Embeddings::driver($driver);

        if ($model !== null) {
            $embeddings->withModel($model);
        }

        return $embeddings;
    }

    /**
     * Create a new task builder with the given name and type.
     */
    public static function task(
        ?string $name = null,
        TaskType $type = TaskType::Text,
    ): TextTaskBuilder|StructuredTaskBuilder {
        $builder = $type->builder();

        if ($name !== null) {
            $builder->name($name);
        }

        return $builder;
    }
}