feature #252 [Platform] Enhance in memory platform to support all ResultInterface types (RamyHakam)

This PR was squashed before being merged into the main branch.

Discussion
----------

[Platform] Enhance in memory platform to support all `ResultInterface` types

| Q             | A
| ------------- | ---
| Bug fix?      | no
| New feature?  | yes <!-- please update src/**/CHANGELOG.md files -->
| Docs?         | yes <!-- required for new features -->
| Issues        | none <!-- prefix each issue number with "Fix #", no need to create an issue if none exist, explain below instead -->
| License       | MIT

Enhances the existing `InMemoryPlatform` to support all `ResultInterface` types, making it easier to test all result types.
Previously limited to text responses, it now supports vectors, images, binary data, choices, and any custom result types.

Commits
-------

82e873f [Platform] Enhance in memory platform to support all `ResultInterface` types

Author: chr-hertel

src/platform/doc/index.rst

@@ -289,7 +289,7 @@ For unit or integration testing, you can use the `InMemoryPlatform`, which imple
 It supports returning either:
 
 - A fixed string result
-- A callable that dynamically returns a response based on the model, input, and options::
+- A callable that dynamically returns a simple string or any ``ResultInterface`` based on the model, input, and options::
 
     use Symfony\AI\Platform\InMemoryPlatform;
     use Symfony\AI\Platform\Model;
@@ -300,8 +300,41 @@ It supports returning either:
 
     echo $result->asText(); // "Fake result"
 
+**Dynamic Text Results**::
 
-Internally, it uses `InMemoryRawResult` to simulate the behavior of real API responses and support `ResultPromise`.
+    $platform = new InMemoryPlatform(
+        fn($model, $input, $options) => "Echo: {$input}"
+    );
+
+    $result = $platform->invoke(new Model('test'), 'Hello AI');
+    echo $result->asText(); // "Echo: Hello AI"
+
+**Vector Results**::
+
+    use Symfony\AI\Platform\Result\VectorResult;
+
+    $platform = new InMemoryPlatform(
+        fn() => new VectorResult(new Vector([0.1, 0.2, 0.3, 0.4]))
+    );
+
+    $result = $platform->invoke(new Model('test'), 'vectorize this text');
+    $vectors = $result->asVectors(); // Returns Vector object with [0.1, 0.2, 0.3, 0.4]
+
+**Binary Results**::
+
+    use Symfony\AI\Platform\Result\BinaryResult;
+
+    $platform = new InMemoryPlatform(
+        fn() => new BinaryResult('fake-pdf-content', 'application/pdf')
+    );
+
+    $result = $platform->invoke(new Model('test'), 'generate PDF document');
+    $binary = $result->asBinary(); // Returns Binary object with content and MIME type
+
+
+**Raw Results**
+
+The platform automatically uses the ``getRawResult()`` from any ``ResultInterface`` returned by closures. For string results, it creates an ``InMemoryRawResult`` to simulate real API response metadata.
 
 This allows fast and isolated testing of AI-powered features without relying on live providers or HTTP requests.
 

src/platform/src/InMemoryPlatform.php

@@ -12,6 +12,7 @@
 namespace Symfony\AI\Platform;
 
 use Symfony\AI\Platform\Result\InMemoryRawResult;
+use Symfony\AI\Platform\Result\ResultInterface;
 use Symfony\AI\Platform\Result\ResultPromise;
 use Symfony\AI\Platform\Result\TextResult;
 
@@ -26,27 +27,36 @@ class InMemoryPlatform implements PlatformInterface
 {
     /**
      * The mock result can be a string or a callable that returns a string.
-     * If it's a closure, it receives the model, input, and  optionally options as parameters like a real platform call.
+     * If it's a closure, it receives the model, input, and optionally options as parameters like a real platform call.
      */
     public function __construct(private readonly \Closure|string $mockResult)
     {
     }
 
     public function invoke(Model $model, array|string|object $input, array $options = []): ResultPromise
     {
-        $resultText = $this->mockResult instanceof \Closure
-            ? ($this->mockResult)($model, $input, $options)
-            : $this->mockResult;
-
-        $textResult = new TextResult($resultText);
-
-        return new ResultPromise(
-            static fn () => $textResult,
-            rawResult: new InMemoryRawResult(
-                ['text' => $resultText],
-                (object) ['text' => $resultText],
-            ),
-            options: $options
+        $result = \is_string($this->mockResult) ? $this->mockResult : ($this->mockResult)($model, $input, $options);
+
+        if ($result instanceof ResultInterface) {
+            return $this->createPromise($result, $options);
+        }
+
+        return $this->createPromise(new TextResult($result), $options);
+    }
+
+    /**
+     * Creates a ResultPromise from a ResultInterface.
+     *
+     * @param ResultInterface      $result  The result to wrap in a promise
+     * @param array<string, mixed> $options Additional options for the promise
+     */
+    private function createPromise(ResultInterface $result, array $options): ResultPromise
+    {
+        $rawResult = $result->getRawResult() ?? new InMemoryRawResult(
+            ['text' => $result->getContent()],
+            (object) ['text' => $result->getContent()],
         );
+
+        return new ResultPromise(static fn () => $result, $rawResult, $options);
     }
 }

src/platform/tests/InMemoryPlatformTest.php

@@ -13,6 +13,8 @@
 use PHPUnit\Framework\TestCase;
 use Symfony\AI\Platform\InMemoryPlatform;
 use Symfony\AI\Platform\Model;
+use Symfony\AI\Platform\Result\VectorResult;
+use Symfony\AI\Platform\Vector\Vector;
 
 #[CoversClass(InMemoryPlatform::class)]
 class InMemoryPlatformTest extends TestCase
@@ -37,4 +39,15 @@ public function testPlatformInvokeWithCallableResult()
 
         $this->assertSame('DYNAMIC TEXT', $result->asText());
     }
+
+    public function testPlatformInvokeWithVectorResultResponse()
+    {
+        $platform = new InMemoryPlatform(
+            fn () => new VectorResult(new Vector([0.1, 0.1, 0.5]))
+        );
+
+        $result = $platform->invoke(new Model('test'), 'dynamic text');
+
+        $this->assertEquals([0.1, 0.1, 0.5], $result->asVectors()[0]->getData());
+    }
 }