ai-microcore 4.0.0.dev15__tar.gz → 4.0.0.dev17__tar.gz

{ai_microcore-4.0.0.dev15 → ai_microcore-4.0.0.dev17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-microcore
-Version: 4.0.0.dev15
+Version: 4.0.0.dev17
 Summary: # Minimalistic Foundation for AI Applications
 Keywords: llm,large language models,ai,similarity search,ai search,gpt,openai,framework,adapter
 Author-email: Vitalii Stepanenko <mail@vitalii.in>

{ai_microcore-4.0.0.dev15 → ai_microcore-4.0.0.dev17}/microcore/__init__.py

@@ -184,4 +184,4 @@ __all__ = [
     # "wrappers",
 ]
 
-__version__ = "4.0.0-dev15"
+__version__ = "4.0.0-dev17"

{ai_microcore-4.0.0.dev15 → ai_microcore-4.0.0.dev17}/microcore/configuration.py

@@ -113,7 +113,7 @@ class BaseConfig:
         if self.USE_DOT_ENV:
             if self.DOT_ENV_FILE or not _default_dotenv_loaded:
                 fp = self.DOT_ENV_FILE
-                if fp and "~" in fp:
+                if fp and "~" in str(fp):
                     fp = Path(fp).expanduser()
                 dotenv.load_dotenv(
                     override=True,

{ai_microcore-4.0.0.dev15 → ai_microcore-4.0.0.dev17}/microcore/mcp.py

@@ -138,9 +138,14 @@ class MCPConnection:
         mcp_tools = await self.session.list_tools()
         self.tools = Tools.from_list([Tool.from_mcp(tool) for tool in mcp_tools.tools])
         if use_cache:
-            ToolsCache.write(self.url, self.tools)
+            self.update_tools_cache()
         return self.tools
 
+    def update_tools_cache(self):
+        if self.tools is None:
+            raise RuntimeError("Tools are not fetched yet. Call fetch_tools() first.")
+        ToolsCache.write(self.url, self.tools)
+
     def __del__(self):
         self._del_event.set()
 
@@ -264,6 +269,9 @@ class MCPServer:
     ) -> MCPConnection:
         return await MCPConnection.init(self.url, fetch_tools=fetch_tools, use_cache=use_cache)
 
+    def get_tools_cache(self) -> Tools | None:
+        return ToolsCache.read(self.url)
+
 
 class MCPRegistry(dict[str, MCPServer]):
     def __init__(self, server_configs: list[dict | str]):
@@ -281,6 +289,19 @@ class MCPRegistry(dict[str, MCPServer]):
             raise ValueError(f"MCP server '{server_name}' not found in registry")
         return self[server_name]
 
+    async def precache_tools(self):
+        async def precache_server_tools(server_name):
+            conn = await self.get(server_name).connect(
+                fetch_tools=True,
+                use_cache=False
+            )
+            try:
+                conn.update_tools_cache()
+            finally:
+                await conn.close()
+
+        await asyncio.gather(*[precache_server_tools(srv) for srv in self.keys()])
+
     async def connect_to(
         self,
         server_name: str,