wei-py
3 weeks ago
7 changed files with 560 additions and 62 deletions
-
50src/main/java/com/wok/supportbot/app/AssistantApp.java
-
10src/main/java/com/wok/supportbot/app/ChatResult.java
-
59src/main/java/com/wok/supportbot/app/SourceReference.java
-
73src/main/java/com/wok/supportbot/controller/AiController.java
-
14src/main/java/com/wok/supportbot/controller/OpenApiController.java
-
250src/test/java/com/wok/supportbot/AnswerTransportTests.java
-
166src/test/java/com/wok/supportbot/ChatResultEndpointTests.java
@ -0,0 +1,59 @@ |
|||
package com.wok.supportbot.app; |
|||
|
|||
import com.fasterxml.jackson.annotation.JsonInclude; |
|||
import org.springframework.ai.document.Document; |
|||
|
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
/** Public citation metadata from the documents actually used by this answer. */ |
|||
@JsonInclude(JsonInclude.Include.ALWAYS) |
|||
public record SourceReference(String documentId, String title, String sourceName, |
|||
Integer chunkIndex, Double score, String snippet) { |
|||
|
|||
public SourceReference { |
|||
if (snippet != null && snippet.length() > 160) { |
|||
int end = Character.isHighSurrogate(snippet.charAt(158)) ? 158 : 159; |
|||
snippet = snippet.substring(0, end) + "…"; |
|||
} |
|||
} |
|||
|
|||
public static List<SourceReference> fromDocuments(List<Document> documents) { |
|||
if (documents == null || documents.isEmpty()) { |
|||
return List.of(); |
|||
} |
|||
return documents.stream().map(SourceReference::fromDocument).toList(); |
|||
} |
|||
|
|||
private static SourceReference fromDocument(Document document) { |
|||
Map<String, Object> metadata = document.getMetadata(); |
|||
// Preserve the published source endpoint's distance semantics; other retrievers expose score. |
|||
Object score = metadata.get("distance") != null ? metadata.get("distance") : metadata.get("score"); |
|||
return new SourceReference(stringValue(metadata.get("documentId")), |
|||
stringValue(metadata.get("title")), stringValue(metadata.get("sourceName")), |
|||
integerValue(metadata.get("chunkIndex")), doubleValue(score), document.getText()); |
|||
} |
|||
|
|||
private static String stringValue(Object value) { |
|||
return value == null ? null : value.toString(); |
|||
} |
|||
|
|||
private static Integer integerValue(Object value) { |
|||
if (value == null) return null; |
|||
try { |
|||
return Integer.valueOf(value.toString()); |
|||
} catch (NumberFormatException ignored) { |
|||
return null; |
|||
} |
|||
} |
|||
|
|||
private static Double doubleValue(Object value) { |
|||
if (value == null) return null; |
|||
try { |
|||
double number = Double.parseDouble(value.toString()); |
|||
return Double.isFinite(number) ? number : null; |
|||
} catch (NumberFormatException ignored) { |
|||
return null; |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,250 @@ |
|||
package com.wok.supportbot; |
|||
|
|||
import com.fasterxml.jackson.databind.JsonNode; |
|||
import com.fasterxml.jackson.databind.ObjectMapper; |
|||
import com.wok.supportbot.app.AssistantApp; |
|||
import com.wok.supportbot.app.ChatContext; |
|||
import com.wok.supportbot.app.ChatPipeline; |
|||
import com.wok.supportbot.app.ChatRequest; |
|||
import com.wok.supportbot.app.ChatResult; |
|||
import com.wok.supportbot.app.SourceReference; |
|||
import com.wok.supportbot.chatmemory.DatabaseChatMemory; |
|||
import com.wok.supportbot.config.ChatModelFactory; |
|||
import com.wok.supportbot.config.SimpleCircuitBreaker; |
|||
import com.wok.supportbot.entity.LlmCallTrace; |
|||
import com.wok.supportbot.service.AiModelConfigService; |
|||
import com.wok.supportbot.service.ContentSafetyService; |
|||
import com.wok.supportbot.service.LlmCallTraceService; |
|||
import org.junit.jupiter.api.BeforeEach; |
|||
import org.junit.jupiter.api.Test; |
|||
import org.junit.jupiter.params.ParameterizedTest; |
|||
import org.junit.jupiter.params.provider.ValueSource; |
|||
import org.mockito.ArgumentCaptor; |
|||
import org.springframework.ai.chat.client.ChatClient; |
|||
import org.springframework.ai.chat.messages.AssistantMessage; |
|||
import org.springframework.ai.chat.model.ChatResponse; |
|||
import org.springframework.ai.chat.model.Generation; |
|||
import org.springframework.ai.document.Document; |
|||
import org.springframework.test.util.ReflectionTestUtils; |
|||
import reactor.core.publisher.Flux; |
|||
|
|||
import java.time.Duration; |
|||
import java.util.ArrayList; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
import java.util.Optional; |
|||
import java.util.concurrent.atomic.AtomicBoolean; |
|||
|
|||
import static org.junit.jupiter.api.Assertions.*; |
|||
import static org.mockito.ArgumentMatchers.*; |
|||
import static org.mockito.Mockito.*; |
|||
|
|||
class AnswerTransportTests { |
|||
private static final ObjectMapper JSON = new ObjectMapper(); |
|||
private AssistantApp app; |
|||
private ChatPipeline pipeline; |
|||
private ChatClient.ChatClientRequestSpec spec; |
|||
private ChatClient.CallResponseSpec call; |
|||
private ChatClient.StreamResponseSpec stream; |
|||
private LlmCallTraceService traces; |
|||
|
|||
@BeforeEach |
|||
@SuppressWarnings("unchecked") |
|||
void setUp() { |
|||
app = new AssistantApp(mock(ChatModelFactory.class), mock(DatabaseChatMemory.class)); |
|||
pipeline = mock(ChatPipeline.class); |
|||
traces = mock(LlmCallTraceService.class); |
|||
ContentSafetyService safety = mock(ContentSafetyService.class); |
|||
when(safety.mask(any())).thenAnswer(invocation -> invocation.getArgument(0)); |
|||
ReflectionTestUtils.setField(app, "chatPipeline", pipeline); |
|||
ReflectionTestUtils.setField(app, "llmCallTraceService", traces); |
|||
ReflectionTestUtils.setField(app, "contentSafetyService", safety); |
|||
ReflectionTestUtils.setField(app, "aiModelConfigService", mock(AiModelConfigService.class)); |
|||
ChatClient client = mock(ChatClient.class); |
|||
spec = mock(ChatClient.ChatClientRequestSpec.class, RETURNS_SELF); |
|||
call = mock(ChatClient.CallResponseSpec.class); |
|||
stream = mock(ChatClient.StreamResponseSpec.class); |
|||
when(client.prompt()).thenReturn(spec); |
|||
when(spec.call()).thenReturn(call); |
|||
when(spec.stream()).thenReturn(stream); |
|||
Map<String, ChatClient> cache = (Map<String, ChatClient>) ReflectionTestUtils.getField(app, "chatClientCache"); |
|||
cache.put("CHAT:none", client); |
|||
} |
|||
|
|||
@ParameterizedTest |
|||
@ValueSource(booleans = {true, false}) |
|||
void synchronousAnswerUsesOnlyThisBuildsDocuments(boolean rag) throws Exception { |
|||
ChatContext ctx = context(rag); |
|||
List<Document> documents = rag ? List.of(document()) : List.of(); |
|||
when(pipeline.buildRequest(ctx)).thenReturn(request(ctx, documents, null)); |
|||
when(call.chatResponse()).thenReturn(response("回答")); |
|||
|
|||
ChatResult result = app.chatWithEvents(ctx); |
|||
|
|||
assertEquals("回答", result.text()); |
|||
assertEquals(SourceReference.fromDocuments(documents), result.sources()); |
|||
assertTrue(result.suggestions().isEmpty()); |
|||
JsonNode json = JSON.valueToTree(result); |
|||
if (rag) { |
|||
assertEquals("9223372036854775807", json.at("/sources/0/documentId").asText()); |
|||
assertTrue(json.at("/sources/0/documentId").isTextual()); |
|||
} |
|||
verify(pipeline).buildRequest(ctx); |
|||
verifyNoMoreInteractions(pipeline); |
|||
verify(call).chatResponse(); |
|||
} |
|||
|
|||
@ParameterizedTest |
|||
@ValueSource(booleans = {true, false}) |
|||
void completedStreamCarriesExactlyOneMetadataChunkBeforeStop(boolean rag) throws Exception { |
|||
ChatContext ctx = context(rag); |
|||
List<Document> documents = rag ? List.of(document()) : List.of(); |
|||
when(pipeline.buildRequest(ctx)).thenReturn(request(ctx, documents, null)); |
|||
when(stream.chatResponse()).thenReturn(Flux.just(response("## "), response("回答\n"))); |
|||
|
|||
Flux<String> result = app.chatStreamOpenAi(ctx); |
|||
verifyNoInteractions(pipeline); |
|||
List<String> chunks = result.collectList().block(Duration.ofSeconds(5)); |
|||
assertEnvelope(chunks, SourceReference.fromDocuments(documents), "## 回答\n"); |
|||
verify(pipeline).buildRequest(ctx); |
|||
verifyNoMoreInteractions(pipeline); |
|||
verify(stream).chatResponse(); |
|||
ArgumentCaptor<LlmCallTrace> trace = ArgumentCaptor.forClass(LlmCallTrace.class); |
|||
verify(traces, timeout(1000)).recordAsync(trace.capture()); |
|||
assertEquals("## 回答\n", trace.getValue().getAiResponse()); |
|||
assertEquals("COMPLETE", trace.getValue().getStatus()); |
|||
} |
|||
|
|||
@Test |
|||
void faqHasEmptySourcesWithoutModelInvocation() throws Exception { |
|||
ChatContext ctx = context(true); |
|||
when(pipeline.buildRequest(ctx)).thenReturn(request(ctx, List.of(), "标准答案")); |
|||
assertEnvelope(app.chatStreamOpenAi(ctx).collectList().block(Duration.ofSeconds(5)), List.of(), "标准答案"); |
|||
verify(pipeline).buildRequest(ctx); |
|||
verifyNoMoreInteractions(pipeline); |
|||
verifyNoInteractions(call, stream); |
|||
} |
|||
|
|||
@Test |
|||
void synchronousFaqHasEmptySources() { |
|||
ChatContext ctx = context(true); |
|||
when(pipeline.buildRequest(ctx)).thenReturn(request(ctx, List.of(), "标准答案")); |
|||
ChatResult result = app.chatWithEvents(ctx); |
|||
assertEquals("标准答案", result.text()); |
|||
assertTrue(result.sources().isEmpty()); |
|||
verify(pipeline).buildRequest(ctx); |
|||
verifyNoMoreInteractions(pipeline); |
|||
verifyNoInteractions(call, stream); |
|||
} |
|||
|
|||
@Test |
|||
void circuitFallbackDoesNotBuildOrRetrieve() throws Exception { |
|||
SimpleCircuitBreaker breaker = (SimpleCircuitBreaker) ReflectionTestUtils.getField(app, "aiCircuitBreaker"); |
|||
for (int i = 0; i < 3; i++) breaker.recordFailure(-1L); |
|||
ChatContext ctx = context(true); |
|||
assertTrue(app.chatWithEvents(ctx).sources().isEmpty()); |
|||
assertEnvelope(app.chatStreamOpenAi(ctx).collectList().block(Duration.ofSeconds(5)), |
|||
List.of(), "AI 服务暂时不可用，请稍后重试。"); |
|||
verifyNoInteractions(pipeline, call, stream); |
|||
} |
|||
|
|||
@Test |
|||
void streamFailureDoesNotRepeatGenerationOrExposeUnusedSources() throws Exception { |
|||
ChatContext ctx = context(true); |
|||
when(pipeline.buildRequest(ctx)).thenReturn(request(ctx, List.of(document()), null)); |
|||
when(stream.chatResponse()).thenReturn(Flux.concat(Flux.just(response("部分回答")), |
|||
Flux.error(new IllegalStateException("模型超时")))); |
|||
List<String> chunks = app.chatStreamOpenAi(ctx).collectList().block(Duration.ofSeconds(5)); |
|||
assertEnvelope(chunks, List.of(), "部分回答抱歉，AI 服务调用失败：模型超时"); |
|||
verify(pipeline).buildRequest(ctx); |
|||
verifyNoMoreInteractions(pipeline); |
|||
verify(stream).chatResponse(); |
|||
verifyNoInteractions(call); |
|||
} |
|||
|
|||
@Test |
|||
void cancellationPropagatesWithoutMetadataOrSecondRequest() throws Exception { |
|||
ChatContext ctx = context(true); |
|||
AtomicBoolean cancelled = new AtomicBoolean(); |
|||
when(pipeline.buildRequest(ctx)).thenReturn(request(ctx, List.of(document()), null)); |
|||
when(stream.chatResponse()).thenReturn(Flux.concat(Flux.just(response("首段")), Flux.<ChatResponse>never()) |
|||
.doOnCancel(() -> cancelled.set(true))); |
|||
List<String> chunks = app.chatStreamOpenAi(ctx).take(2).collectList().block(Duration.ofSeconds(5)); |
|||
assertEquals(2, chunks.size()); |
|||
assertEquals("首段", JSON.readTree(chunks.get(1)).at("/choices/0/delta/content").asText()); |
|||
assertFalse(chunks.stream().anyMatch(chunk -> chunk.contains("\"sources\"") || chunk.equals("[DONE]"))); |
|||
verify(traces, timeout(1000)).recordAsync(argThat(trace -> "CANCEL".equals(trace.getStatus()))); |
|||
assertTrue(cancelled.get()); |
|||
verify(pipeline).buildRequest(ctx); |
|||
verifyNoMoreInteractions(pipeline); |
|||
verify(stream).chatResponse(); |
|||
verifyNoInteractions(call); |
|||
} |
|||
|
|||
@Test |
|||
void sourceSerializationPreservesNullableFieldsDistanceAndSnippetBoundaries() { |
|||
Document doc = document(); |
|||
SourceReference source = SourceReference.fromDocuments(List.of(doc)).get(0); |
|||
assertEquals("9223372036854775807", source.documentId()); |
|||
assertEquals(0.25, source.score()); |
|||
assertEquals(2, source.chunkIndex()); |
|||
assertEquals(160, source.snippet().length()); |
|||
assertTrue(source.snippet().endsWith("…")); |
|||
SourceReference nullable = SourceReference.fromDocuments(List.of(new Document("short"))).get(0); |
|||
JsonNode json = JSON.copy().setSerializationInclusion(com.fasterxml.jackson.annotation.JsonInclude.Include.NON_NULL) |
|||
.valueToTree(nullable); |
|||
assertEquals(6, json.size()); |
|||
assertTrue(json.get("documentId").isNull()); |
|||
assertTrue(json.get("score").isNull()); |
|||
SourceReference unicode = new SourceReference(null, null, null, null, null, "a".repeat(158) + "😀xx"); |
|||
assertTrue(unicode.snippet().length() <= 160); |
|||
assertFalse(Character.isHighSurrogate(unicode.snippet().charAt(unicode.snippet().length() - 2))); |
|||
ArrayList<SourceReference> mutable = new ArrayList<>(List.of(source)); |
|||
ChatResult result = new ChatResult("answer", null, null, mutable); |
|||
mutable.clear(); |
|||
assertEquals(List.of(source), result.sources()); |
|||
} |
|||
|
|||
private static void assertEnvelope(List<String> chunks, List<SourceReference> sources, String text) throws Exception { |
|||
assertNotNull(chunks); |
|||
assertEquals("[DONE]", chunks.get(chunks.size() - 1)); |
|||
JsonNode metadata = JSON.readTree(chunks.get(chunks.size() - 3)); |
|||
JsonNode stop = JSON.readTree(chunks.get(chunks.size() - 2)); |
|||
assertEquals("stop", stop.at("/choices/0/finish_reason").asText()); |
|||
assertEquals(0, metadata.get("choices").size()); |
|||
assertEquals(JSON.valueToTree(sources), metadata.get("sources")); |
|||
StringBuilder answer = new StringBuilder(); |
|||
int metadataCount = 0; |
|||
for (String raw : chunks.subList(0, chunks.size() - 1)) { |
|||
JsonNode chunk = JSON.readTree(raw); |
|||
assertEquals("chat.completion.chunk", chunk.get("object").asText()); |
|||
assertEquals(metadata.get("id"), chunk.get("id")); |
|||
assertEquals(metadata.get("model"), chunk.get("model")); |
|||
assertEquals(metadata.get("created"), chunk.get("created")); |
|||
if (chunk.has("sources")) metadataCount++; |
|||
answer.append(chunk.at("/choices/0/delta/content").asText("")); |
|||
} |
|||
assertEquals(1, metadataCount); |
|||
assertEquals(text, answer.toString()); |
|||
} |
|||
|
|||
private static Document document() { |
|||
return new Document("知识".repeat(100), Map.of("documentId", Long.MAX_VALUE, "title", "授权文档", |
|||
"sourceName", "manual.pdf", "chunkIndex", "2", "distance", 0.25, "score", 0.9)); |
|||
} |
|||
|
|||
private static ChatContext context(boolean rag) { |
|||
return new ChatContext("退货", "transport-chat", "CHAT", null, null, List.of(7L), |
|||
"NONE", rag, false, 11L, "售后", "account", null, null); |
|||
} |
|||
|
|||
private static ChatRequest request(ChatContext ctx, List<Document> documents, String faq) { |
|||
return new ChatRequest(ctx, ctx.message(), "system", Optional.ofNullable(faq), "system", |
|||
documents.isEmpty() ? null : "资料", documents.size(), faq != null ? "FAQ" : "RAG", |
|||
"VECTOR", documents, null); |
|||
} |
|||
|
|||
private static ChatResponse response(String text) { |
|||
return new ChatResponse(List.of(new Generation(new AssistantMessage(text)))); |
|||
} |
|||
} |
|||
@ -0,0 +1,166 @@ |
|||
package com.wok.supportbot; |
|||
|
|||
import com.wok.supportbot.app.AssistantApp; |
|||
import com.wok.supportbot.app.ChatContext; |
|||
import com.wok.supportbot.app.ChatResult; |
|||
import com.wok.supportbot.app.SourceReference; |
|||
import com.wok.supportbot.config.RoleAccessConfig; |
|||
import com.wok.supportbot.controller.AiController; |
|||
import com.wok.supportbot.controller.OpenApiController; |
|||
import com.wok.supportbot.entity.ApiKey; |
|||
import com.wok.supportbot.rag.CategoryFilter; |
|||
import com.wok.supportbot.security.JwtTokenProvider; |
|||
import com.wok.supportbot.security.SdkAuthFilter; |
|||
import com.wok.supportbot.security.SdkJwtTokenProvider; |
|||
import com.wok.supportbot.service.ConversationService; |
|||
import com.wok.supportbot.service.CustomerServiceRoleService; |
|||
import com.wok.supportbot.service.CustomerServiceRoleService.RoleScope; |
|||
import io.jsonwebtoken.Claims; |
|||
import org.junit.jupiter.api.BeforeEach; |
|||
import org.junit.jupiter.api.Test; |
|||
import org.mockito.ArgumentCaptor; |
|||
import org.springframework.jdbc.core.JdbcTemplate; |
|||
import org.springframework.mock.web.MockHttpServletRequest; |
|||
import org.springframework.test.util.ReflectionTestUtils; |
|||
import org.springframework.test.web.servlet.MockMvc; |
|||
import org.springframework.test.web.servlet.setup.MockMvcBuilders; |
|||
|
|||
import java.util.List; |
|||
import java.util.Set; |
|||
|
|||
import static org.junit.jupiter.api.Assertions.*; |
|||
import static org.mockito.ArgumentMatchers.any; |
|||
import static org.mockito.Mockito.*; |
|||
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get; |
|||
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*; |
|||
|
|||
class ChatResultEndpointTests { |
|||
private AssistantApp assistant; |
|||
private CustomerServiceRoleService roles; |
|||
private ConversationService conversations; |
|||
private RoleAccessConfig access; |
|||
private AiController controller; |
|||
private MockMvc mvc; |
|||
private SdkJwtTokenProvider tokens; |
|||
|
|||
@BeforeEach |
|||
void setUp() { |
|||
assistant = mock(AssistantApp.class); |
|||
roles = mock(CustomerServiceRoleService.class); |
|||
conversations = mock(ConversationService.class); |
|||
access = new RoleAccessConfig(); |
|||
controller = new AiController(); |
|||
ReflectionTestUtils.setField(controller, "assistantApp", assistant); |
|||
ReflectionTestUtils.setField(controller, "customerServiceRoleService", roles); |
|||
ReflectionTestUtils.setField(controller, "conversationService", conversations); |
|||
ReflectionTestUtils.setField(controller, "roleAccessConfig", access); |
|||
ReflectionTestUtils.setField(controller, "categoryFilter", mock(CategoryFilter.class)); |
|||
tokens = mock(SdkJwtTokenProvider.class); |
|||
SdkAuthFilter filter = new SdkAuthFilter(tokens, mock(JwtTokenProvider.class), mock(JdbcTemplate.class)); |
|||
mvc = MockMvcBuilders.standaloneSetup(controller).addFilters(filter).build(); |
|||
} |
|||
|
|||
@Test |
|||
void resultRejectsMissingTokenAndUnauthorizedRoleBeforeGeneration() throws Exception { |
|||
mvc.perform(get("/ai/chat/result").servletPath("/ai/chat/result").param("message", "问题")) |
|||
.andExpect(status().isUnauthorized()); |
|||
authorize(); |
|||
mvc.perform(get("/ai/chat/result").servletPath("/ai/chat/result") |
|||
.header("Authorization", "Bearer valid").param("message", "问题").param("roleId", "99")) |
|||
.andExpect(status().isForbidden()); |
|||
verifyNoInteractions(assistant, roles, conversations); |
|||
} |
|||
|
|||
@Test |
|||
void resultKeepsRoleScopeAccountBindingAndDecodedImages() throws Exception { |
|||
authorize(); |
|||
when(roles.getRoleScope(11L)).thenReturn(new RoleScope(true, "售后", "授权人设", List.of(7L), List.of("tool"))); |
|||
SourceReference source = new SourceReference("9223372036854775807", "授权文档", null, null, null, "片段"); |
|||
when(assistant.chatWithEvents(any())).thenReturn(new ChatResult("回答", List.of(), List.of(), List.of(source))); |
|||
mvc.perform(get("/ai/chat/result").servletPath("/ai/chat/result") |
|||
.header("Authorization", "Bearer valid").param("message", "问题").param("chatId", "chat") |
|||
.param("roleId", "11").param("accountId", " account ").param("systemPrompt", "越权人设") |
|||
.param("enableRag", "true").param("categoryIds", "99").param("imageUrls", "https%3A%2F%2Fexample.org%2Fa.png")) |
|||
.andExpect(status().isOk()).andExpect(content().contentTypeCompatibleWith("application/json")) |
|||
.andExpect(jsonPath("$.text").value("回答")) |
|||
.andExpect(jsonPath("$.data").doesNotExist()) |
|||
.andExpect(jsonPath("$.sources[0].documentId").value("9223372036854775807")) |
|||
.andExpect(jsonPath("$.mcpEvents").isArray()).andExpect(jsonPath("$.suggestions").isArray()); |
|||
ArgumentCaptor<ChatContext> context = ArgumentCaptor.forClass(ChatContext.class); |
|||
verify(assistant).chatWithEvents(context.capture()); |
|||
ChatContext ctx = context.getValue(); |
|||
assertEquals(List.of(7L), ctx.categoryIds()); |
|||
assertEquals("授权人设", ctx.systemPrompt()); |
|||
assertEquals(List.of("tool"), ctx.allowedMcpTools()); |
|||
assertEquals(List.of("https://example.org/a.png"), ctx.imageUrls()); |
|||
assertEquals("NONE", ctx.rewriteStrategy()); |
|||
assertTrue(ctx.enableRag()); |
|||
verify(conversations).bindConversation("chat", "account", 11L); |
|||
verifyNoMoreInteractions(assistant); |
|||
} |
|||
|
|||
@Test |
|||
void strictIsolationAndExplicitRewriteApplyToBothSyncEndpoints() { |
|||
access.setStrictIsolation(true); |
|||
when(roles.getRoleScope(11L)).thenReturn(new RoleScope(true, "售后", "人设", List.of(), List.of())); |
|||
when(assistant.chatWithEvents(any())).thenReturn(new ChatResult("回答", List.of(), List.of(), List.of())); |
|||
assertEquals("回答", controller.chatSync("问题", "chat", 11L, "account", null, true, |
|||
"REWRITE", null, "99", null)); |
|||
ArgumentCaptor<ChatContext> context = ArgumentCaptor.forClass(ChatContext.class); |
|||
verify(assistant).chatWithEvents(context.capture()); |
|||
assertFalse(context.getValue().enableRag()); |
|||
assertTrue(context.getValue().categoryIds().isEmpty()); |
|||
assertEquals("REWRITE", context.getValue().rewriteStrategy()); |
|||
verifyNoMoreInteractions(assistant); |
|||
} |
|||
|
|||
@Test |
|||
void openApiRetainsEnvelopeAndAddsSameAnswerSourcesWithNoneDefault() { |
|||
OpenApiController open = new OpenApiController(); |
|||
ReflectionTestUtils.setField(open, "assistantApp", assistant); |
|||
ReflectionTestUtils.setField(open, "customerServiceRoleService", roles); |
|||
ReflectionTestUtils.setField(open, "categoryFilter", mock(CategoryFilter.class)); |
|||
ReflectionTestUtils.setField(open, "roleAccessConfig", access); |
|||
when(roles.getRoleScope(11L)).thenReturn(new RoleScope(true, "售后", "人设", List.of(7L), List.of())); |
|||
SourceReference source = new SourceReference("9223372036854775807", "授权文档", null, null, null, "片段"); |
|||
when(assistant.chatWithEvents(any())).thenReturn(new ChatResult("回答", List.of(), List.of(), List.of(source))); |
|||
ApiKey key = new ApiKey(); |
|||
key.setId(3L); |
|||
key.setRoleIds("[11]"); |
|||
MockHttpServletRequest request = new MockHttpServletRequest(); |
|||
request.setAttribute("apiKey", key); |
|||
var result = open.chat("问题", "11", "chat", "99", null, true, request); |
|||
assertEquals(200, result.getStatusCode().value()); |
|||
assertEquals(true, result.getBody().get("success")); |
|||
var data = (java.util.Map<?, ?>) result.getBody().get("data"); |
|||
assertEquals("回答", data.get("reply")); |
|||
assertEquals(List.of(source), data.get("sources")); |
|||
ArgumentCaptor<ChatContext> context = ArgumentCaptor.forClass(ChatContext.class); |
|||
verify(assistant).chatWithEvents(context.capture()); |
|||
assertEquals("NONE", context.getValue().rewriteStrategy()); |
|||
assertEquals(List.of(7L), context.getValue().categoryIds()); |
|||
verifyNoMoreInteractions(assistant); |
|||
} |
|||
|
|||
@Test |
|||
void explicitSourcesApiUsesTheSameCitationSerializer() { |
|||
ReflectionTestUtils.setField(controller, "chatPipeline", mock(com.wok.supportbot.app.ChatPipeline.class)); |
|||
when(roles.getRoleScope(11L)).thenReturn(new RoleScope(true, "售后", "人设", List.of(7L), List.of())); |
|||
var document = new org.springframework.ai.document.Document("片段".repeat(100), |
|||
java.util.Map.of("documentId", Long.MAX_VALUE, "distance", 0.2)); |
|||
when(assistant.retrieveSources(any())).thenReturn(List.of(document)); |
|||
var result = controller.chatSources("问题", "chat", null, 11L, "account", null, "99"); |
|||
assertEquals(SourceReference.fromDocuments(List.of(document)), result.get("data")); |
|||
ArgumentCaptor<ChatContext> context = ArgumentCaptor.forClass(ChatContext.class); |
|||
verify(assistant).retrieveSources(context.capture()); |
|||
assertEquals(List.of(7L), context.getValue().categoryIds()); |
|||
assertEquals("NONE", context.getValue().rewriteStrategy()); |
|||
verifyNoMoreInteractions(assistant); |
|||
} |
|||
|
|||
private void authorize() { |
|||
Claims claims = mock(Claims.class); |
|||
when(tokens.parseToken("valid")).thenReturn(claims); |
|||
when(tokens.getAllowedRoleIds(claims)).thenReturn(Set.of(11L)); |
|||
} |
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue