mono/agent-desk-ng
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: harden support, localize knowledge and add operator availability [skip ci]

Browse Source 
Source-only publication of reviewed commit 64e96a7f423673dc29ad56485be9fc79bce8af1c.

Passed in isolation: focused frontend/backend tests, race checks, TypeScript typecheck, lint, Go compilation, Flowgram build, and deterministic SDK build/tests.

Not accepted as a production release: Next.js export/native assembly remains blocked by environment memory limits; browser interaction and MySQL staging acceptance remain unverified. No production server, database/schema, runtime configuration, deployment, release tag, or workflow execution was performed.

Expected source tree: 0a4a7528e9.
This commit is contained in:
mono
2026-10-16 15:21:42 +00:00
parent 299ea50fb5
commit 2f7a491666
176 changed files with 11869 additions and 2716 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
internal/ai/rag/retrieve_context.go
View File

@@ -1,12 +1,19 @@
// Modified in the independent AgentDesk fork: localized knowledge workflows and safe display errors.
package rag
import (
"context"
"fmt"
"strings"
"agent-desk/internal/pkg/i18nx"
)
func (s *retrieve) SelectContextResults(results []RetrieveResult, maxTokens int) []RetrieveResult {
return s.SelectContextResultsWithContext(context.Background(), results, maxTokens)
}
func (s *retrieve) SelectContextResultsWithContext(ctx context.Context, results []RetrieveResult, maxTokens int) []RetrieveResult {
if len(results) == 0 {
return nil
}
@@ -20,7 +27,7 @@ func (s *retrieve) SelectContextResults(results []RetrieveResult, maxTokens int)
if documentUsage[item.DocumentID] >= 2 {
continue
}
chunkText := buildContextChunkText(item)
chunkText := buildContextChunkText(ctx, item)
estimatedTokens := len(chunkText) / 2
if totalTokens+estimatedTokens > maxTokens {
break
@@ -32,15 +39,15 @@ func (s *retrieve) SelectContextResults(results []RetrieveResult, maxTokens int)
return selected
}
func (s *retrieve) BuildContext(_ context.Context, results []RetrieveResult, maxTokens int) string {
func (s *retrieve) BuildContext(ctx context.Context, results []RetrieveResult, maxTokens int) string {
if len(results) == 0 {
return ""
}
normalizedResults := s.SelectContextResults(results, maxTokens)
normalizedResults := s.SelectContextResultsWithContext(ctx, results, maxTokens)
var builder strings.Builder
for _, r := range normalizedResults {
builder.WriteString(buildContextChunkText(r))
builder.WriteString(buildContextChunkText(ctx, r))
}
return builder.String()
@@ -125,26 +132,27 @@ func buildSectionKey(item RetrieveResult) string {
return fmt.Sprintf("%d|chunk:%d", item.DocumentID, item.ChunkNo)
}
func buildContextChunkText(item RetrieveResult) string {
func buildContextChunkText(ctx context.Context, item RetrieveResult) string {
locale := i18nx.LocaleFromContext(ctx)
if item.FaqID > 0 {
title := strings.TrimSpace(item.FaqQuestion)
if title == "" {
title = strings.TrimSpace(item.Title)
}
if title == "" {
title = fmt.Sprintf("FAQ#%d", item.FaqID)
title = i18nx.Getf(locale, "knowledge.rag.context.faqTitle", item.FaqID)
}
return fmt.Sprintf("【FAQ：%s】\n%s\n\n", title, item.Content)
return i18nx.Getf(locale, "knowledge.rag.context.faq", title, item.Content)
}
title := strings.TrimSpace(item.DocumentTitle)
if title == "" {
title = fmt.Sprintf("文档#%d", item.DocumentID)
title = i18nx.Getf(locale, "knowledge.rag.context.documentTitle", item.DocumentID)
}
if item.SectionPath != "" {
return fmt.Sprintf("【文档：%s｜章节：%s】\n%s\n\n", title, item.SectionPath, item.Content)
return i18nx.Getf(locale, "knowledge.rag.context.documentSection", title, item.SectionPath, item.Content)
}
if item.Title != "" {
return fmt.Sprintf("【文档：%s｜标题：%s】\n%s\n\n", title, item.Title, item.Content)
return i18nx.Getf(locale, "knowledge.rag.context.documentHeading", title, item.Title, item.Content)
}
return fmt.Sprintf("【文档：%s】\n%s\n\n", title, item.Content)
return i18nx.Getf(locale, "knowledge.rag.context.document", title, item.Content)
}