universal-mcp 0.1.6__py3-none-any.whl → 0.1.7rc1__py3-none-any.whl

universal_mcp/applications/e2b/app.py

@@ -1,6 +1,6 @@
 from e2b_code_interpreter import Sandbox
 from loguru import logger
-
+from typing import Annotated
 from universal_mcp.applications.application import APIApplication
 from universal_mcp.integrations import Integration
 
@@ -50,7 +50,9 @@ class E2BApp(APIApplication):
             return "Execution finished with no output (stdout/stderr)."
         return "\n\n".join(output_parts)
 
-    def execute_python_code(self, code: str) -> str:
+    def execute_python_code(
+        self, code: Annotated[str, "The Python code to execute."]
+    ) -> str:
         """
         Executes Python code within a secure E2B cloud sandbox.
 
@@ -61,6 +63,9 @@ class E2BApp(APIApplication):
             A string containing the formatted standard output (stdout) and standard error (stderr)
             from the execution. If an error occurs during setup or execution, an
             error message string is returned.
+
+        Raises:
+            NotAuthorizedError: If the API key is not set.
         """
         self._set_api_key()
         with Sandbox(api_key=self.api_key) as sandbox:

universal_mcp/applications/perplexity/app.py

@@ -2,6 +2,7 @@ from typing import Any, Literal
 
 from universal_mcp.applications.application import APIApplication
 from universal_mcp.integrations import Integration
+from loguru import logger
 
 
 class PerplexityApp(APIApplication):
@@ -18,26 +19,40 @@ class PerplexityApp(APIApplication):
             raise ValueError("Integration is None. Cannot retrieve Perplexity API Key.")
 
         credentials = self.integration.get_credentials()
-        if not credentials:
-             raise ValueError(
+        if not credentials or "apiKey" not in credentials:
+            raise ValueError(
                 f"Failed to retrieve Perplexity API Key using integration '{self.integration.name}'. "
-             )
-        
-        if not isinstance(credentials, str) or not credentials.strip():
-             raise ValueError(
-                f"Invalid credential format received for Perplexity API Key via integration '{self.integration.name}'. "
             )
-        self.api_key = credentials
+
+        # if not isinstance(credentials, str) or not credentials.strip():
+        #     raise ValueError(
+        #         f"Invalid credential format received for Perplexity API Key via integration '{self.integration.name}'. "
+        #     )
+        self.api_key = credentials["apiKey"]
 
     def _get_headers(self) -> dict[str, str]:
         self._set_api_key()
+        logger.info(f"Perplexity API Key: {self.api_key}")
         return {
             "Authorization": f"Bearer {self.api_key}",
             "Content-Type": "application/json",
             "Accept": "application/json",
         }
 
-    def chat(self, query: str, model: Literal["r1-1776","sonar","sonar-pro","sonar-reasoning","sonar-reasoning-pro", "sonar-deep-research"] = "sonar" , temperature: float = 1, system_prompt: str = "Be precise and concise.") -> dict[str, Any] | str:
+    def chat(
+        self,
+        query: str,
+        model: Literal[
+            "r1-1776",
+            "sonar",
+            "sonar-pro",
+            "sonar-reasoning",
+            "sonar-reasoning-pro",
+            "sonar-deep-research",
+        ] = "sonar",
+        temperature: float = 1,
+        system_prompt: str = "Be precise and concise.",
+    ) -> dict[str, Any] | str:
         """
         Sends a query to a Perplexity Sonar online model and returns the response.
 
@@ -69,11 +84,11 @@ class PerplexityApp(APIApplication):
 
         data = self._post(endpoint, data=payload)
         response = data.json()
-        content = response['choices'][0]['message']['content']
-        citations = response.get('citations', [])
+        content = response["choices"][0]["message"]["content"]
+        citations = response.get("citations", [])
         return {"content": content, "citations": citations}
 
     def list_tools(self):
         return [
             self.chat,
-        ]
+        ]

universal_mcp/config.py

@@ -10,7 +10,7 @@ class StoreConfig(BaseModel):
 
 class IntegrationConfig(BaseModel):
     name: str
-    type: Literal["api_key", "oauth", "agentr"]
+    type: Literal["api_key", "oauth", "agentr", "oauth2"]
     credentials: dict | None = None
     store: StoreConfig | None = None
 

universal_mcp/servers/server.py

@@ -115,7 +115,11 @@ class LocalServer(Server):
                     tool_name = tool.__name__
                     name = app.name + "_" + tool_name
                     description = tool.__doc__
-                    if app_config.actions is None or tool_name in app_config.actions:
+                    if (
+                        app_config.actions is None
+                        or len(app_config.actions) == 0
+                        or name in app_config.actions
+                    ):
                         self.add_tool(tool, name=name, description=description)
             except Exception as e:
                 logger.error(f"Error loading app {app_config.name}: {e}")
@@ -167,7 +171,11 @@ class AgentRServer(Server):
                     tool_name = tool.__name__
                     name = app.name + "_" + tool_name
                     description = tool.__doc__
-                    if app_config.actions is None or tool_name in app_config.actions:
+                    if (
+                        app_config.actions is None
+                        or len(app_config.actions) == 0
+                        or name in app_config.actions
+                    ):
                         self.add_tool(tool, name=name, description=description)
             except Exception as e:
                 logger.error(f"Error loading app {app_config.name}: {e}")

{universal_mcp-0.1.6.dist-info → universal_mcp-0.1.7rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp
-Version: 0.1.6
+Version: 0.1.7rc1
 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>
 Requires-Python: >=3.11

{universal_mcp-0.1.6.dist-info → universal_mcp-0.1.7rc1.dist-info}/RECORD

@@ -1,13 +1,13 @@
 universal_mcp/__init__.py,sha256=2gdHpHaDDcsRjZjJ01FLN-1iidN_wbDAolNpxhGoFB4,59
 universal_mcp/cli.py,sha256=DG-Qxc5vQIdbhAIQuU7bKKJuRGzwyOigjfCKSWBRhBI,5258
-universal_mcp/config.py,sha256=9eb3DDg4PBBr1MlGeBrA4bja3Y6howOH-UKpo7JIbs8,828
+universal_mcp/config.py,sha256=sJaPI4q51CDPPG0z32rMJiE7a64eaa9nxbjJgYnaFA4,838
 universal_mcp/exceptions.py,sha256=Zp2_v_m3L7GDAmD1ZyuwFtY6ngapdhxuIygrvpZAQtM,271
 universal_mcp/logger.py,sha256=W6A868vyvpdkEQ4Dd0rWdC_7ErSxSQ1z2uxCb77IM8Y,2015
 universal_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 universal_mcp/applications/__init__.py,sha256=qeWnbdIudyMR7ST4XTc0gpEM9o6TsM1ZnZ92dMAPSBA,754
 universal_mcp/applications/application.py,sha256=dqp8lgIi2xhY62imwo7C6769URQtNmqd6Ok6PiTr6wc,3399
 universal_mcp/applications/e2b/README.md,sha256=S4lTp-vEZ8VTCKPXqjUXu5nYlUMAF8lw8CQyBGPgxjs,700
-universal_mcp/applications/e2b/app.py,sha256=5piCipi1TC_KuKLFP17do1Y1r28fqApvRsZy76equ9w,2573
+universal_mcp/applications/e2b/app.py,sha256=dC6TFhN9NYzZqmJno47ZCU01Un7M3HlqNxuXx7UEOso,2733
 universal_mcp/applications/firecrawl/README.md,sha256=KAWe_TQbrc9eA6bSyde5dapMP1CNvarVItV_YJH3d_0,1430
 universal_mcp/applications/firecrawl/app.py,sha256=RSy8zRn4k1A1tIpJNqrUnPI8ctEv1nKWuOuJQcp9mGo,9264
 universal_mcp/applications/github/README.md,sha256=6ID-__gUJ5ZxzAS_OjzmoUAag1LamSvEB75DHcj3m-g,1294
@@ -26,7 +26,7 @@ universal_mcp/applications/notion/README.md,sha256=45NmPOmSQv99qBvWdwmnV5vbaYc9_
 universal_mcp/applications/notion/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 universal_mcp/applications/notion/app.py,sha256=XpLnmeXj0Gnf_RYHlbAFnwtSCTYsrNzk6MSMSyDmHGQ,17283
 universal_mcp/applications/perplexity/README.md,sha256=QGV1iReH5p-Np7vvkZsVHxxDKQ0YaitHEwomNmGEyQs,732
-universal_mcp/applications/perplexity/app.py,sha256=J27IDcz9pRC_uW4wABpn-EcI4RvKIzoh_o2hzysM5wA,3187
+universal_mcp/applications/perplexity/app.py,sha256=81cR_UUyBlAzonq0GsSuV6w2XxLsKIiMelJDc8h5Fxw,3444
 universal_mcp/applications/reddit/README.md,sha256=YVbJ1RN6NWlB-P6w2LxCk_DuUWl7mwaKZScY-mIMnNc,1271
 universal_mcp/applications/reddit/app.py,sha256=leU__w5VxX1vMK-kfuy-dvY97Pn8Mn80X2payVshirU,13562
 universal_mcp/applications/resend/README.md,sha256=k-sb2UwbFvDPEz6qQPLWd2cJj8hDx5f3NW7dz2jAfjI,719
@@ -42,7 +42,7 @@ universal_mcp/integrations/__init__.py,sha256=8e11JZyctaR9CmlNkfEZ6HhGDvhlvf9iug
 universal_mcp/integrations/agentr.py,sha256=l0mo79oeDML19udFfoCo9lyhbDAf0X94_lnpOgbTrb0,3331
 universal_mcp/integrations/integration.py,sha256=8TYr7N1F6oV8PPOSLo0eA_6nD-vrwIWbuplqwD03uuQ,5819
 universal_mcp/servers/__init__.py,sha256=dgRW_khG537GeLKC5_U5jhxCuu1L_1YeTujeDg0601E,654
-universal_mcp/servers/server.py,sha256=uHArnw_kYIhvlE-Zp5GharGqapGy4jaOlxRQGsOafPM,6431
+universal_mcp/servers/server.py,sha256=9GazpAzNiMgCMg4Vb4VeyoviddEt6pQzi7vLli8vC5A,6677
 universal_mcp/stores/__init__.py,sha256=Sc4AWtee_qtK5hpEVUAH2XM_6EBhcfikQXWiGXdNfes,560
 universal_mcp/stores/store.py,sha256=CNOnmKeOCkSU2ZA9t12AIWJcmqZZX_LSyZaV8FQf8Xk,4545
 universal_mcp/utils/__init__.py,sha256=8wi4PGWu-SrFjNJ8U7fr2iFJ1ktqlDmSKj1xYd7KSDc,41
@@ -52,7 +52,7 @@ universal_mcp/utils/docgen.py,sha256=yK6Ijo8G-wHPU3E1AnFpnXS9vXt2j9FM77w0etTaNOA
 universal_mcp/utils/dump_app_tools.py,sha256=cLB9SumKsbs-rXJ_02lpMyyNkOmKZ57gekhCjhAlcHg,2009
 universal_mcp/utils/installation.py,sha256=uSL_H76fG_7yN4QNxkfp1mEF_00iAPyiXqtdWEMVJe8,3747
 universal_mcp/utils/openapi.py,sha256=ud_ZB7_60BcS1Vao7ESKDqo0gry9JN5wzy-CFssrjm8,13140
-universal_mcp-0.1.6.dist-info/METADATA,sha256=8R7LIzbEuCr2cB6yaLxZG8HD9ISZz_H0jJqquPXSiaw,10852
-universal_mcp-0.1.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-universal_mcp-0.1.6.dist-info/entry_points.txt,sha256=QlBrVKmA2jIM0q-C-3TQMNJTTWOsOFQvgedBq2rZTS8,56
-universal_mcp-0.1.6.dist-info/RECORD,,
+universal_mcp-0.1.7rc1.dist-info/METADATA,sha256=mwq5fSI-AWh4t7sJXfgooEnhI19wmVeD75uPjutsboI,10855
+universal_mcp-0.1.7rc1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+universal_mcp-0.1.7rc1.dist-info/entry_points.txt,sha256=QlBrVKmA2jIM0q-C-3TQMNJTTWOsOFQvgedBq2rZTS8,56
+universal_mcp-0.1.7rc1.dist-info/RECORD,,