seeseemelk/ollama4j
1
0
Fork 0
forked from Mirror/ollama4j
Code Issues Pull Requests Actions Packages Releases 1 Activity

init

Browse Source
This commit is contained in:
Amith Koujalgi
2023-11-07 20:52:42 +05:30
parent c1615a2005
commit 153f516d9f
15 changed files with 434 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/test/java/io/github/amithkoujalgi/ollama4j/TestMockedAPIs.java
View File

@@ -1,5 +1,8 @@
package io.github.amithkoujalgi.ollama4j;
import io.github.amithkoujalgi.ollama4j.core.OllamaAPI;
import io.github.amithkoujalgi.ollama4j.core.exceptions.OllamaBaseException;
import io.github.amithkoujalgi.ollama4j.core.types.OllamaModelType;
import org.apache.hc.core5.http.ParseException;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
@@ -12,7 +15,7 @@ public class TestMockedAPIs {
@Test
public void testMockPullModel() {
OllamaAPI ollamaAPI = Mockito.mock(OllamaAPI.class);
OllamaModel model = OllamaModel.LLAMA2;
String model = OllamaModelType.LLAMA2;
try {
doNothing().when(ollamaAPI).pullModel(model);
ollamaAPI.pullModel(model);