SplitfireUptown commented on code in PR #8551:
URL: https://github.com/apache/seatunnel/pull/8551#discussion_r1957275977
##########
seatunnel-transforms-v2/src/test/java/org/apache/seatunnel/transform/llm/LLMRequestJsonTest.java:
##########
@@ -209,4 +215,80 @@ void testCustomRequestJson() throws IOException {
"{\"messages\":[{\"role\":\"system\",\"content\":\"Determine
whether someone is Chinese or American by their
name\"},{\"role\":\"user\",\"content\":\"{\\\"id\\\":1,
\\\"name\\\":\\\"John\\\"}\"}],\"model\":\"custom-model\"}",
OBJECT_MAPPER.writeValueAsString(node));
}
+
+ @Mock
+ private HttpClient httpClient;
+
+ @Test
+ void testCustomOllamaRequestJson() throws IOException {
+
+ MockWebServer mockWebServer = new MockWebServer();
+ mockWebServer.start(11434);
+ // 设置模拟的 JSON 响应
+ String jsonResponse = "{\n" +
+ " \"model\": \"qwen:7b\",\n" +
+ " \"created_at\": \"2025-02-07T01:22:46.589856Z\",\n" +
+ " \"message\": {\n" +
+ " \"role\": \"assistant\",\n" +
+ " \"content\": \"Based on the information provided in
the JSON object, \\\"John\\\" does not inherently indicate if the person is
Chinese or American. The name \\\"John\\\" is commonly used across many
cultures. To determine a person's nationality based solely on their name, more
context would be needed.\"\n" +
+ " },\n" +
+ " \"done_reason\": \"stop\",\n" +
+ " \"done\": true,\n" +
+ " \"total_duration\": 14435322300,\n" +
+ " \"load_duration\": 28998200,\n" +
+ " \"prompt_eval_count\": 34,\n" +
+ " \"prompt_eval_duration\": 302000000,\n" +
+ " \"eval_count\": 56,\n" +
+ " \"eval_duration\": 14102000000\n" +
+ "}";
+
+ // 向 MockWebServer 添加响应
Review Comment:
code is ok please check it
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: commits-unsubscr...@seatunnel.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
