增加超时时间和读取超时时间设置

This commit is contained in:
choweli
2025-06-04 10:58:13 +08:00
parent 3be0111884
commit 82399a185e
9 changed files with 102 additions and 19 deletions

View File

@@ -51,7 +51,7 @@ class DeepSeekServiceTest {
deepSeekService.chat(prompt, data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {
@@ -92,7 +92,7 @@ class DeepSeekServiceTest {
deepSeekService.beta(beta, data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {

View File

@@ -55,7 +55,7 @@ class DoubaoServiceTest {
doubaoService.chat(prompt, data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {
@@ -111,7 +111,7 @@ class DoubaoServiceTest {
AtomicBoolean isDone = new AtomicBoolean(false);
doubaoService.chatVision(prompt,images, data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {
@@ -188,7 +188,7 @@ class DoubaoServiceTest {
AtomicBoolean isDone = new AtomicBoolean(false);
doubaoService.botsChat(messages, data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {
@@ -286,7 +286,7 @@ class DoubaoServiceTest {
AtomicBoolean isDone = new AtomicBoolean(false);
doubaoService.chatContext(messages,contextId, data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {

View File

@@ -56,7 +56,7 @@ class GrokServiceTest {
grokService.chat(prompt, data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {
@@ -96,7 +96,7 @@ class GrokServiceTest {
AtomicBoolean isDone = new AtomicBoolean(false);
grokService.message(prompt, 4096, data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {
@@ -130,7 +130,7 @@ class GrokServiceTest {
AtomicBoolean isDone = new AtomicBoolean(false);
grokService.chatVision(prompt,images, data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {

View File

@@ -59,7 +59,7 @@ class HutoolServiceTest {
hutoolService.chat(prompt, data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {
@@ -102,7 +102,7 @@ class HutoolServiceTest {
AtomicBoolean isDone = new AtomicBoolean(false);
hutoolService.chatVision(prompt,images, data -> {
assertNotNull(data);
if (data.equals("data:[DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {

View File

@@ -62,7 +62,7 @@ class OpenaiServiceTest {
openaiService.chat(prompt, data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {
@@ -107,7 +107,7 @@ class OpenaiServiceTest {
AtomicBoolean isDone = new AtomicBoolean(false);
openaiService.chatVision(prompt,images, data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {
@@ -229,7 +229,7 @@ class OpenaiServiceTest {
AtomicBoolean isDone = new AtomicBoolean(false);
openaiService.chatReasoning(messages,OpenaiCommon.OpenaiReasoning.HIGH.getEffort(), data -> {
assertNotNull(data);
if (data.equals("data: [DONE]")) {
if (data.contains("[DONE]")) {
// 设置结束标志
isDone.set(true);
} else if (data.contains("\"error\"")) {