fix(tool): fix system tool icon (#158)

Signed-off-by: Milan Gallas <[email protected]>
i-am-bee · Dec 18, 2024 · 38a73cb · 38a73cb
1 parent 5851900
commit 38a73cb
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 8 deletions.
diff --git a/src/modules/assistants/tools/ToolNameWithTooltip.tsx b/src/modules/assistants/tools/ToolNameWithTooltip.tsx
@@ -15,10 +15,7 @@
  */
 
 import { ToolReference } from '@/app/api/tools/types';
-import {
-  getStaticToolName,
-  useToolInfo,
-} from '@/modules/tools/hooks/useToolInfo';
+import { useToolInfo } from '@/modules/tools/hooks/useToolInfo';
 import classes from './ToolNameWithTooltip.module.scss';
 import { ToolInfoButton } from './ToolInfoButton';
 import { useProjectContext } from '@/layout/providers/ProjectProvider';

diff --git a/src/modules/auth/AcceptToUForm.module.scss b/src/modules/auth/AcceptToUForm.module.scss
@@ -44,8 +44,8 @@
   }
 
   :global(
-      .#{$prefix}--checkbox:not(:checked) ~ .#{$prefix}--checkbox-label::before
-    ) {
+    .#{$prefix}--checkbox:not(:checked) ~ .#{$prefix}--checkbox-label::before
+  ) {
     border-color: currentColor;
   }
 }

diff --git a/src/modules/tools/hooks/useToolInfo.tsx b/src/modules/tools/hooks/useToolInfo.tsx
@@ -39,6 +39,7 @@ import { encodeEntityWithMetadata } from '@/app/api/utils';
 import { Organization } from '@/app/api/organization/types';
 import { Project } from '@/app/api/projects/types';
 import { useTools } from './useTools';
+import capitalize from 'lodash/capitalize';
 
 export function useToolInfo({
   toolReference,
@@ -87,7 +88,7 @@ export function useToolInfo({
         if (tool?.name === 'DuckDuckGo') return DuckDuckGo;
         return SkeletonIcon;
       }
-      return SYSTEM_TOOL_ICONS[id];
+      return SYSTEM_TOOL_ICONS[id] ?? Tools;
     }
     if (type === 'file_search') return SearchLocate;
     if (type === 'code_interpreter' || type === 'function') return Code;
@@ -123,7 +124,7 @@ export function getStaticToolName({ type, id, tool }: ToolReference) {
 
   switch (type) {
     case 'system':
-      return SYSTEM_TOOL_NAME[id];
+      return SYSTEM_TOOL_NAME[id] ?? id.split('_').map(capitalize).join(' ');
     case 'code_interpreter':
       return 'Python Intepreter';
     case 'function':