Skip to content

Commit 5cd16b8

Browse files
authored
Merge pull request #2745 from ModelEngine-Group/develop_fix_kb_2
✨ Enhance ToolConfigModal and ToolTestPanel with knowledge base selec…
2 parents 6f28802 + 048689d commit 5cd16b8

File tree

2 files changed

+300
-23
lines changed

2 files changed

+300
-23
lines changed

frontend/app/[locale]/agents/components/agentConfig/tool/ToolConfigModal.tsx

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1621,6 +1621,28 @@ export default function ToolConfigModal({
16211621
tool={tool}
16221622
onClose={handleCloseTestPanel}
16231623
configParams={currentParams}
1624+
toolRequiresKbSelection={toolRequiresKbSelection}
1625+
knowledgeBases={knowledgeBases}
1626+
kbLoading={kbLoading}
1627+
selectedKbIds={selectedKbIds}
1628+
selectedKbDisplayNames={selectedKbDisplayNames}
1629+
onOpenKbSelector={(paramIndex) => openKbSelector(paramIndex === -1 ? 0 : paramIndex)}
1630+
onKbSelectionChange={(ids, displayNames) => {
1631+
setSelectedKbIds(ids);
1632+
setSelectedKbDisplayNames(displayNames);
1633+
}}
1634+
onRemoveKb={(index, paramIndex) => {
1635+
if (paramIndex === -1) {
1636+
// Called from test panel - remove from selectedKbIds
1637+
const newIds = selectedKbIds.filter((_, i) => i !== index);
1638+
const newDisplayNames = selectedKbDisplayNames.filter((_, i) => i !== index);
1639+
setSelectedKbIds(newIds);
1640+
setSelectedKbDisplayNames(newDisplayNames);
1641+
} else {
1642+
// Called from config panel
1643+
removeKbFromSelection(index, paramIndex);
1644+
}
1645+
}}
16241646
/>
16251647
)}
16261648
</div>

0 commit comments

Comments
 (0)