simplify ai provider interface

Author: ilvalerione

src/Providers/AIProviderInterface.php

@@ -3,7 +3,6 @@
 namespace NeuronAI\Providers;
 
 use NeuronAI\Messages\Message;
-use NeuronAI\Providers\Embeddings\EmbeddingsProviderInterface;
 
 interface AIProviderInterface
 {
@@ -38,14 +37,14 @@ public function chat(array|string $prompt): Message;
     /**
      * The context window limitation of the LLM.
      *
-     * @return int
+     * @return ?int
      */
-    public function contextWindow(): int;
+    //public function contextWindow(): ?int;
 
     /**
      * The maximum number of tokens to generate before stopping.
      *
-     * @return int
+     * @return ?int
      */
-    public function maxCompletionTokens(): int;
+    //public function maxCompletionTokens(): ?int;
 }

src/Providers/Anthropic.php

@@ -2,7 +2,10 @@
 
 namespace NeuronAI\Providers;
 
+use GuzzleHttp\Exception\GuzzleException;
+use NeuronAI\Messages\AssistantMessage;
 use NeuronAI\Messages\Message;
+use NeuronAI\Messages\UserMessage;
 use NeuronAI\Tools\Tool;
 use NeuronAI\Tools\ToolInterface;
 use NeuronAI\Tools\ToolCallMessage;
@@ -40,9 +43,8 @@ class Anthropic implements AIProviderInterface
     public function __construct(
         protected string $key,
         protected string $model,
-        protected string $version,
-        protected int $max_tokens,
-        protected int $context_window,
+        protected string $version = '2023-06-01',
+        protected int $max_tokens = 8192,
         protected ?float $temperature = null,
         protected ?array $stop_sequences = null,
     ) {
@@ -68,14 +70,14 @@ public function systemPrompt(string $prompt): AIProviderInterface
     /**
      * Send a prompt to the AI agent.
      *
-     * @param array|string $prompt
-     * @return Message
-     * @throws \Exception
+     * @throws GuzzleException
      */
     public function chat(array|string $prompt): Message
     {
         if (\is_string($prompt)) {
-            $prompt = [['role' => 'user', 'content' => $prompt]];
+            $prompt = [
+                new UserMessage($prompt)
+            ];
         }
 
         $json = \array_filter([
@@ -106,7 +108,7 @@ public function chat(array|string $prompt): Message
                 $content['input']
             );
         } else {
-            $response = new Message('assistant', \last($result['content'])['text']);
+            $response = new AssistantMessage(\last($result['content'])['text']);
         }
 
         // Attach the usage for the current interaction
@@ -122,26 +124,6 @@ public function chat(array|string $prompt): Message
         return $response;
     }
 
-    /**
-     * The context window limitation of the LLM.
-     *
-     * @return int
-     */
-    public function contextWindow(): int
-    {
-        return $this->context_window;
-    }
-
-    /**
-     * The maximum number of tokens to generate before stopping.
-     *
-     * @return int
-     */
-    public function maxCompletionTokens(): int
-    {
-        return  $this->max_tokens;
-    }
-
     public function setTools(array $tools): self
     {
         $this->tools = $tools;

src/Providers/Log.php

@@ -44,16 +44,6 @@ public function chat(array|string $prompt): Message
         return new AssistantMessage("I'm the log Neuron AI driver");
     }
 
-    public function contextWindow(): int
-    {
-        return 1000000000;
-    }
-
-    public function maxCompletionTokens(): int
-    {
-        return 1000000000;
-    }
-
     public function setTools(array $tools): AIProviderInterface
     {
         return $this;

src/Providers/Mistral.php

@@ -27,7 +27,7 @@ class Mistral implements AIProviderInterface
     public function __construct(
         protected string $key,
         protected string $model,
-        protected int $max_tokens
+        protected int $max_tokens = 1024
     ) {
         $this->client = new Client([
             'base_uri' => 'https://api.mistral.ai/v1',
@@ -71,16 +71,6 @@ public function chat(array|string $prompt): Message
         return new AssistantMessage($result['choices'][0]['message']['content']);
     }
 
-    public function contextWindow(): int
-    {
-        return $this->max_tokens;
-    }
-
-    public function maxCompletionTokens(): int
-    {
-        return $this->max_tokens;
-    }
-
     public function setTools(array $tools): AIProviderInterface
     {
         throw new \LogicException('Not implemented');

src/Providers/OpenAI.php

@@ -27,8 +27,7 @@ class OpenAI implements AIProviderInterface
     public function __construct(
         protected string $key,
         protected string $model,
-        protected int $context_window,
-        protected int $max_tokens,
+        protected int $max_tokens = 1024,
     ) {
         $this->client = new Client([
             'base_uri' => 'https://api.openai.com/v1',
@@ -72,16 +71,6 @@ public function chat(array|string $prompt): Message
         return new AssistantMessage($result['choices'][0]['message']['content']);
     }
 
-    public function contextWindow(): int
-    {
-        return $this->context_window;
-    }
-
-    public function maxCompletionTokens(): int
-    {
-        return $this->max_tokens;
-    }
-
     public function setTools(array $tools): AIProviderInterface
     {
         throw new \LogicException('Not implemented.');