🐛 fix: allow builtin tools to trigger AI message (lobehub#3135)

geekhuashan · Jul 3, 2024 · 6c4c8f7 · 6c4c8f7
1 parent 4dbf795
commit 6c4c8f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/store/chat/slices/plugin/action.ts b/src/store/chat/slices/plugin/action.ts
@@ -117,7 +117,7 @@ export const chatPlugin: StateCreator<
 
     if (!content) return;
 
-    await action(id, content);
+    return await action(id, content);
   },
 
   invokeDefaultTypePlugin: async (id, payload) => {
@@ -220,7 +220,7 @@ export const chatPlugin: StateCreator<
       // trigger the plugin call
       const data = await get().internal_invokeDifferentTypePlugin(id, payload);
 
-      if (payload.type === 'default' && data) {
+      if ((payload.type === 'default' || payload.type === 'builtin') && data) {
         shouldCreateMessage = true;
         latestToolId = id;
       }