test/agents-flex
1
0
Fork 0
mirror of https://gitee.com/agents-flex/agents-flex.git synced 2024-11-29 18:38:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

test: add test for ollama vision Models

Browse Source
This commit is contained in:
Michael Yang 2024-09-15 10:19:09 +08:00
parent 8641a63537
commit 452af07ec7
1 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
agents-flex-llm/agents-flex-llm-ollama/src/test/java/com/agentsflex/llm/ollama/OllamaLlmTest.java
View File

@ -2,8 +2,11 @@ package com.agentsflex.llm.ollama;
import com.agentsflex.core.document.Document;
import com.agentsflex.core.llm.Llm;
import com.agentsflex.core.llm.response.AiMessageResponse;
import com.agentsflex.core.llm.response.FunctionMessageResponse;
import com.agentsflex.core.message.AiMessage;
import com.agentsflex.core.prompt.FunctionPrompt;
import com.agentsflex.core.prompt.ImagePrompt;
import com.agentsflex.core.store.VectorData;
import org.junit.Test;
@ -49,7 +52,6 @@ public class OllamaLlmTest {
}
@Test
public void testFunctionCall() throws InterruptedException {
OllamaLlmConfig config = new OllamaLlmConfig();
@ -67,4 +69,23 @@ public class OllamaLlmTest {
System.out.println(result);
}
@Test
public void testVisionModel() {
OllamaLlmConfig config = new OllamaLlmConfig();
config.setEndpoint("http://localhost:11434");
config.setModel("llava");
config.setDebug(true);
Llm llm = new OllamaLlm(config);
ImagePrompt imagePrompt = new ImagePrompt("What's in the picture?",
"https://agentsflex.com/assets/images/logo.png");
AiMessageResponse response = llm.chat(imagePrompt);
AiMessage message = response.getMessage();
System.out.println(message);
}
}