16
16
use PHPUnit \Framework \TestCase ;
17
17
use Symfony \AI \Platform \Bridge \Ollama \Ollama ;
18
18
use Symfony \AI \Platform \Bridge \Ollama \OllamaClient ;
19
+ use Symfony \AI \Platform \Bridge \Ollama \OllamaResultConverter ;
19
20
use Symfony \AI \Platform \Bridge \Ollama \PlatformFactory ;
20
21
use Symfony \AI \Platform \Model ;
22
+ use Symfony \AI \Platform \Result \RawHttpResult ;
21
23
use Symfony \AI \Platform \Result \StreamResult ;
22
24
use Symfony \Component \HttpClient \MockHttpClient ;
23
25
use Symfony \Component \HttpClient \Response \JsonMockResponse ;
@@ -153,8 +155,8 @@ public function testStreamingConverterWithDirectResponse()
153
155
]);
154
156
155
157
$ mockResponse = $ mockHttpClient ->request ('GET ' , 'http://test.example ' );
156
- $ rawResult = new \ Symfony \ AI \ Platform \ Result \ RawHttpResult ($ mockResponse );
157
- $ converter = new \ Symfony \ AI \ Platform \ Bridge \ Ollama \ OllamaResultConverter ();
158
+ $ rawResult = new RawHttpResult ($ mockResponse );
159
+ $ converter = new OllamaResultConverter ();
158
160
159
161
$ result = $ converter ->convert ($ rawResult , ['stream ' => true ]);
160
162
@@ -170,7 +172,7 @@ public function testStreamingConverterWithDirectResponse()
170
172
]);
171
173
172
174
$ regularMockResponse = $ regularMockHttpClient ->request ('GET ' , 'http://test.example ' );
173
- $ regularRawResult = new \ Symfony \ AI \ Platform \ Result \ RawHttpResult ($ regularMockResponse );
175
+ $ regularRawResult = new RawHttpResult ($ regularMockResponse );
174
176
$ regularResult = $ converter ->convert ($ regularRawResult , ['stream ' => false ]);
175
177
176
178
$ this ->assertNotInstanceOf (StreamResult::class, $ regularResult );
0 commit comments