fix/Add isModel flag to AgentTools component (#2876)

This commit is contained in:
Yeuoly 2024-03-18 17:01:25 +08:00 committed by GitHub
parent 95b74c211d
commit 01108e6172
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 5 deletions

View File

@ -210,6 +210,7 @@ const AgentTools: FC = () => {
setting={currentTool?.tool_parameters as any} setting={currentTool?.tool_parameters as any}
collection={currentTool?.collection as Collection} collection={currentTool?.collection as Collection}
isBuiltIn={currentTool?.collection?.type === CollectionType.builtIn} isBuiltIn={currentTool?.collection?.type === CollectionType.builtIn}
isModel={currentTool?.collection?.type === CollectionType.model}
onSave={handleToolSettingChange} onSave={handleToolSettingChange}
onHide={() => setIsShowSettingTool(false)} onHide={() => setIsShowSettingTool(false)}
/>) />)

View File

@ -58,11 +58,16 @@ const SettingBuiltInTool: FC<Props> = ({
(async () => { (async () => {
setIsLoading(true) setIsLoading(true)
try { try {
const list = isBuiltIn const list = await new Promise<Tool[]>((resolve) => {
? await fetchBuiltInToolList(collection.name) (async function () {
: isModel if (isModel)
? await fetchModelToolList(collection.name) resolve(await fetchModelToolList(collection.name))
: await fetchCustomToolList(collection.name) else if (isBuiltIn)
resolve(await fetchBuiltInToolList(collection.name))
else
resolve(await fetchCustomToolList(collection.name))
}())
})
setTools(list) setTools(list)
const currTool = list.find(tool => tool.name === toolName) const currTool = list.find(tool => tool.name === toolName)
if (currTool) { if (currTool) {