mirror of
https://github.com/amithkoujalgi/ollama4j.git
synced 2025-11-02 17:40:41 +01:00
Added surefire and failsafe plugins
This commit is contained in:
@@ -101,7 +101,7 @@ public class TestRealAPIs {
|
||||
String model = OllamaModelType.LLAMA2;
|
||||
String prompt = "some prompt text";
|
||||
try {
|
||||
when(ollamaAPI.ask(model, prompt)).thenReturn(new OllamaResult("", 0));
|
||||
when(ollamaAPI.ask(model, prompt)).thenReturn(new OllamaResult("", 0,200));
|
||||
ollamaAPI.ask(model, prompt);
|
||||
verify(ollamaAPI, times(1)).ask(model, prompt);
|
||||
} catch (IOException | OllamaBaseException | InterruptedException e) {
|
||||
|
||||
Reference in New Issue
Block a user