Pass selected folder to backend when searching for models.

transformerlab · May 27, 2024 · 702efc5 · 702efc5
1 parent 61d74d3
commit 702efc5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/src/renderer/components/ModelZoo/ImportModelsBar.tsx b/src/renderer/components/ModelZoo/ImportModelsBar.tsx
@@ -89,7 +89,6 @@ export default function ModelDetails({}) {
             <Button
               size="sm"
               sx={{ height: '30px' }}
-              disabled={process.env.NODE_ENV != "development"}
               endDecorator={<PlusIcon />}
               onClick={() => {
                 setImportModelsModalOpen(true);

diff --git a/src/renderer/components/ModelZoo/ImportModelsModal.tsx b/src/renderer/components/ModelZoo/ImportModelsModal.tsx
@@ -25,14 +25,14 @@ const fetcher = (url) => fetch(url).then((res) => res.json());
 
 export default function ImportModelsModal({ open, setOpen}) {
     const [importing, setImporting] = useState(false);
-    const [modelFolder, setModelFolder] = useState(null); // Disabled folder support
+    const [modelFolder, setModelFolder] = useState("");
 
     const {
         data: modelsData,
         error: modelsError,
         isLoading: isLoading,
     } = useSWR(
-        chatAPI.Endpoints.Models.GetLocalUninstalled(),
+        chatAPI.Endpoints.Models.SearchForLocalUninstalledModels(modelFolder),
         fetcher
     );
     const models = modelsData?.data;
@@ -166,12 +166,12 @@ export default function ImportModelsModal({ open, setOpen}) {
                       const webkitRelativePath = firstfile.webkitRelativePath;
                       const parentPath = firstfilepath.slice(0, -1*webkitRelativePath.length);
                       const topRelativePathDir = webkitRelativePath.split('/')[0];
-                      const fullPath = parentPath + '/' + topRelativePathDir;
+                      const fullPath = parentPath + topRelativePathDir;
                       console.log(firstfile);
                       console.log(webkitRelativePath);
                       setModelFolder(fullPath);
                     } else {
-                      setModelFolder(null);
+                      setModelFolder("");
                     }
                 }}
                 />

diff --git a/src/renderer/lib/transformerlab-api-sdk.ts b/src/renderer/lib/transformerlab-api-sdk.ts
@@ -608,7 +608,8 @@ Endpoints.Models = {
     API_URL() + 'model/details/' + convertSlashInUrl(modelId),
   GetLocalHFConfig: (modelId: string) =>
     API_URL() + 'model/get_local_hfconfig?model_id=' + modelId,
-  GetLocalUninstalled: () => API_URL() + 'model/list_local_uninstalled',
+  SearchForLocalUninstalledModels: (path: string) =>
+    API_URL() + 'model/list_local_uninstalled?path=' + path,
   ImportLocal: (modelSource: string, modelId: string) =>
     API_URL() +
     'model/import_local?model_source=' +