xref: /plugin/aichat/Model/Ollama/EmbeddingModel.php (revision e7611489c85f333e5e56d206ee2182115fd03f58) 
1<?php
2
3namespace dokuwiki\plugin\aichat\Model\Ollama;
4
5use dokuwiki\plugin\aichat\Model\EmbeddingInterface;
6
7class EmbeddingModel extends AbstractOllama implements EmbeddingInterface
8{
9    /** @inheritdoc */
10    public function getEmbedding($text): array
11    {
12        $data = [
13            'model' => $this->getModelName(),
14            'prompt' => $text,
15        ];
16        $response = $this->request('embeddings', $data);
17
18        return $response['embedding'];
19    }
20}
21