feat(Context.php): add getModelParameters method to Context class

feat(InvalidArgumentException.php): add InvalidArgumentException class
feat(LLamaCPP.php): add validation for embedding generation in createEmbedding method

test(LLamaCPPTest.php): add test case for createEmbedding method with embedding set to false and expect InvalidArgumentException

src/Context.php

@@ -49,4 +49,12 @@ public function getCtx(): CData
     {
         return $this->ctx;
     }
+
+    /**
+     * @return ModelParameters
+     */
+    public function getModelParameters(): ModelParameters
+    {
+        return $this->modelParameters;
+    }
 }

src/Exception/InvalidArgumentException.php

@@ -0,0 +1,7 @@
+<?php
+
+namespace Kambo\LLamaCPP\Exception;
+
+final class InvalidArgumentException extends LLamaCppException
+{
+}

src/LLamaCPP.php

@@ -6,6 +6,7 @@
 use Symfony\Component\EventDispatcher\EventDispatcherInterface;
 use Kambo\LLamaCPP\Native\LLamaCPPFFI;
 use Kambo\LLamaCPP\Events\TokenGeneratedEvent;
+use Kambo\LLamaCPP\Exception\InvalidArgumentException;
 use Generator;
 
 use function strlen;
@@ -95,6 +96,10 @@ public function generateAll(string $prompt, ?GenerationParameters $generation =
 
     public function createEmbedding(string $text, int $noOfThreads = 10): array
     {
+        if (!$this->context->getModelParameters()->isEmbedding()) {
+            throw new InvalidArgumentException('Generation must of embedding must be turned on.');
+        }
+
         $input  = $this->ffi->newArray('llama_token', strlen($text));
         $nOfTok = $this->ffi->llama_tokenize($this->context->getCtx(), $text, $input, strlen($text), true);
 

tests/LLamaCPPTest.php

@@ -7,6 +7,8 @@
 use Kambo\LLamaCPP\Context;
 use Kambo\LLamaCPP\LLamaCPP;
 use Kambo\LLamaCPP\Native\LLamaCPPFFI;
+use Kambo\LLamaCPP\Parameters\ModelParameters;
+use Kambo\LLamaCPP\Exception\InvalidArgumentException;
 use Symfony\Component\EventDispatcher\EventDispatcherInterface;
 use Generator;
 use FFI;
@@ -106,6 +108,13 @@ public function testGenerateAll()
 
     public function testCreateEmbedding()
     {
+        $modelParameters = new ModelParameters(
+            modelPath: 'test',
+            embedding: true,
+        );
+        $this->contextMock->method('getModelParameters')
+            ->willReturn($modelParameters);
+
         $this->contextMock->method('getCtx')
             ->willReturn(FFI::new('int'));
 
@@ -139,4 +148,19 @@ public function testCreateEmbedding()
             $result
         );
     }
+
+    public function testCreateEmbeddingFail()
+    {
+        $modelParameters = new ModelParameters(
+            modelPath: 'test',
+            embedding: false,
+        );
+        $this->contextMock->method('getModelParameters')
+            ->willReturn($modelParameters);
+
+        $llamaCPP = new LLamaCPP($this->contextMock, $this->eventDispatcherMock, $this->ffiMock);
+
+        $this->expectException(InvalidArgumentException::class);
+        $llamaCPP->createEmbedding('test');
+    }
 }