Fix: Accept provider enum (#24)

echolabsdev · Oct 14, 2024 · 339c4dc · 339c4dc
1 parent 397fa8c
commit 339c4dc
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/src/Concerns/HasProvider.php b/src/Concerns/HasProvider.php
@@ -5,12 +5,13 @@
 namespace EchoLabs\Prism\Concerns;
 
 use EchoLabs\Prism\Contracts\Provider;
+use EchoLabs\Prism\Enums\Provider as ProviderEnum;
 
 trait HasProvider
 {
  protected Provider $provider;
 
- public function using(string $provider, string $model): self
+ public function using(ProviderEnum|string $provider, string $model): self
  {
  $this->provider = app('prism-manager')->resolve($provider);
  $this->provider->usingModel($model);

diff --git a/tests/Drivers/AnthropicTextTest.php → tests/Providers/AnthropicTextTest.php b/tests/Drivers/AnthropicTextTest.php → tests/Providers/AnthropicTextTest.php
@@ -2,7 +2,7 @@
 
 declare(strict_types=1);
 
-namespace Tests\Drivers;
+namespace Tests\Providers;
 
 use EchoLabs\Prism\Facades\Tool;
 use EchoLabs\Prism\Prism;

diff --git a/tests/Drivers/OllamaTextTest.php → tests/Providers/OllamaTextTest.php b/tests/Drivers/OllamaTextTest.php → tests/Providers/OllamaTextTest.php
diff --git a/tests/Drivers/OpenAITextTest.php → tests/Providers/OpenAITextTest.php b/tests/Drivers/OpenAITextTest.php → tests/Providers/OpenAITextTest.php
@@ -2,7 +2,7 @@
 
 declare(strict_types=1);
 
-namespace Tests\Drivers;
+namespace Tests\Providers;
 
 use EchoLabs\Prism\Facades\Tool;
 use EchoLabs\Prism\Prism;