From 7266533af857d5d476cb43ec402f237ee6d3f3fe Mon Sep 17 00:00:00 2001
From: Christopher Hertel <mail@christopher-hertel.de>
Date: Mon, 21 Apr 2025 11:33:53 +0200
Subject: [PATCH] test: fix namespaces and coverage

---
 tests/Bridge/Google/GooglePromptConverterTest.php | 2 ++
 tests/Bridge/HuggingFace/ModelClientTest.php      | 4 ++++
 tests/Bridge/Meta/LlamaPromptConverterTest.php    | 3 +--
 tests/Model/Message/Content/ImageTest.php         | 2 +-
 4 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/tests/Bridge/Google/GooglePromptConverterTest.php b/tests/Bridge/Google/GooglePromptConverterTest.php
index a70947b..1b695aa 100644
--- a/tests/Bridge/Google/GooglePromptConverterTest.php
+++ b/tests/Bridge/Google/GooglePromptConverterTest.php
@@ -6,6 +6,7 @@
 
 use PhpLlm\LlmChain\Bridge\Google\GooglePromptConverter;
 use PhpLlm\LlmChain\Model\Message\AssistantMessage;
+use PhpLlm\LlmChain\Model\Message\Content\File;
 use PhpLlm\LlmChain\Model\Message\Content\Image;
 use PhpLlm\LlmChain\Model\Message\Message;
 use PhpLlm\LlmChain\Model\Message\MessageBag;
@@ -22,6 +23,7 @@
 #[UsesClass(UserMessage::class)]
 #[UsesClass(AssistantMessage::class)]
 #[UsesClass(MessageBag::class)]
+#[UsesClass(File::class)]
 final class GooglePromptConverterTest extends TestCase
 {
     #[Test]
diff --git a/tests/Bridge/HuggingFace/ModelClientTest.php b/tests/Bridge/HuggingFace/ModelClientTest.php
index fc1ad76..d58c977 100644
--- a/tests/Bridge/HuggingFace/ModelClientTest.php
+++ b/tests/Bridge/HuggingFace/ModelClientTest.php
@@ -14,10 +14,14 @@
 use PhpLlm\LlmChain\Model\Model as BaseModel;
 use PHPUnit\Framework\Attributes\CoversClass;
 use PHPUnit\Framework\Attributes\DataProvider;
+use PHPUnit\Framework\Attributes\Small;
+use PHPUnit\Framework\Attributes\UsesClass;
 use PHPUnit\Framework\TestCase;
 use Symfony\Component\HttpClient\MockHttpClient;
 
 #[CoversClass(ModelClient::class)]
+#[Small]
+#[UsesClass(Model::class)]
 final class ModelClientTest extends TestCase
 {
     public function testSupportsWithHuggingFaceModel(): void
diff --git a/tests/Bridge/Meta/LlamaPromptConverterTest.php b/tests/Bridge/Meta/LlamaPromptConverterTest.php
index 86b7ae3..74e8fb4 100644
--- a/tests/Bridge/Meta/LlamaPromptConverterTest.php
+++ b/tests/Bridge/Meta/LlamaPromptConverterTest.php
@@ -6,7 +6,6 @@
 
 use PhpLlm\LlmChain\Bridge\Meta\LlamaPromptConverter;
 use PhpLlm\LlmChain\Model\Message\AssistantMessage;
-use PhpLlm\LlmChain\Model\Message\Content\Image;
 use PhpLlm\LlmChain\Model\Message\Content\ImageUrl;
 use PhpLlm\LlmChain\Model\Message\Content\Text;
 use PhpLlm\LlmChain\Model\Message\Message;
@@ -23,7 +22,7 @@
 #[CoversClass(LlamaPromptConverter::class)]
 #[Small]
 #[UsesClass(AssistantMessage::class)]
-#[UsesClass(Image::class)]
+#[UsesClass(ImageUrl::class)]
 #[UsesClass(Text::class)]
 #[UsesClass(Message::class)]
 #[UsesClass(MessageBag::class)]
diff --git a/tests/Model/Message/Content/ImageTest.php b/tests/Model/Message/Content/ImageTest.php
index 1ffc7c9..42b5c8f 100644
--- a/tests/Model/Message/Content/ImageTest.php
+++ b/tests/Model/Message/Content/ImageTest.php
@@ -2,7 +2,7 @@
 
 declare(strict_types=1);
 
-namespace Model\Message\Content;
+namespace PhpLlm\LlmChain\Tests\Model\Message\Content;
 
 use PhpLlm\LlmChain\Model\Message\Content\Image;
 use PHPUnit\Framework\Attributes\CoversClass;