embedding added

app/Http/Controllers/CollectionController.php

@@ -64,7 +64,6 @@ public function update(Collection $collection)
         return to_route('collections.show', $collection);
     }
 
-
     public function show(Collection $collection)
     {
         $chatResource = $collection->chats()->where('user_id', auth()->user()->id)

app/LlmDriver/OllamaClient.php

@@ -4,6 +4,7 @@
 
 use App\LlmDriver\Requests\MessageInDto;
 use App\LlmDriver\Responses\CompletionResponse;
+use App\LlmDriver\Responses\EmbeddingsResponseDto;
 use Illuminate\Contracts\Container\BindingResolutionException;
 use Illuminate\Support\Facades\Http;
 use Illuminate\Support\Facades\Log;
@@ -12,6 +13,23 @@ class OllamaClient extends BaseClient
 {
     protected string $driver = 'ollama';
 
+    public function embedData(string $prompt): EmbeddingsResponseDto
+    {
+        Log::info('LlmDriver::Ollama::completion');
+
+        $response = $this->getClient()->post('/embeddings', [
+            'model' => $this->getConfig('ollama')['models']['embedding_model'],
+            'prompt' => $prompt,
+        ]);
+
+        $results = $response->json();
+
+        return EmbeddingsResponseDto::from([
+            'embedding' => data_get($results, 'embedding'),
+            'token_count' => 1000,
+        ]);
+    }
+
     /**
      * @param  MessageInDto[]  $messages
      *

config/llmdriver.php

@@ -31,6 +31,7 @@
             'models' => [
                 //@see https://github.com/ollama/ollama/blob/main/docs/openai.md
                 'completion_model' => env('CLAUDE_COMPLETION_MODEL', 'llama2'),
+                'embedding_model' => env('CLAUDE_EMBEDDING_MODEL', 'llama2'),
             ],
         ],
     ],

resources/js/Pages/Collection/Show.vue

@@ -36,7 +36,6 @@ const closeEditCollectionSlideOut = () => {
     showEditCollection.value = false;
 };
 
-
 onMounted(() => {
     Echo.private(`collection.${props.collection.data.id}`)
     .listen('.status', (e) => {

tests/Feature/Http/Controllers/CollectionControllerTest.php

@@ -69,7 +69,7 @@ public function test_update(): void
             'description' => 'Test Description',
         ])->assertStatus(302);
         $this->assertDatabaseCount('collections', 1);
-        
+
         $this->assertEquals(DriversEnum::Claude, $collection->refresh()->embedding_driver);
 
     }

tests/Feature/OllamaClientTest.php

@@ -2,7 +2,6 @@
 
 namespace Tests\Feature;
 
-use App\LlmDriver\ClaudeClient;
 use App\LlmDriver\OllamaClient;
 use App\LlmDriver\Requests\MessageInDto;
 use App\LlmDriver\Responses\CompletionResponse;
@@ -17,9 +16,14 @@ class OllamaClientTest extends TestCase
      */
     public function test_embeddings(): void
     {
-        $this->markTestSkipped('@TODO: Requires another server');
 
-        $client = new ClaudeClient();
+        $client = new OllamaClient();
+
+        $data = get_fixture('ollama_embedings.json');
+
+        Http::fake([
+            '127.0.0.1:11434/*' => Http::response($data, 200),
+        ]);
 
         $results = $client->embedData('test');