fix tool type to be TTool instead of string

PersonaFlow · Sep 24, 2024 · b31d822 · b31d822
1 parent e43ca5e
commit b31d822
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/ui/src/components/features/build-panel/components/create-assistant.tsx b/ui/src/components/features/build-panel/components/create-assistant.tsx
@@ -81,7 +81,7 @@ export function CreateAssistant() {
         (tool) => tool.type === RetrievalType,
       );
       const retrievalTools = [retrievalTool];
-      const containsCodeInterpreter = tools.includes("Code interpretor");
+      // const containsCodeInterpreter = tools.includes("Code interpretor");
       // if (containsCodeInterpreter) retrievalTools.push("Code interpreter");
       form.setValue("config.configurable.tools", retrievalTools);
     }

diff --git a/ui/src/components/features/build-panel/components/edit-assistant.tsx b/ui/src/components/features/build-panel/components/edit-assistant.tsx
@@ -96,10 +96,10 @@ function EditAssistantForm({
       const retrievalTool = availableTools?.find(
         (tool) => tool.type === RetrievalType,
       );
-      const containsCodeInterpreter = tools.includes("Code interpretor");
+      // const containsCodeInterpreter = tools.includes("Code interpretor");
       // if (containsCodeInterpreter) retrievalTools.push("Code interpreter");
       if (retrievalTool) {
-        form.setValue("config.configurable.tools", [retrievalTool.toString()]);
+        form.setValue("config.configurable.tools", [retrievalTool]);
       }
     }
   }, [architectureType]);

diff --git a/ui/src/components/features/build-panel/components/files-dialog.tsx b/ui/src/components/features/build-panel/components/files-dialog.tsx
@@ -125,7 +125,6 @@ export default function FilesDialog({ form, classNames }: TFilesDialog) {
                         defaultValues={formattedAssistantFiles}
                         onValueChange={(selections) => {
                           const fileIds = getFileIds(selections);
-                          console.log(fileIds);
                           field.onChange(fileIds);
                         }}
                       />

diff --git a/ui/src/data-provider/types.ts b/ui/src/data-provider/types.ts
@@ -47,7 +47,7 @@ export type TConfigurable = {
   interrupt_before_action: boolean;
   retrieval_description: string;
   system_message: string;
-  tools: string[];
+  tools: TTool[];
   llm_type: string;
 };