chore: lib v0.24 compatibility (#101)

Author: chr-hertel

src/ai-bundle/src/DependencyInjection/AIExtension.php

@@ -49,6 +49,7 @@
 use Symfony\AI\Store\Bridge\ChromaDB\Store as ChromaDBStore;
 use Symfony\AI\Store\Bridge\MongoDB\Store as MongoDBStore;
 use Symfony\AI\Store\Bridge\Pinecone\Store as PineconeStore;
+use Symfony\AI\Store\Document\Vectorizer;
 use Symfony\AI\Store\Indexer;
 use Symfony\AI\Store\StoreInterface;
 use Symfony\AI\Store\VectorStoreInterface;
@@ -484,9 +485,14 @@ private function processIndexerConfig(int|string $name, array $config, Container
         $modelDefinition->addTag('symfony_ai.model.embeddings_model');
         $container->setDefinition('symfony_ai.indexer.'.$name.'.model', $modelDefinition);
 
-        $definition = new Definition(Indexer::class, [
-            '$model' => new Reference('symfony_ai.indexer.'.$name.'.model'),
+        $vectorizerDefinition = new Definition(Vectorizer::class, [
             '$platform' => new Reference($config['platform']),
+            '$model' => new Reference('symfony_ai.indexer.'.$name.'.model'),
+        ]);
+        $container->setDefinition('symfony_ai.indexer.'.$name.'.vectorizer', $vectorizerDefinition);
+
+        $definition = new Definition(Indexer::class, [
+            '$vectorizer' => new Reference('symfony_ai.indexer.'.$name.'.vectorizer'),
             '$store' => new Reference($config['store']),
         ]);
 

src/ai-bundle/src/Profiler/DataCollector.php

@@ -12,6 +12,7 @@
 namespace Symfony\AI\AIBundle\Profiler;
 
 use Symfony\AI\Agent\Toolbox\ToolboxInterface;
+use Symfony\AI\Platform\Model;
 use Symfony\AI\Platform\Tool\Tool;
 use Symfony\Bundle\FrameworkBundle\DataCollector\AbstractDataCollector;
 use Symfony\Component\DependencyInjection\Attribute\TaggedIterator;
@@ -55,7 +56,7 @@ public function collect(Request $request, Response $response, ?\Throwable $excep
     {
         $this->data = [
             'tools' => $this->defaultToolBox->getTools(),
-            'platform_calls' => array_merge(...array_map(fn (TraceablePlatform $platform) => $platform->calls, $this->platforms)),
+            'platform_calls' => array_merge(...array_map($this->awaitCallResults(...), $this->platforms)),
             'tool_calls' => array_merge(...array_map(fn (TraceableToolbox $toolbox) => $toolbox->calls, $this->toolboxes)),
         ];
     }
@@ -88,4 +89,23 @@ public function getToolCalls(): array
     {
         return $this->data['tool_calls'] ?? [];
     }
+
+    /**
+     * @return array{
+     *     model: Model,
+     *     input: array<mixed>|string|object,
+     *     options: array<string, mixed>,
+     *     response: string|iterable<mixed>|object|null
+     * }[]
+     */
+    private function awaitCallResults(TraceablePlatform $platform): array
+    {
+        $calls = $platform->calls;
+        foreach ($calls as $key => $call) {
+            $call['response'] = $call['response']->await()->getContent();
+            $calls[$key] = $call;
+        }
+
+        return $calls;
+    }
 }

src/ai-bundle/src/Profiler/TraceablePlatform.php

@@ -14,7 +14,7 @@
 use Symfony\AI\Platform\Message\Content\File;
 use Symfony\AI\Platform\Model;
 use Symfony\AI\Platform\PlatformInterface;
-use Symfony\AI\Platform\Response\ResponseInterface;
+use Symfony\AI\Platform\Response\ResponsePromise;
 
 /**
  * @author Christopher Hertel <mail@christopher-hertel.de>
@@ -23,7 +23,7 @@
  *     model: Model,
  *     input: array<mixed>|string|object,
  *     options: array<string, mixed>,
- *     response: ResponseInterface,
+ *     response: ResponsePromise,
  * }
  */
 final class TraceablePlatform implements PlatformInterface
@@ -38,7 +38,7 @@ public function __construct(
     ) {
     }
 
-    public function request(Model $model, array|string|object $input, array $options = []): ResponseInterface
+    public function request(Model $model, array|string|object $input, array $options = []): ResponsePromise
     {
         $response = $this->platform->request($model, $input, $options);
 
@@ -50,7 +50,7 @@ public function request(Model $model, array|string|object $input, array $options
             'model' => $model,
             'input' => \is_object($input) ? clone $input : $input,
             'options' => $options,
-            'response' => $response->getContent(),
+            'response' => $response,
         ];
 
         return $response;