/*** 版权所有 2021 涂聚文有限公司* 许可信息查看： 言語成了邀功盡責的功臣，還需要行爲每日來值班嗎* 描述：DB 数据库帮助，连接SQL Server,Oracle,PostgreSQL,MySQL,SQLite,neo4j* 数据库：MS SQL Server 2019* IDE: IntelliJ IDEA 2025.1.3* OS: Windows 10 x64* 历史版本： JDK 21* author: Geovin Du 涂聚文，geovindu*/import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.ObjectNode;import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.nio.charset.StandardCharsets;
import java.util.Scanner;public class Main {// Default Ollama API URLprivate static final String OLLAMA_API_URL = "http://localhost:11434/api/chat";// Model name - change this if your local model has a different tagprivate static final String MODEL_NAME = "deepseek-r1:1.5b";public static void main(String[] args) {System.out.println("Connecting to Ollama (" + MODEL_NAME + ")...");System.out.println("Type 'exit' or 'quit' to end the chat.");HttpClient client = HttpClient.newHttpClient();ObjectMapper mapper = new ObjectMapper();Scanner scanner = new Scanner(System.in);// Chat historyArrayNode messages = mapper.createArrayNode();while (true) {System.out.print("\nYou: ");String userInput = scanner.nextLine();if ("exit".equalsIgnoreCase(userInput) || "quit".equalsIgnoreCase(userInput)) {System.out.println("Goodbye!");break;}if (userInput.trim().isEmpty()) {continue;}// Add user message to historyObjectNode userMessage = mapper.createObjectNode();userMessage.put("role", "user");userMessage.put("content", userInput);messages.add(userMessage);// Prepare request bodyObjectNode requestBody = mapper.createObjectNode();requestBody.put("model", MODEL_NAME);requestBody.set("messages", messages);requestBody.put("stream", true); // Enable streamingtry {String jsonRequest = mapper.writeValueAsString(requestBody);HttpRequest request = HttpRequest.newBuilder().uri(URI.create(OLLAMA_API_URL)).header("Content-Type", "application/json").POST(HttpRequest.BodyPublishers.ofString(jsonRequest)).build();System.out.print("AI: ");// Send request and handle streaming responseclient.sendAsync(request, HttpResponse.BodyHandlers.ofInputStream()).thenAccept(response -> {if (response.statusCode() != 200) {System.err.println("\nError: " + response.statusCode());return;}try (InputStream is = response.body();BufferedReader reader = new BufferedReader(new InputStreamReader(is, StandardCharsets.UTF_8))) {StringBuilder fullResponseContent = new StringBuilder();String line;while ((line = reader.readLine()) != null) {JsonNode jsonResponse = mapper.readTree(line);if (jsonResponse.has("message")) {JsonNode messageNode = jsonResponse.get("message");if (messageNode.has("content")) {String content = messageNode.get("content").asText();System.out.print(content);fullResponseContent.append(content);}}if (jsonResponse.has("done") && jsonResponse.get("done").asBoolean()) {// Message complete, add to historyObjectNode assistantMessage = mapper.createObjectNode();assistantMessage.put("role", "assistant");assistantMessage.put("content", fullResponseContent.toString());messages.add(assistantMessage);}}} catch (IOException e) {e.printStackTrace();}}).join(); // Wait for completionSystem.out.println(); // New line after response} catch (Exception e) {System.err.println("Error communicating with Ollama: " + e.getMessage());e.printStackTrace();}}}
}
输出：
