agentr 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

agentr/utils/openapi.py

@@ -1,185 +1,185 @@
-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.4.dist-info/METADATA

@@ -0,0 +1,156 @@
+Metadata-Version: 2.4
+Name: agentr
+Version: 0.1.4
+Summary: Add your description here
+Author-email: Manoj Bajaj <manojbajaj95@gmail.com>
+Requires-Python: >=3.11
+Requires-Dist: loguru>=0.7.3
+Requires-Dist: mcp>=1.5.0
+Requires-Dist: pyyaml>=6.0.2
+Requires-Dist: typer>=0.15.2
+Description-Content-Type: text/markdown
+
+# AgentR
+
+AgentR is a Python framework for building powerful agent applications with integrated third-party services. It provides a clean abstraction for incorporating various API services as "applications" into your agent workflow.
+
+## 🌟 Features
+
+- **Simplified API Integration**: Connect to services like GitHub, Google Calendar, Gmail, Reddit, Tavily, and more with minimal code
+- **Authentication Handling**: Built-in support for API keys and OAuth-based authentication flows
+- **Extensible Architecture**: Easily build and add new app integrations with minimal boilerplate
+- **MCP (Model Context Protocol) Integration**: Seamlessly works with MCP server architecture
+- **Credential Management**: Flexible storage options for API credentials with memory and environment-based implementations
+
+## 🔧 Installation
+
+Install AgentR using pip:
+
+```bash
+pip install agentr
+```
+
+## 🚀 Quick Start
+
+### 1. Get an API Key
+Before using AgentR with services that require authorization (like GitHub, Gmail, etc.), you'll need an AgentR API key:
+
+Visit https://agentr.dev to create an account
+Generate an API key from your dashboard
+Set it as an environment variable or include it directly in your code:
+
+```bash
+export AGENTR_API_KEY="your_api_key_here"
+```
+
+### 2. Create a basic server
+
+```bash
+from agentr.server import TestServer
+
+# Define your applications list
+apps_list = [
+    {
+        "name": "tavily",
+        "integration": {
+            "name": "tavily_api_key",
+            "type": "api_key",
+            "store": {
+                "type": "environment",
+            }
+        },        
+    },
+    {
+        "name": "zenquotes",
+        "integration": None
+    },
+    {
+        "name": "github",
+        "integration": {
+            "name": "github",
+            "type": "agentr",
+        }
+    }
+]
+
+# Create a server with these applications
+server = TestServer(name="My Agent Server", description="A server for my agent apps", apps_list=apps_list)
+
+# Run the server
+if __name__ == "__main__":
+    server.run()
+```
+
+## 🧩 Available Applications
+AgentR comes with several pre-built applications:
+
+| Application | Description | Authentication Type |
+|-------------|-------------|---------------------|
+| GitHub | Star repositories and more | OAuth (AgentR) |
+| Google Calendar | Retrieve calendar events | OAuth (AgentR) |
+| Gmail | Send emails | OAuth (AgentR) |
+| Reddit | Access Reddit data | OAuth (AgentR) |
+| Resend | Send emails via Resend API | API Key |
+| Tavily | Web search capabilities | API Key |
+| ZenQuotes | Get inspirational quotes | None |
+
+> **Note**: More applications are coming soon! Stay tuned for updates to our application catalog.
+
+## 🔐 Integration Types
+AgentR supports two primary integration types:
+
+### 1. API Key Integration
+For services that authenticate via API keys:
+```python
+{
+    "name": "service_name",
+    "integration": {
+        "name": "service_api_key",
+        "type": "api_key",
+        "store": {
+            "type": "environment",  # or "memory"
+        }
+    }
+}
+```
+
+### 2. OAuth Integration (via AgentR)
+For services requiring OAuth flow:
+```python
+{
+    "name": "service_name",
+    "integration": {
+        "name": "service_name",
+        "type": "agentr"
+    }
+}
+```
+When using OAuth integrations, users will be directed to authenticate with the service provider through a secure flow managed by AgentR.
+
+## 🤖 CLI Usage
+AgentR includes a command-line interface for common operations:
+
+```bash
+# Get version information
+agentr version
+
+# Generate API client from OpenAPI schema
+agentr generate --schema path/to/openapi.yaml
+
+# Run the test server
+agentr run
+
+# Install AgentR for specific applications
+agentr install claude
+```
+
+## 📋 Requirements
+
+- Python 3.11+
+- Dependencies (automatically installed):
+  - loguru >= 0.7.3
+  - mcp >= 1.5.0
+  - pyyaml >= 6.0.2
+  - typer >= 0.15.2
+
+

agentr-0.1.4.dist-info/RECORD

@@ -0,0 +1,22 @@
+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.4.dist-info/METADATA,sha256=hMmkU1mQZbV2otCFm-Kirx7XS_Y4ca08NWRME5xm9y0,4173
+agentr-0.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agentr-0.1.4.dist-info/entry_points.txt,sha256=13fGFeVhgF6_8T-VFiIkNxYO7gDQaUwwTcUNWdvaLQg,42
+agentr-0.1.4.dist-info/RECORD,,

agentr-0.1.3.dist-info/METADATA

@@ -1,10 +0,0 @@
-Metadata-Version: 2.4
-Name: agentr
-Version: 0.1.3
-Summary: Add your description here
-Author-email: Manoj Bajaj <manojbajaj95@gmail.com>
-Requires-Python: >=3.11
-Requires-Dist: loguru>=0.7.3
-Requires-Dist: mcp>=1.5.0
-Requires-Dist: pyyaml>=6.0.2
-Requires-Dist: typer>=0.15.2

agentr-0.1.3.dist-info/RECORD

@@ -1,21 +0,0 @@
-agentr/__init__.py,sha256=LOWhgQayrQV7f5ro4rlBJ_6WevhbWIbjAOHnqP7b_-4,30
-agentr/application.py,sha256=2K7jL5QUDwlD_zRS7adN_R3_yPt-EJBYpAmr1abDew8,1551
-agentr/cli.py,sha256=KcmYrnzoVzkh4SWB6UDjxGJboQbMcphj1juNbfqrIAE,1676
-agentr/integration.py,sha256=jRwpp_bUXJWl8OkZsnRDqoScoRSxCMDBMrDMfRBhgN4,3298
-agentr/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentr/server.py,sha256=24cD8e3ks8ckftPMphI7K3SfGs8LZZyXpfiKU2daY-Q,4220
-agentr/store.py,sha256=fB3uAaobnWf2ILcDBmg3ToDaqAIPYlLtmHBdpmkcGcI,1585
-agentr/test.py,sha256=A0t1fdI9N89eF9xm903fKtvTzibnrHeMKkPwZnSBEc4,852
-agentr/applications/github/app.py,sha256=_PW_xzkCzNB468lEOvFzdQk0m7Ol6y_PQ5OqCkVkDVE,1811
-agentr/applications/google_calendar/app.py,sha256=WW-4MTqPl-NU2w5DKKyEvfTho4d6tS7t52jDRyq-tv0,2841
-agentr/applications/google_mail/app.py,sha256=SqzWyYW0xF_fdZC8kf_8FZfQK6OOddI-EVpdgYAengg,2334
-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=yjiPYs-_JsYQ7T3aPh7oimHUCf8pMblIR8IPCaAeNCg,6279
-agentr-0.1.3.dist-info/METADATA,sha256=YWWFhc2Z9sG48D4zr7lIDJ39JO_pK13RZ5mQO1jiWEY,273
-agentr-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agentr-0.1.3.dist-info/entry_points.txt,sha256=13fGFeVhgF6_8T-VFiIkNxYO7gDQaUwwTcUNWdvaLQg,42
-agentr-0.1.3.dist-info/RECORD,,