test/agents-flex
1
0
Fork 0
mirror of https://gitee.com/agents-flex/agents-flex.git synced 2024-11-30 02:47:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

test: add chat with image test

Browse Source
This commit is contained in:
Michael Yang 2024-06-14 18:29:46 +08:00
parent de60f5e6ac
commit 91d39001ed
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
agents-flex-llm/agents-flex-llm-openai/src/test/java/com/agentsflex/llm/openai/OpenAiLlmTest.java
View File

@ -7,6 +7,7 @@ import com.agentsflex.llm.response.AiMessageResponse;
import com.agentsflex.llm.response.FunctionMessageResponse;
import com.agentsflex.message.AiMessage;
import com.agentsflex.prompt.FunctionPrompt;
import com.agentsflex.prompt.ImagePrompt;
import org.junit.Test;
public class OpenAiLlmTest {
@ -48,6 +49,33 @@ public class OpenAiLlmTest {
@Test
public void testChatWithImage() {
OpenAiLlmConfig config = new OpenAiLlmConfig();
config.setApiKey("sk-5gqOclb****0");
config.setModel("gpt-4-turbo");
Llm llm = new OpenAiLlm(config);
ImagePrompt prompt = new ImagePrompt("What's in this image?");
prompt.setImageUrl("https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg");
// llm.chatStream(prompt, (StreamResponseListener<AiMessageResponse, AiMessage>)
// (context, response) -> System.out.println(response.getMessage().getContent())
// );
AiMessageResponse response = llm.chat(prompt);
System.out.println(response);
try {
Thread.sleep(12000);
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
}
@Test
public void testFunctionCalling() throws InterruptedException {
OpenAiLlmConfig config = new OpenAiLlmConfig();