初始化

easy-agents-chat/easy-agents-chat-ollama/src/test/java/com/easyagents/llm/ollama/OllamaChatModelTest.java

@@ -0,0 +1,94 @@
+package com.easyagents.llm.ollama;
+
+import com.easyagents.core.message.AiMessage;
+import com.easyagents.core.model.chat.ChatModel;
+import com.easyagents.core.model.chat.response.AiMessageResponse;
+import com.easyagents.core.model.exception.ModelException;
+import com.easyagents.core.prompt.SimplePrompt;
+import org.junit.Test;
+
+public class OllamaChatModelTest {
+
+    @Test(expected = ModelException.class)
+    public void testChat() {
+        OllamaChatConfig config = new OllamaChatConfig();
+        config.setEndpoint("http://localhost:11434");
+        config.setModel("llama3");
+        config.setLogEnabled(true);
+
+        ChatModel chatModel = new OllamaChatModel(config);
+        String chat = chatModel.chat("Why is the sky blue?");
+        System.out.println(">>>" + chat);
+    }
+
+
+    @Test
+    public void testChatStream() throws InterruptedException {
+        OllamaChatConfig config = new OllamaChatConfig();
+        config.setEndpoint("http://localhost:11434");
+        config.setModel("llama3");
+        config.setLogEnabled(true);
+
+        ChatModel chatModel = new OllamaChatModel(config);
+        chatModel.chatStream("Why is the sky blue?", (context, response) -> System.out.println(response.getMessage().getContent()));
+
+        Thread.sleep(2000);
+    }
+
+
+    @Test
+    public void testFunctionCall1() throws InterruptedException {
+        OllamaChatConfig config = new OllamaChatConfig();
+        config.setEndpoint("http://localhost:11434");
+        config.setModel("llama3.1");
+        config.setLogEnabled(true);
+
+        ChatModel chatModel = new OllamaChatModel(config);
+
+        SimplePrompt prompt = new SimplePrompt("What's the weather like in Beijing?");
+        prompt.addToolsFromClass(WeatherFunctions.class);
+        AiMessageResponse response = chatModel.chat(prompt);
+
+        System.out.println(response.executeToolCallsAndGetResults());
+    }
+
+
+    @Test
+    public void testFunctionCall2() throws InterruptedException {
+        OllamaChatConfig config = new OllamaChatConfig();
+        config.setEndpoint("http://localhost:11434");
+        config.setModel("llama3.1");
+        config.setLogEnabled(true);
+
+        ChatModel chatModel = new OllamaChatModel(config);
+
+        SimplePrompt prompt = new SimplePrompt("What's the weather like in Beijing?");
+        prompt.addToolsFromClass(WeatherFunctions.class);
+        AiMessageResponse response = chatModel.chat(prompt);
+
+        if (response.hasToolCalls()) {
+            prompt.setToolMessages(response.executeToolCallsAndGetToolMessages());
+            AiMessageResponse response1 = chatModel.chat(prompt);
+            System.out.println(response1.getMessage().getContent());
+        }
+    }
+
+
+    @Test
+    public void testVisionModel() {
+        OllamaChatConfig config = new OllamaChatConfig();
+        config.setEndpoint("http://localhost:11434");
+        config.setModel("llava");
+        config.setLogEnabled(true);
+
+        ChatModel chatModel = new OllamaChatModel(config);
+
+        SimplePrompt imagePrompt = new SimplePrompt("What's in the picture?");
+        imagePrompt.addImageUrl("https://agentsflex.com/assets/images/logo.png");
+
+        AiMessageResponse response = chatModel.chat(imagePrompt);
+        AiMessage message = response == null ? null : response.getMessage();
+        System.out.println(message);
+    }
+
+}

easy-agents-chat/easy-agents-chat-ollama/src/test/java/com/easyagents/llm/ollama/WeatherFunctions.java

@@ -0,0 +1,22 @@
+package com.easyagents.llm.ollama;
+
+import com.easyagents.core.model.chat.tool.annotation.ToolDef;
+import com.easyagents.core.model.chat.tool.annotation.ToolParam;
+
+public class WeatherFunctions {
+
+    @ToolDef(name = "get_the_weather_info", description = "get the weather info")
+    public static String getWeatherInfo(
+        @ToolParam(name = "city", description = "the city name") String name
+    ) {
+        return "Snowy days";
+    }
+
+
+    @ToolDef(name = "get_the_temperature", description = "get the temperature")
+    public static String getTemperature(
+        @ToolParam(name = "city", description = "the city name") String name
+    ) {
+        return "The temperature in " + name + " is 15°C";
+    }
+}