Update app.py

app.py

@@ -12,7 +12,8 @@ from tooluniverse import ToolUniverse
 # Patch PyTorch to allow loading old numpy pickles
 torch.serialization.add_safe_globals([
     numpy.core.multiarray._reconstruct,
-    numpy.ndarray
+    numpy.ndarray,
+    numpy.dtype
 ])
 
 logging.basicConfig(
@@ -63,7 +64,11 @@ def patch_embedding_loading():
             tools = tooluniverse.get_all_tools() if hasattr(tooluniverse, "get_all_tools") else getattr(tooluniverse, "tools", [])
             if len(tools) != len(self.tool_desc_embedding):
                 logger.warning("Tool count mismatch.")
-                self.tool_desc_embedding = self.tool_desc_embedding[:len(tools)]
+                if len(self.tool_desc_embedding) > len(tools):
+                    self.tool_desc_embedding = self.tool_desc_embedding[:len(tools)]
+                else:
+                    padding = self.tool_desc_embedding[-1].unsqueeze(0).repeat(len(tools) - len(self.tool_desc_embedding), 1)
+                    self.tool_desc_embedding = torch.cat([self.tool_desc_embedding, padding], dim=0)
             return True
         except Exception as e:
             logger.error(f"Embedding load failed: {e}")