agentr 0.1.6__py3-none-any.whl → 0.1.7__py3-none-any.whl

agentr/utils/openapi.py

@@ -1,185 +1,186 @@
-import json
-import yaml
-from pathlib import Path
-
-def load_schema(path: Path):
-    if path.suffix == '.yaml':
-        type = 'yaml'
-    else:
-        type = 'json'
-    with open(path, 'r') as f:
-        if type == 'yaml':
-            return yaml.safe_load(f)
-        else:
-            return json.load(f)
-
-def generate_api_client(schema):
-    """
-    Generate a Python API client class from an OpenAPI schema.
-    
-    Args:
-        schema (dict): The OpenAPI schema as a dictionary.
-    
-    Returns:
-        str: A string containing the Python code for the API client class.
-    """
-    methods = []
-    
-    # Iterate over paths and their operations
-    for path, path_info in schema.get('paths', {}).items():
-        for method in path_info:
-            if method in ['get', 'post', 'put', 'delete', 'patch', 'options', 'head']:
-                operation = path_info[method]
-                method_code = generate_method_code(path, method, operation)
-                methods.append(method_code)
-    
-    # Construct the class code
-    class_code = (
-        "import requests\n\n"
-        "class APIClient:\n"
-        "    def __init__(self, base_url):\n"
-        "        self.base_url = base_url\n\n" +
-        '\n\n'.join(methods)
-    )
-    return class_code
-
-def generate_method_code(path, method, operation):
-    """
-    Generate the code for a single API method.
-    
-    Args:
-        path (str): The API path (e.g., '/users/{user_id}').
-        method (str): The HTTP method (e.g., 'get').
-        operation (dict): The operation details from the schema.
-    
-    Returns:
-        str: The Python code for the method.
-    """
-    # Determine function name
-    if 'operationId' in operation:
-        func_name = operation['operationId']
-    else:
-        # Generate name from path and method
-        path_parts = path.strip('/').split('/')
-        name_parts = [method]
-        for part in path_parts:
-            if part.startswith('{') and part.endswith('}'):
-                name_parts.append('by_' + part[1:-1])
-            else:
-                name_parts.append(part)
-        func_name = '_'.join(name_parts).replace('-', '_').lower()
-    
-    # Get parameters and request body
-    parameters = operation.get('parameters', [])
-    has_body = 'requestBody' in operation
-    body_required = has_body and operation['requestBody'].get('required', False)
-    
-    # Build function arguments
-    args = []
-    for param in parameters:
-        if param.get('required', False):
-            args.append(param['name'])
-        else:
-            args.append(f"{param['name']}=None")
-    if has_body:
-        args.append('body' if body_required else 'body=None')
-    signature = f"def {func_name}(self, {', '.join(args)}):"
-    
-    # Build method body
-    body_lines = []
-    
-    # Path parameters
-    path_params = [p for p in parameters if p['in'] == 'path']
-    path_params_dict = ', '.join([f"'{p['name']}': {p['name']}" for p in path_params])
-    body_lines.append(f"    path_params = {{{path_params_dict}}}")
-    
-    # Query parameters
-    query_params = [p for p in parameters if p['in'] == 'query']
-    query_params_items = ', '.join([f"('{p['name']}', {p['name']})" for p in query_params])
-    body_lines.append(
-        f"    query_params = {{k: v for k, v in [{query_params_items}] if v is not None}}"
-    )
-    
-    # Format URL
-    body_lines.append(f"    url = f\"{{self.base_url}}{path}\".format_map(path_params)")
-    
-    # Make HTTP request
-    method_func = method.lower()
-    if has_body:
-        body_lines.append("    if body is not None:")
-        body_lines.append(f"        response = requests.{method_func}(url, params=query_params, json=body)")
-        body_lines.append("    else:")
-        body_lines.append(f"        response = requests.{method_func}(url, params=query_params)")
-    else:
-        body_lines.append(f"    response = requests.{method_func}(url, params=query_params)")
-    
-    # Handle response
-    body_lines.append("    response.raise_for_status()")
-    body_lines.append("    return response.json()")
-    
-    return signature + '\n' + '\n'.join(body_lines)
-
-# Example usage
-if __name__ == "__main__":
-    # Sample OpenAPI schema
-    schema = {
-        "paths": {
-            "/users": {
-                "get": {
-                    "summary": "Get a list of users",
-                    "parameters": [
-                        {
-                            "name": "limit",
-                            "in": "query",
-                            "required": False,
-                            "schema": {"type": "integer"}
-                        }
-                    ],
-                    "responses": {
-                        "200": {
-                            "description": "A list of users",
-                            "content": {"application/json": {"schema": {"type": "array"}}}
-                        }
-                    }
-                },
-                "post": {
-                    "summary": "Create a user",
-                    "requestBody": {
-                        "required": True,
-                        "content": {
-                            "application/json": {
-                                "schema": {
-                                    "type": "object",
-                                    "properties": {"name": {"type": "string"}}
-                                }
-                            }
-                        }
-                    },
-                    "responses": {
-                        "201": {"description": "User created"}
-                    }
-                }
-            },
-            "/users/{user_id}": {
-                "get": {
-                    "summary": "Get a user by ID",
-                    "parameters": [
-                        {
-                            "name": "user_id",
-                            "in": "path",
-                            "required": True,
-                            "schema": {"type": "string"}
-                        }
-                    ],
-                    "responses": {
-                        "200": {"description": "User details"}
-                    }
-                }
-            }
-        }
-    }
-    
-
-    schema = load_schema('openapi.yaml')
-    code = generate_api_client(schema)
+import json
+import yaml
+from pathlib import Path
+
+
+def load_schema(path: Path):
+    if path.suffix == '.yaml':
+        type = 'yaml'
+    else:
+        type = 'json'
+    with open(path, 'r') as f:
+        if type == 'yaml':
+            return yaml.safe_load(f)
+        else:
+            return json.load(f)
+
+def generate_api_client(schema):
+    """
+    Generate a Python API client class from an OpenAPI schema.
+    
+    Args:
+        schema (dict): The OpenAPI schema as a dictionary.
+    
+    Returns:
+        str: A string containing the Python code for the API client class.
+    """
+    methods = []
+    
+    # Iterate over paths and their operations
+    for path, path_info in schema.get('paths', {}).items():
+        for method in path_info:
+            if method in ['get', 'post', 'put', 'delete', 'patch', 'options', 'head']:
+                operation = path_info[method]
+                method_code = generate_method_code(path, method, operation)
+                methods.append(method_code)
+    
+    # Construct the class code
+    class_code = (
+        "import requests\n\n"
+        "class APIClient:\n"
+        "    def __init__(self, base_url):\n"
+        "        self.base_url = base_url\n\n" +
+        '\n\n'.join(methods)
+    )
+    return class_code
+
+def generate_method_code(path, method, operation):
+    """
+    Generate the code for a single API method.
+    
+    Args:
+        path (str): The API path (e.g., '/users/{user_id}').
+        method (str): The HTTP method (e.g., 'get').
+        operation (dict): The operation details from the schema.
+    
+    Returns:
+        str: The Python code for the method.
+    """
+    # Determine function name
+    if 'operationId' in operation:
+        func_name = operation['operationId']
+    else:
+        # Generate name from path and method
+        path_parts = path.strip('/').split('/')
+        name_parts = [method]
+        for part in path_parts:
+            if part.startswith('{') and part.endswith('}'):
+                name_parts.append('by_' + part[1:-1])
+            else:
+                name_parts.append(part)
+        func_name = '_'.join(name_parts).replace('-', '_').lower()
+    
+    # Get parameters and request body
+    parameters = operation.get('parameters', [])
+    has_body = 'requestBody' in operation
+    body_required = has_body and operation['requestBody'].get('required', False)
+    
+    # Build function arguments
+    args = []
+    for param in parameters:
+        if param.get('required', False):
+            args.append(param['name'])
+        else:
+            args.append(f"{param['name']}=None")
+    if has_body:
+        args.append('body' if body_required else 'body=None')
+    signature = f"def {func_name}(self, {', '.join(args)}):"
+    
+    # Build method body
+    body_lines = []
+    
+    # Path parameters
+    path_params = [p for p in parameters if p['in'] == 'path']
+    path_params_dict = ', '.join([f"'{p['name']}': {p['name']}" for p in path_params])
+    body_lines.append(f"    path_params = {{{path_params_dict}}}")
+    
+    # Query parameters
+    query_params = [p for p in parameters if p['in'] == 'query']
+    query_params_items = ', '.join([f"('{p['name']}', {p['name']})" for p in query_params])
+    body_lines.append(
+        f"    query_params = {{k: v for k, v in [{query_params_items}] if v is not None}}"
+    )
+    
+    # Format URL
+    body_lines.append(f"    url = f\"{{self.base_url}}{path}\".format_map(path_params)")
+    
+    # Make HTTP request
+    method_func = method.lower()
+    if has_body:
+        body_lines.append("    if body is not None:")
+        body_lines.append(f"        response = requests.{method_func}(url, params=query_params, json=body)")
+        body_lines.append("    else:")
+        body_lines.append(f"        response = requests.{method_func}(url, params=query_params)")
+    else:
+        body_lines.append(f"    response = requests.{method_func}(url, params=query_params)")
+    
+    # Handle response
+    body_lines.append("    response.raise_for_status()")
+    body_lines.append("    return response.json()")
+    
+    return signature + '\n' + '\n'.join(body_lines)
+
+# Example usage
+if __name__ == "__main__":
+    # Sample OpenAPI schema
+    schema = {
+        "paths": {
+            "/users": {
+                "get": {
+                    "summary": "Get a list of users",
+                    "parameters": [
+                        {
+                            "name": "limit",
+                            "in": "query",
+                            "required": False,
+                            "schema": {"type": "integer"}
+                        }
+                    ],
+                    "responses": {
+                        "200": {
+                            "description": "A list of users",
+                            "content": {"application/json": {"schema": {"type": "array"}}}
+                        }
+                    }
+                },
+                "post": {
+                    "summary": "Create a user",
+                    "requestBody": {
+                        "required": True,
+                        "content": {
+                            "application/json": {
+                                "schema": {
+                                    "type": "object",
+                                    "properties": {"name": {"type": "string"}}
+                                }
+                            }
+                        }
+                    },
+                    "responses": {
+                        "201": {"description": "User created"}
+                    }
+                }
+            },
+            "/users/{user_id}": {
+                "get": {
+                    "summary": "Get a user by ID",
+                    "parameters": [
+                        {
+                            "name": "user_id",
+                            "in": "path",
+                            "required": True,
+                            "schema": {"type": "string"}
+                        }
+                    ],
+                    "responses": {
+                        "200": {"description": "User details"}
+                    }
+                }
+            }
+        }
+    }
+    
+
+    schema = load_schema('openapi.yaml')
+    code = generate_api_client(schema)
     print(code)

{agentr-0.1.6.dist-info → agentr-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentr
-Version: 0.1.6
+Version: 0.1.7
 Summary: A python framework to build MCP servers
 Author-email: Manoj Bajaj <manojbajaj95@gmail.com>
 License-File: LICENSE
@@ -9,6 +9,9 @@ Requires-Dist: loguru>=0.7.3
 Requires-Dist: mcp>=1.5.0
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: typer>=0.15.2
+Provides-Extra: test
+Requires-Dist: pytest-asyncio>=0.26.0; extra == 'test'
+Requires-Dist: pytest>=8.3.5; extra == 'test'
 Description-Content-Type: text/markdown
 
 # AgentR

agentr-0.1.7.dist-info/RECORD

@@ -0,0 +1,25 @@
+agentr/__init__.py,sha256=LOWhgQayrQV7f5ro4rlBJ_6WevhbWIbjAOHnqP7b_-4,30
+agentr/application.py,sha256=r2o4PP0uDBFTkziseuOh7AobcVqnEWDoYJxN_UA7UVo,2932
+agentr/cli.py,sha256=z0QOdyiBcypNE_npN2lYIkEcYZXG8Ji0027IcEuhDTs,2880
+agentr/config.py,sha256=YTygfFJPUuL-epRuILvt5tc5ACzByWIFFNhpFwHlDCE,387
+agentr/exceptions.py,sha256=hHlyXUZBjG4DfUurvqd0ZiruHC67gbpT6EHKxifwUhg,271
+agentr/integration.py,sha256=BmtPpn2lXV7i6k6Uqu5g-LWpD9wRiMz7R3SgztLAktc,2663
+agentr/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+agentr/server.py,sha256=vg9YlApxlLvxcIn9h_Rz2IgatXv7aBj0pTTb6XWMRN4,4279
+agentr/store.py,sha256=fB3uAaobnWf2ILcDBmg3ToDaqAIPYlLtmHBdpmkcGcI,1585
+agentr/test.py,sha256=wtupsEMrn-Q9Ocjlhelk4YrVSrWMM5vwMBhnvepEaHo,917
+agentr/applications/__init__.py,sha256=huqhhfMkSMjcc3eqVAprW03Drr98OHbH2Rh0GwGQHjs,942
+agentr/applications/github/app.py,sha256=N-f9TX5btcUDr19H-eTG7NlFlb25fGENLZ07fFfQBg4,17301
+agentr/applications/google_calendar/app.py,sha256=3mpqP0MsTYsJyZDF2Skgs1VLfLRr6E7OSOhbcAQ_L3E,25710
+agentr/applications/google_mail/app.py,sha256=GWmJwgdpBKyNufsrHp2PkYzXNzyCb7XOHxZPYw6XaBA,23710
+agentr/applications/reddit/app.py,sha256=iNkSRVGScj7Mae3e8N-Mkov_hvZrvH5NynbpO8mKToQ,842
+agentr/applications/resend/app.py,sha256=ihRzP65bwoNIq3EzBqIghxgLZRxvy-LbHy-rER20ODo,1428
+agentr/applications/tavily/app.py,sha256=D4FOhm2yxNbuTVHTo3T8ZsuE5AgwK874YutfYx2Njcw,1805
+agentr/applications/zenquotes/app.py,sha256=4LjYeWdERI8ZMzkajOsDgy9IRTA9plUKem-rW4M03sA,607
+agentr/utils/bridge.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+agentr/utils/openapi.py,sha256=DT4jcWue_Fx0J-wg7T53w3uMb5CIzn4u2QswmtpSFVU,6280
+agentr-0.1.7.dist-info/METADATA,sha256=BUoooDf1WgcRVE-rVBD3W78W-FrhSGnvAmhbYbuV-TY,4388
+agentr-0.1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agentr-0.1.7.dist-info/entry_points.txt,sha256=13fGFeVhgF6_8T-VFiIkNxYO7gDQaUwwTcUNWdvaLQg,42
+agentr-0.1.7.dist-info/licenses/LICENSE,sha256=CPslwL9mT3MH-lEljRJQHKe646096G-szURVmOD18Lc,1063
+agentr-0.1.7.dist-info/RECORD,,

{agentr-0.1.6.dist-info → agentr-0.1.7.dist-info}/licenses/LICENSE

@@ -1,21 +1,21 @@
-MIT License
-
-Copyright (c) 2025 AgentR
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+MIT License
+
+Copyright (c) 2025 AgentR
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

agentr-0.1.6.dist-info/RECORD

@@ -1,23 +0,0 @@
-agentr/__init__.py,sha256=oESTO4raGTykM5npHqhuwinRY-zxPtdmeOzXV-se_Bo,31
-agentr/application.py,sha256=jDcYk8DArOJuE-HEn6_aCUtMn_ckrFpJuoAv4qzyr6c,1607
-agentr/cli.py,sha256=UndJFUcJ7MCZzlhQB3W70j1oi1a-aYb1pZyKODNtdX8,2872
-agentr/exceptions.py,sha256=7VusZ_ltcYK3pRcew_Kc8rSllQoxi2RvmxdntJgufqM,275
-agentr/integration.py,sha256=Btd7VwKb1T94s9ZKApMDAEFv7tlK4oxHfBvIYRAmUe8,3657
-agentr/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentr/server.py,sha256=WB98K-BrSt77pp_wGTUnJa7Sjeyj_hFbbaB3hS9qnRA,4325
-agentr/store.py,sha256=IlCGsYjFIzVLsBnse5cuzS4egOn7osPr3zSUt3FwPvA,1655
-agentr/test.py,sha256=ijzm58aoCSJDAN-ObK2HludNMr2bbvPMURC7ah8wcu4,920
-agentr/applications/github/app.py,sha256=7ZOPDq95IaYza7IhNkB4zkeZmGDRxffhtRju6asUwLg,2005
-agentr/applications/google_calendar/app.py,sha256=9PFV_p0KbyuJslcaFnhDmqR6p3G7BR-3Ij5KIaPomTc,2914
-agentr/applications/google_mail/app.py,sha256=Pf_u0HE1Im7J9xe77RbknfaOmnspHRrKY40CpL270zU,2401
-agentr/applications/reddit/app.py,sha256=NLBPS0sAqUt5EVHeqcnAn4s6XD-KoapJjlLhSnsUgaY,871
-agentr/applications/resend/app.py,sha256=FWomwUKrrOHMptXO0P1wbe40JDyBvq_CC7Hrk0qcyWE,1471
-agentr/applications/tavily/app.py,sha256=WgnEqD0bROiDpliGEUSTww4vH5drRsHTBBGf8WRZacU,1862
-agentr/applications/zenquotes/app.py,sha256=WRK96H8cTWXRV-Z42QnOa1qfqzs6xmYTpxj1MJooYZM,627
-agentr/utils/bridge.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentr/utils/openapi.py,sha256=S9MyxrxAs87-WTYgzfKVWhroxIB98_nBG1Ksac0x_pU,6463
-agentr-0.1.6.dist-info/METADATA,sha256=i9BGPVoqy0tnhkOdG49N8GGMUq7awTTMoc8GPj3WeH4,4266
-agentr-0.1.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agentr-0.1.6.dist-info/entry_points.txt,sha256=13fGFeVhgF6_8T-VFiIkNxYO7gDQaUwwTcUNWdvaLQg,42
-agentr-0.1.6.dist-info/licenses/LICENSE,sha256=vXWvmZhTURz0cuRxzRypSABVSncu9X8-XhwdC1yo6J0,1084
-agentr-0.1.6.dist-info/RECORD,,