feat: adding llama support via replicate and ollama

.env

@@ -9,6 +9,12 @@ ANTHROPIC_API_KEY=
 # For using Voyage
 VOYAGE_API_KEY=
 
+# For using Replicate
+REPLICATE_API_KEY=
+
+# For using Ollama
+OLLAMA_HOST_URL=
+
 # For using GPT on Azure
 AZURE_OPENAI_BASEURL=
 AZURE_OPENAI_DEPLOYMENT=

examples/chat-llama-ollama.php

@@ -0,0 +1,29 @@
+<?php
+
+use PhpLlm\LlmChain\Chain;
+use PhpLlm\LlmChain\Message\Message;
+use PhpLlm\LlmChain\Message\MessageBag;
+use PhpLlm\LlmChain\Model\Language\Llama;
+use PhpLlm\LlmChain\Platform\Ollama;
+use Symfony\Component\Dotenv\Dotenv;
+use Symfony\Component\HttpClient\HttpClient;
+
+require_once dirname(__DIR__).'/vendor/autoload.php';
+(new Dotenv())->loadEnv(dirname(__DIR__).'/.env');
+
+if (empty($_ENV['OLLAMA_HOST_URL'])) {
+    echo 'Please set the OLLAMA_HOST_URL environment variable.'.PHP_EOL;
+    exit(1);
+}
+
+$platform = new Ollama(HttpClient::create(), $_ENV['OLLAMA_HOST_URL']);
+$llm = new Llama($platform);
+
+$chain = new Chain($llm);
+$messages = new MessageBag(
+    Message::forSystem('You are a helpful assistant.'),
+    Message::ofUser('Tina has one brother and one sister. How many sisters do Tina\'s siblings have?'),
+);
+$response = $chain->call($messages);
+
+echo $response.PHP_EOL;

examples/chat-llama-replicate.php

@@ -0,0 +1,29 @@
+<?php
+
+use PhpLlm\LlmChain\Chain;
+use PhpLlm\LlmChain\Message\Message;
+use PhpLlm\LlmChain\Message\MessageBag;
+use PhpLlm\LlmChain\Model\Language\Llama;
+use PhpLlm\LlmChain\Platform\Replicate;
+use Symfony\Component\Dotenv\Dotenv;
+use Symfony\Component\HttpClient\HttpClient;
+
+require_once dirname(__DIR__).'/vendor/autoload.php';
+(new Dotenv())->loadEnv(dirname(__DIR__).'/.env');
+
+if (empty($_ENV['REPLICATE_API_KEY'])) {
+    echo 'Please set the REPLICATE_API_KEY environment variable.'.PHP_EOL;
+    exit(1);
+}
+
+$platform = new Replicate(HttpClient::create(), $_ENV['REPLICATE_API_KEY']);
+$llm = new Llama($platform);
+
+$chain = new Chain($llm);
+$messages = new MessageBag(
+    Message::forSystem('You are a helpful assistant.'),
+    Message::ofUser('Tina has one brother and one sister. How many sisters do Tina\'s siblings have?'),
+);
+$response = $chain->call($messages);
+
+echo $response.PHP_EOL;

src/Model/Language/Llama.php

@@ -0,0 +1,48 @@
+<?php
+
+declare(strict_types=1);
+
+namespace PhpLlm\LlmChain\Model\Language;
+
+use PhpLlm\LlmChain\LanguageModel;
+use PhpLlm\LlmChain\Message\MessageBag;
+use PhpLlm\LlmChain\Platform\Ollama;
+use PhpLlm\LlmChain\Platform\Replicate;
+use PhpLlm\LlmChain\Response\Choice;
+use PhpLlm\LlmChain\Response\Response;
+
+final readonly class Llama implements LanguageModel
+{
+    public function __construct(
+        private Replicate|Ollama $platform,
+    ) {
+    }
+
+    public function call(MessageBag $messages, array $options = []): Response
+    {
+        $systemMessages = $messages->getSystemMessage();
+        $endpoint = $this->platform instanceof Replicate ? 'predictions' : 'chat';
+
+        $response = $this->platform->request('meta/meta-llama-3.1-405b-instruct', 'predictions', [
+            'system' => $systemMessages->content,
+            'prompt' => $messages->withoutSystemMessage()->getIterator()->current()->content[0]->text,
+        ]);
+
+        return new Response(new Choice(implode('', $response['output'])));
+    }
+
+    public function supportsToolCalling(): bool
+    {
+        // TODO: Implement supportsToolCalling() method.
+    }
+
+    public function supportsImageInput(): bool
+    {
+        // TODO: Implement supportsImageInput() method.
+    }
+
+    public function supportsStructuredOutput(): bool
+    {
+        // TODO: Implement supportsStructuredOutput() method.
+    }
+}

src/Platform/Ollama.php

@@ -0,0 +1,36 @@
+<?php
+
+declare(strict_types=1);
+
+namespace PhpLlm\LlmChain\Platform;
+
+use Symfony\Contracts\HttpClient\HttpClientInterface;
+
+final readonly class Ollama
+{
+    public function __construct(
+        private HttpClientInterface $httpClient,
+        private string $hostUrl,
+    ) {
+    }
+
+    /**
+     * @param string               $model The model name on Replicate, e.g. "meta/meta-llama-3.1-405b-instruct"
+     * @param array<string, mixed> $body
+     *
+     * @return array<string, mixed>
+     */
+    public function request(string $model, string $endpoint, array $body): array
+    {
+        $url = sprintf('%s/api/%s', $this->hostUrl, $endpoint);
+
+        $response = $this->httpClient->request('POST', $url, [
+            'headers' => ['Content-Type' => 'application/json'],
+            'json' => array_merge($body, [
+                'model' => $model,
+            ]),
+        ]);
+
+        return dump($response->toArray());
+    }
+}

src/Platform/Replicate.php

@@ -0,0 +1,56 @@
+<?php
+
+declare(strict_types=1);
+
+namespace PhpLlm\LlmChain\Platform;
+
+use Symfony\Contracts\HttpClient\HttpClientInterface;
+
+final readonly class Replicate
+{
+    public function __construct(
+        private HttpClientInterface $httpClient,
+        private string $apiKey,
+    ) {
+    }
+
+    /**
+     * @param string               $model The model name on Replicate, e.g. "meta/meta-llama-3.1-405b-instruct"
+     * @param array<string, mixed> $body
+     *
+     * @return array<string, mixed>
+     */
+    public function request(string $model, string $endpoint, array $body): array
+    {
+        $url = sprintf('https://api.replicate.com/v1/models/%s/%s', $model, $endpoint);
+
+        $response = $this->httpClient->request('POST', $url, [
+            'headers' => ['Content-Type' => 'application/json'],
+            'auth_bearer' => $this->apiKey,
+            'json' => ['input' => $body],
+        ])->toArray();
+
+        while (!in_array($response['status'], ['succeeded', 'failed', 'canceled'], true)) {
+            sleep(1);
+
+            $response = $this->getResponse($response['id']);
+        }
+
+        return $response;
+    }
+
+    /**
+     * @return array<string, mixed>
+     */
+    private function getResponse(string $id): array
+    {
+        $url = sprintf('https://api.replicate.com/v1/predictions/%s', $id);
+
+        $response = $this->httpClient->request('GET', $url, [
+            'headers' => ['Content-Type' => 'application/json'],
+            'auth_bearer' => $this->apiKey,
+        ]);
+
+        return $response->toArray();
+    }
+}