File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -44,7 +44,7 @@ public function getModel(string $modelName): Ollama
4444 $ capabilities = array_map (
4545 static fn (string $ capability ): Capability => match ($ capability ) {
4646 'embedding ' => Capability::EMBEDDINGS ,
47- 'completion ' => Capability::INPUT_TEXT ,
47+ 'completion ' => Capability::INPUT_MESSAGES ,
4848 'tools ' => Capability::TOOL_CALLING ,
4949 'thinking ' => Capability::THINKING ,
5050 'vision ' => Capability::INPUT_IMAGE ,
Original file line number Diff line number Diff line change @@ -34,7 +34,7 @@ public function testModelCatalogCanReturnModelFromApi()
3434
3535 $ this ->assertSame ('foo ' , $ model ->getName ());
3636 $ this ->assertSame ([
37- Capability::INPUT_TEXT ,
37+ Capability::INPUT_MESSAGES ,
3838 ], $ model ->getCapabilities ());
3939 $ this ->assertSame (1 , $ httpClient ->getRequestsCount ());
4040 }
@@ -66,7 +66,7 @@ public function testModelCatalogCanReturnModelsFromApi()
6666 $ this ->assertSame (Ollama::class, $ model ['class ' ]);
6767 $ this ->assertCount (1 , $ model ['capabilities ' ]);
6868 $ this ->assertSame ([
69- Capability::INPUT_TEXT ,
69+ Capability::INPUT_MESSAGES ,
7070 ], $ model ['capabilities ' ]);
7171 $ this ->assertSame (2 , $ httpClient ->getRequestsCount ());
7272 }
You can’t perform that action at this time.
0 commit comments