universal-mcp 0.1.24rc21__py3-none-any.whl → 0.1.24rc23__py3-none-any.whl

universal_mcp/agentr/README.md

@@ -14,29 +14,30 @@ pip install universal-mcp
 
 ## Usage
 
-The AgentR platform is designed to seamlessly integrate a wide array of tools into your agentic applications. The primary entry point for this is the `Agentr` class, which provides a high-level interface for loading and listing tools.
+The AgentR platform is designed to seamlessly integrate a wide array of tools into your agentic applications. The primary entry point for this is the `AgentrRegistry` class, which provides a high-level interface for loading and listing tools.
 
-### High-Level Client (`Agentr`)
+### High-Level Client (`AgentrRegistry`)
 
-This is the recommended way to get started. It abstracts away the details of the registry and tool management.
+This is the recommended way to get started. It abstracts away the details of the tool management.
 
 ```python
 import os
-from universal_mcp.agentr import Agentr
+from universal_mcp.agentr import AgentrRegistry
 from universal_mcp.tools import ToolFormat
 
 # Initialize the main client
 # It reads from environment variables by default (AGENTR_API_KEY, AGENTR_BASE_URL)
-agentr = Agentr(
+registry = AgentrRegistry(
     api_key=os.environ.get("AGENTR_API_KEY")
 )
 
 # Load specific tools from the AgentR server into the tool manager
-agentr.load_tools(["reddit__search_subreddits", "google-drive__list_files"])
-
-# List the tools that are now loaded and ready to be used
-# You can specify a format compatible with your LLM (e.g., OPENAI)
-tools = agentr.list_tools(format=ToolFormat.OPENAI)
+# The export_tools method loads the tools and then exports them to the specified format.
+tools_config = {
+    "reddit": ["search_subreddits"],
+    "google-drive": ["list_files"]
+}
+tools = await registry.export_tools(tools_config, format=ToolFormat.OPENAI)
 print(tools)
 ```
 
@@ -59,7 +60,7 @@ client = AgentrClient(
 )
 
 # Fetch a list of available applications from the AgentR server
-apps = client.list_apps()
+apps = client.list_all_apps()
 print("Available Apps:", apps)
 
 # Get credentials for a specific app by its ID (e.g., 'reddit')
@@ -71,7 +72,7 @@ except NotAuthorizedError as e:
     print(e)  # "Please ask the user to visit the following url to authorize..."
 
 # List all available tools globally
-all_tools = client.list_tools()
+all_tools = client.list_all_tools()
 print("All Available Tools:", all_tools)
 
 # Example of fetching a single app and a single tool
@@ -80,14 +81,14 @@ if apps:
     app_id = apps[0]['id']
 
     # Fetch a single app's details
-    app_details = client.get_app(app_id)
+    app_details = client.get_app_details(app_id)
     print(f"Fetched details for app '{app_id}':", app_details)
 
 if all_tools:
     tool_id = all_tools[0]['id']
 
     # Fetch a single tool's details
-    tool_details = client.get_tool(tool_id)
+    tool_details = client.get_tool_details(tool_id)
     print(f"Fetched details for tool '{tool_id}':", tool_details)
 ```
 
@@ -134,7 +135,7 @@ registry = AgentrRegistry(client=client)
 
 async def main():
     # List all apps available on the AgentR platform
-    available_apps = await registry.list_apps()
+    available_apps = await registry.list_all_apps()
     print(available_apps)
 
     if available_apps:
@@ -143,14 +144,6 @@ async def main():
         app_details = await registry.get_app_details(app_id)
         print(f"Details for {app_id}:", app_details)
 
-    # The load_tools method is used internally by the high-level Agentr client
-    # but can be called directly if needed.
-    # from universal_mcp.tools import ToolManager
-    # tool_manager = ToolManager()
-    # registry.load_tools(["reddit_search_subreddits"], tool_manager)
-    # print(tool_manager.list_tools())
-
-
 if __name__ == "__main__":
     asyncio.run(main())
 ```
@@ -179,27 +172,30 @@ print(tool_manager.list_tools())
 
 ## Executing Tools
 
-Once tools are loaded, you can execute them using the `call_tool` method on the `ToolManager` instance, which is available via `agentr.manager`.
+Once tools are loaded, you can execute them using the `call_tool` method on the `AgentrRegistry` instance.
 
 ```python
 import os
 import asyncio
-from universal_mcp.agentr import Agentr
+from universal_mcp.agentr import AgentrRegistry
+from universal_mcp.tools import ToolFormat
 
 async def main():
-    # 1. Initialize Agentr
-    agentr = Agentr(api_key=os.environ.get("AGENTR_API_KEY"))
+    # 1. Initialize AgentrRegistry
+    registry = AgentrRegistry(api_key=os.environ.get("AGENTR_API_KEY"))
 
-    # 2. Load the tool(s) you want to use
+    # 2. Load the tool(s) you want to use by exporting them
     tool_name = "reddit__search_subreddits"
-    agentr.load_tools([tool_name])
+    tools_config = {"reddit": ["search_subreddits"]}
+    await registry.export_tools(tools_config, format=ToolFormat.OPENAI)
+
 
-    # 3. Execute the tool using the tool manager
+    # 3. Execute the tool using the registry's call_tool method
     try:
         # Note the 'await' since call_tool is an async method
-        result = await agentr.manager.call_tool(
-            name=tool_name,
-            arguments={"query": "elon musk", "limit": 5, "sort": "relevance"}
+        result = await registry.call_tool(
+            tool_name=tool_name,
+            tool_args={"query": "elon musk", "limit": 5, "sort": "relevance"}
         )
         print("Execution result:", result)
     except Exception as e:
@@ -207,4 +203,4 @@ async def main():
 
 if __name__ == "__main__":
     asyncio.run(main())
-```
+```

universal_mcp/agentr/client.py

@@ -183,7 +183,7 @@ class AgentrClient:
         response.raise_for_status()
         return response.json()
 
-    def search_all_apps(self, query: str, limit: int = 2):
+    def search_all_apps(self, query: str, limit: int = 2, distance_threshold: float = 0.6):
         """Search for apps from the AgentR API.
 
         Args:
@@ -193,11 +193,13 @@ class AgentrClient:
         Returns:
             List[Dict[str, Any]]: A list of app data dictionaries.
         """
-        response = self.client.get("/apps/", params={"search": query, "limit": limit})
+        response = self.client.get(
+            "/apps/", params={"search": query, "limit": limit, "distance_threshold": distance_threshold}
+        )
         response.raise_for_status()
         return response.json().get("items", [])
 
-    def search_all_tools(self, query: str, limit: int = 2, app_id: str | None = None):
+    def search_all_tools(self, query: str, limit: int = 2, app_id: str | None = None, distance_threshold: float = 0.6):
         """Search for tools from the AgentR API.
 
         Args:
@@ -205,7 +207,7 @@ class AgentrClient:
             limit (int, optional): The number of tools to return. Defaults to 2.
             app_id (str, optional): The ID of the app to search tools for.
         """
-        params = {"search": query, "limit": limit}
+        params = {"search": query, "limit": limit, "distance_threshold": distance_threshold}
         if app_id:
             params["app_id"] = app_id
         response = self.client.get("/tools/", params=params)

universal_mcp/agentr/registry.py

@@ -72,7 +72,7 @@ class AgentrRegistry(ToolRegistry):
             Dictionary containing app details
         """
         try:
-            app_info = self.client.get_app_details(app_id)
+            app_info = self.client.get_app_details(app_id=app_id)
             return app_info
         except Exception as e:
             logger.error(f"Error getting details for app {app_id}: {e}")
@@ -82,6 +82,7 @@ class AgentrRegistry(ToolRegistry):
         self,
         query: str,
         limit: int = 10,
+        distance_threshold: float = 0.6,
     ) -> list[dict[str, Any]]:
         """Search for apps by a query.
 
@@ -93,7 +94,7 @@ class AgentrRegistry(ToolRegistry):
             List of app dictionaries matching the query
         """
         try:
-            apps = self.client.search_all_apps(query, limit)
+            apps = self.client.search_all_apps(query=query, limit=limit, distance_threshold=distance_threshold)
             return apps
         except Exception as e:
             logger.error(f"Error searching apps from AgentR: {e}")
@@ -123,6 +124,7 @@ class AgentrRegistry(ToolRegistry):
         query: str,
         limit: int = 2,
         app_id: str | None = None,
+        distance_threshold: float = 0.6,
     ) -> list[dict[str, Any]]:
         """Search for tools by a query.
 
@@ -134,7 +136,9 @@ class AgentrRegistry(ToolRegistry):
             List of tool dictionaries matching the query
         """
         try:
-            tools = self.client.search_all_tools(query, limit, app_id)
+            tools = self.client.search_all_tools(
+                query=query, limit=limit, app_id=app_id, distance_threshold=distance_threshold
+            )
             return tools
         except Exception as e:
             logger.error(f"Error searching tools from AgentR: {e}")
@@ -230,6 +234,6 @@ class AgentrRegistry(ToolRegistry):
         except Exception as e:
             raise e
 
-    async def list_connected_apps(self) -> list[str]:
+    async def list_connected_apps(self) -> list[dict[str, Any]]:
         """List all apps that the user has connected."""
         return self.client.list_my_connections()

universal_mcp/applications/application.py

@@ -11,6 +11,8 @@ from loguru import logger
 
 from universal_mcp.integrations.integration import Integration
 
+DEFAULT_API_TIMEOUT = 30  # seconds
+
 
 class BaseApplication(ABC):
     """Defines the foundational structure for applications in Universal MCP.
@@ -90,7 +92,7 @@ class APIApplication(BaseApplication):
                              BaseApplication.
         """
         super().__init__(name, **kwargs)
-        self.default_timeout: int = 180
+        self.default_timeout: int = DEFAULT_API_TIMEOUT
         self.integration = integration
         logger.debug(f"Initializing APIApplication '{name}' with integration: {integration}")
         self._client: httpx.Client | None = client
@@ -452,6 +454,7 @@ class GraphQLApplication(BaseApplication):
         super().__init__(name, **kwargs)
         self.base_url = base_url
         self.integration = integration
+        self.default_timeout: float = DEFAULT_API_TIMEOUT
         logger.debug(f"Initializing Application '{name}' with kwargs: {kwargs}")
         self._client: GraphQLClient | None = client
 

universal_mcp/tools/local_registry.py

@@ -104,6 +104,6 @@ class LocalRegistry(ToolRegistry):
         result = await tool.run(tool_args)
         return self._handle_file_output(result)
 
-    async def list_connected_apps(self) -> list[str]:
+    async def list_connected_apps(self) -> list[dict[str, Any]]:
         """Not implemented for LocalRegistry."""
         raise NotImplementedError("LocalRegistry does not support listing connected apps.")

universal_mcp/tools/registry.py

@@ -32,7 +32,7 @@ class ToolRegistry(ABC):
         pass
 
     @abstractmethod
-    async def search_apps(self, query: str, limit: int = 2) -> list[dict[str, Any]]:
+    async def search_apps(self, query: str, limit: int = 2, distance_threshold: float = 0.6) -> list[dict[str, Any]]:
         """Search for apps by a query."""
         pass
 
@@ -42,7 +42,9 @@ class ToolRegistry(ABC):
         pass
 
     @abstractmethod
-    async def search_tools(self, query: str, limit: int = 2, app_id: str | None = None) -> list[dict[str, Any]]:
+    async def search_tools(
+        self, query: str, limit: int = 2, app_id: str | None = None, distance_threshold: float = 0.6
+    ) -> list[dict[str, Any]]:
         """Search for tools by a query, optionally filtered by an app."""
         pass
 
@@ -57,7 +59,7 @@ class ToolRegistry(ABC):
         pass
 
     @abstractmethod
-    async def list_connected_apps(self) -> list[str]:
+    async def list_connected_apps(self) -> list[dict[str, Any]]:
         """List all apps that the user has connected."""
         pass
 

{universal_mcp-0.1.24rc21.dist-info → universal_mcp-0.1.24rc23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp
-Version: 0.1.24rc21
+Version: 0.1.24rc23
 Summary: Universal MCP acts as a middle ware for your API applications. It can store your credentials, authorize, enable disable apps on the fly and much more.
 Author-email: Manoj Bajaj <manojbajaj95@gmail.com>
 License: MIT

{universal_mcp-0.1.24rc21.dist-info → universal_mcp-0.1.24rc23.dist-info}/RECORD

@@ -4,13 +4,13 @@ universal_mcp/exceptions.py,sha256=Uen8UFgLHGlSwXgRUyF-nhqTwdiBuL3okgBVRV2AgtA,2
 universal_mcp/logger.py,sha256=VmH_83efpErLEDTJqz55Dp0dioTXfGvMBLZUx5smOLc,2116
 universal_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 universal_mcp/types.py,sha256=DAzYyS7Zqdi38SpnhdOm08_JeN2CJiC0hDIpLLIhlSw,316
-universal_mcp/agentr/README.md,sha256=oF2Ou3czOCLZnRrfcko8xL7zt89j3Im3_wxwpSikz0A,6638
+universal_mcp/agentr/README.md,sha256=ypsA_jD4Fgdg0wQ8bLFIFPmpFwYCCCv-29pNizJDCk4,6522
 universal_mcp/agentr/__init__.py,sha256=fv1ZnOCduIUiJ9oN4e6Ya_hA2oWQvcEuDU3Ek1vEufI,180
-universal_mcp/agentr/client.py,sha256=CwfmOAgWsApm0buxZPni3vW6MFgrwdddXsXmQA5UYrA,7882
+universal_mcp/agentr/client.py,sha256=og1kfnOxtSTVtgYXg4AMcitX0nSX0t_r5D7A_HrcAG8,8054
 universal_mcp/agentr/integration.py,sha256=V5GjqocqS02tRoI8MeV9PL6m-BzejwBzgJhOHo4MxAE,4212
-universal_mcp/agentr/registry.py,sha256=VrB39dq3HMOrunPa9ruB8YbNyu1-18BtLaBUio_gCGk,8722
+universal_mcp/agentr/registry.py,sha256=KBChYv7QkT9J7i-Xks9AN4vc_8NGQCI5hqMRRXdKYOc,8961
 universal_mcp/agentr/server.py,sha256=d_UaxCGTOzdOsIMiHiILgAjeZLbUASCyQkUqhd1x9wA,1911
-universal_mcp/applications/application.py,sha256=do45GC0jnNepZHL8US2yDNq9s0ZnE6bII4Xbw90GRTc,23727
+universal_mcp/applications/application.py,sha256=Zxgrr0LVNUTnKxqnX11ScTd5IWEUPCsNYZO2URL6mbQ,23838
 universal_mcp/applications/utils.py,sha256=8Pp9lZU6IPt9z9BnuJ-vpv-NGuzryt1c4e4-ShDd2XI,1450
 universal_mcp/applications/sample/app.py,sha256=D9zPezC13xXVMlfO2A0fHgJQD_I-bnpf9UOOHveHMek,10537
 universal_mcp/client/oauth.py,sha256=O00zOUfQxINaruFU2zt-64DIR1_mAqrY8ykLQo-teJU,8679
@@ -26,9 +26,9 @@ universal_mcp/tools/__init__.py,sha256=jC8hsqfTdtn32yU57AVFUXiU3ZmUOCfCERSCaNEIH
 universal_mcp/tools/adapters.py,sha256=LmI5J6e-3TPVuyeDki7uHeZVeCES52y7TRcLpUurzNQ,4591
 universal_mcp/tools/docstring_parser.py,sha256=efEOE-ME7G5Jbbzpn7pN2xNuyu2M5zfZ1Tqu1lRB0Gk,8392
 universal_mcp/tools/func_metadata.py,sha256=F4jd--hoZWKPBbZihVtluYKUsIdXdq4a0VWRgMl5k-Q,10838
-universal_mcp/tools/local_registry.py,sha256=lvvpW_reF2lHSr3_s3txRU9b5Ps8rUwH-cZSApw46Ew,4236
+universal_mcp/tools/local_registry.py,sha256=Gp3bXwLwJ1SBzS1ZSDXa5pHU6kapwAYx5lU41NfRKx4,4247
 universal_mcp/tools/manager.py,sha256=73x5RB5Sgew1hNqCOMmoxHZN2DYjiE7pjsytfi0Gj0g,8213
-universal_mcp/tools/registry.py,sha256=PTVDeVIdHDxMHlf5cNuBmvbmSTFBt_YencA6e79DH4s,3868
+universal_mcp/tools/registry.py,sha256=5gXIcLy_ggCMF8RZ3gRyvqS6S9ERS6Vytp_iXdyUcr8,3959
 universal_mcp/tools/tools.py,sha256=Lk-wUO3rfhwdxaRANtC7lQr5fXi7nclf0oHzxNAb79Q,4927
 universal_mcp/utils/__init__.py,sha256=8wi4PGWu-SrFjNJ8U7fr2iFJ1ktqlDmSKj1xYd7KSDc,41
 universal_mcp/utils/installation.py,sha256=PU_GfHPqzkumKk-xG4L9CkBzSmABxmchwblZkx-zY-I,7204
@@ -48,8 +48,8 @@ universal_mcp/utils/openapi/readme.py,sha256=R2Jp7DUXYNsXPDV6eFTkLiy7MXbSULUj1vH
 universal_mcp/utils/openapi/test_generator.py,sha256=vucBh9klWmQOUA740TFwfM9ry2nkwKWQiNRcsiZ9HbY,12229
 universal_mcp/utils/templates/README.md.j2,sha256=Mrm181YX-o_-WEfKs01Bi2RJy43rBiq2j6fTtbWgbTA,401
 universal_mcp/utils/templates/api_client.py.j2,sha256=DS1nczOOD8YkMexVSGpUGeyc0nYGKKTPadL_x1_if7k,900
-universal_mcp-0.1.24rc21.dist-info/METADATA,sha256=uMU2UHEiXuJZSI4-3DhheHEjz_IImx2YTN13upi82d8,3255
-universal_mcp-0.1.24rc21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-universal_mcp-0.1.24rc21.dist-info/entry_points.txt,sha256=QlBrVKmA2jIM0q-C-3TQMNJTTWOsOFQvgedBq2rZTS8,56
-universal_mcp-0.1.24rc21.dist-info/licenses/LICENSE,sha256=NweDZVPslBAZFzlgByF158b85GR0f5_tLQgq1NS48To,1063
-universal_mcp-0.1.24rc21.dist-info/RECORD,,
+universal_mcp-0.1.24rc23.dist-info/METADATA,sha256=qZ4Rn4X_q_pg17d70czju8sIFdwho5diOPgC6XmhVeg,3255
+universal_mcp-0.1.24rc23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+universal_mcp-0.1.24rc23.dist-info/entry_points.txt,sha256=QlBrVKmA2jIM0q-C-3TQMNJTTWOsOFQvgedBq2rZTS8,56
+universal_mcp-0.1.24rc23.dist-info/licenses/LICENSE,sha256=NweDZVPslBAZFzlgByF158b85GR0f5_tLQgq1NS48To,1063
+universal_mcp-0.1.24rc23.dist-info/RECORD,,