[Platform] Use JSON Path to convert responses

examples/composer.json

@@ -22,6 +22,7 @@
         "symfony/event-dispatcher": "^6.4|^7.0",
         "symfony/filesystem": "^6.4|^7.0",
         "symfony/finder": "^6.4|^7.0",
+        "symfony/json-path": "7.3.*",
         "symfony/process": "^6.4|^7.0",
         "symfony/var-dumper": "^6.4|^7.0"
     },

src/platform/composer.json

@@ -28,6 +28,7 @@
         "psr/log": "^3.0",
         "symfony/clock": "^6.4 || ^7.1",
         "symfony/http-client": "^6.4 || ^7.1",
+        "symfony/json-path": "7.3.*",
         "symfony/property-access": "^6.4 || ^7.1",
         "symfony/property-info": "^6.4 || ^7.1",
         "symfony/serializer": "^6.4 || ^7.1",

src/platform/src/Bridge/Albert/PlatformFactory.php

@@ -11,7 +11,6 @@
 
 namespace Symfony\AI\Platform\Bridge\Albert;
 
-use Symfony\AI\Platform\Bridge\OpenAI\Embeddings;
 use Symfony\AI\Platform\Bridge\OpenAI\GPT;
 use Symfony\AI\Platform\Contract;
 use Symfony\AI\Platform\Exception\InvalidArgumentException;
@@ -40,7 +39,7 @@ public static function create(
                 new GPTModelClient($httpClient, $apiKey, $baseUrl),
                 new EmbeddingsModelClient($httpClient, $apiKey, $baseUrl),
             ],
-            [new GPT\ResultConverter(), new Embeddings\ResultConverter()],
+            [new GPT\ResultConverter(), Contract\ResultConverter::create()],
             Contract::create(),
         );
     }

src/platform/src/Bridge/Azure/OpenAI/PlatformFactory.php

@@ -11,7 +11,6 @@
 
 namespace Symfony\AI\Platform\Bridge\Azure\OpenAI;
 
-use Symfony\AI\Platform\Bridge\OpenAI\Embeddings;
 use Symfony\AI\Platform\Bridge\OpenAI\GPT;
 use Symfony\AI\Platform\Bridge\OpenAI\Whisper;
 use Symfony\AI\Platform\Bridge\OpenAI\Whisper\AudioNormalizer;
@@ -41,7 +40,7 @@ public static function create(
 
         return new Platform(
             [$GPTModelClient, $embeddingsModelClient, $whisperModelClient],
-            [new GPT\ResultConverter(), new Embeddings\ResultConverter(), new Whisper\ResultConverter()],
+            [new GPT\ResultConverter(), new Whisper\ResultConverter(), Contract\ResultConverter::create()],
             $contract ?? Contract::create(new AudioNormalizer()),
         );
     }

src/platform/src/Bridge/Gemini/PlatformFactory.php

@@ -13,10 +13,10 @@
 
 use Symfony\AI\Platform\Bridge\Gemini\Contract\GeminiContract;
 use Symfony\AI\Platform\Bridge\Gemini\Embeddings\ModelClient as EmbeddingsModelClient;
-use Symfony\AI\Platform\Bridge\Gemini\Embeddings\ResultConverter as EmbeddingsResultConverter;
 use Symfony\AI\Platform\Bridge\Gemini\Gemini\ModelClient as GeminiModelClient;
-use Symfony\AI\Platform\Bridge\Gemini\Gemini\ResultConverter as GeminiResultConverter;
 use Symfony\AI\Platform\Contract;
+use Symfony\AI\Platform\Contract\JsonPathConverter\VectorResultExtractor;
+use Symfony\AI\Platform\Contract\ResultConverter;
 use Symfony\AI\Platform\Platform;
 use Symfony\Component\HttpClient\EventSourceHttpClient;
 use Symfony\Contracts\HttpClient\HttpClientInterface;
@@ -36,7 +36,9 @@ public static function create(
 
         return new Platform(
             [new EmbeddingsModelClient($httpClient, $apiKey), new GeminiModelClient($httpClient, $apiKey)],
-            [new EmbeddingsResultConverter(), new GeminiResultConverter()],
+            [new Gemini\ResultConverter(), ResultConverter::create([
+                new VectorResultExtractor('$.embeddings[*].values'),
+            ])],
             $contract ?? GeminiContract::create(),
         );
     }

src/platform/src/Bridge/Mistral/PlatformFactory.php

@@ -32,7 +32,7 @@ public static function create(
 
         return new Platform(
             [new Embeddings\ModelClient($httpClient, $apiKey), new Llm\ModelClient($httpClient, $apiKey)],
-            [new Embeddings\ResultConverter(), new Llm\ResultConverter()],
+            [new Llm\ResultConverter(), Contract\ResultConverter::create()],
             $contract ?? Contract::create(new ToolNormalizer()),
         );
     }

src/platform/src/Bridge/OpenAI/PlatformFactory.php

@@ -41,9 +41,9 @@ public static function create(
             ],
             [
                 new GPT\ResultConverter(),
-                new Embeddings\ResultConverter(),
                 new DallE\ResultConverter(),
                 new WhisperResponseConverter(),
+                Contract\ResultConverter::create(),
             ],
             $contract ?? Contract::create(new AudioNormalizer()),
         );

src/platform/src/Bridge/Voyage/PlatformFactory.php

@@ -29,6 +29,6 @@ public static function create(
     ): Platform {
         $httpClient = $httpClient instanceof EventSourceHttpClient ? $httpClient : new EventSourceHttpClient($httpClient);
 
-        return new Platform([new ModelClient($httpClient, $apiKey)], [new ResultConverter()], $contract);
+        return new Platform([new ModelClient($httpClient, $apiKey)], [Contract\ResultConverter::create()], $contract);
     }
 }

src/platform/src/Contract/JsonPathConverter/ResultExtractorInterface.php

@@ -0,0 +1,25 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\AI\Platform\Contract\JsonPathConverter;
+
+use Symfony\AI\Platform\Result\ResultInterface;
+use Symfony\Component\JsonPath\JsonCrawler;
+
+interface ResultExtractorInterface
+{
+    public function supports(JsonCrawler $crawler): bool;
+
+    /**
+     * @return ResultInterface[]
+     */
+    public function extract(JsonCrawler $crawler): array;
+}

src/platform/src/Contract/JsonPathConverter/TextResultExtractor.php

@@ -0,0 +1,36 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\AI\Platform\Contract\JsonPathConverter;
+
+use Symfony\AI\Platform\Result\TextResult;
+use Symfony\Component\JsonPath\JsonCrawler;
+use Symfony\Component\JsonPath\JsonPath;
+
+readonly class TextResultExtractor implements ResultExtractorInterface
+{
+    public function __construct(
+        private string|JsonPath $jsonPath = '$.choices[?length(@.message.content) >= 0].message.content',
+    ) {
+    }
+
+    public function supports(JsonCrawler $crawler): bool
+    {
+        return [] !== array_filter($crawler->find($this->jsonPath));
+    }
+
+    public function extract(JsonCrawler $crawler): array
+    {
+        $data = $crawler->find($this->jsonPath);
+
+        return array_map(static fn (string $text): TextResult => new TextResult($text), $data);
+    }
+}