agentr 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

agentr/application.py

@@ -0,0 +1,50 @@
+from abc import ABC, abstractmethod
+from agentr.store import Store
+import httpx
+
+class Application(ABC):
+    def __init__(self, name: str, **kwargs):
+        self.name = name
+
+    @abstractmethod
+    def list_tools(self):
+        pass
+
+
+class APIApplication(Application):
+    def __init__(self, name: str, store: Store, **kwargs):
+        super().__init__(name, **kwargs)
+        self.store = store
+
+    def _get_headers(self):
+        return {}
+    
+    def _get(self, url, params=None):
+        headers = self._get_headers()
+        response = httpx.get(url, headers=headers, params=params)
+        return response
+    
+    def _post(self, url, data):
+        headers = self._get_headers()
+        response = httpx.post(url, headers=headers, data=data)
+        return response
+
+    def _put(self, url, data):
+        headers = self._get_headers()
+        response = httpx.put(url, headers=headers, data=data)
+        return response
+
+    def _delete(self, url):
+        headers = self._get_headers()
+        response = httpx.delete(url, headers=headers)
+        return response
+
+    def validate(self):
+        pass
+    
+    def authorize(self):
+        pass
+
+    @abstractmethod
+    def list_tools(self):
+        pass

agentr/applications/zenquotes/app.py

@@ -0,0 +1,21 @@
+from agentr.application import APIApplication
+
+
+class ZenQuoteApp(APIApplication):
+    def __init__(self, **kwargs) -> None:
+        super().__init__(name="zenquote", **kwargs)
+
+    def get_quote(self) -> str:
+        """Get an inspirational quote from the Zen Quotes API
+        
+        Returns:
+            A random inspirational quote
+        """
+        url = "https://zenquotes.io/api/random"
+        response = self._get(url)
+        data = response.json()
+        quote_data = data[0]
+        return f"{quote_data['q']} - {quote_data['a']}"
+
+    def list_tools(self):
+        return [self.get_quote]

agentr/cli.py

@@ -25,5 +25,11 @@ def generate(schema_path: Path = typer.Option(..., "--schema", "-s")):
     code = generate_api_client(schema)
     print(code)
 
+@app.command()
+def run():
+    """Run the MCP server"""
+    from agentr.mcp import mcp
+    mcp.run()
+
 if __name__ == "__main__":
     app()

agentr/mcp.py

@@ -0,0 +1,9 @@
+from agentr.server import TestServer
+from agentr.store import MemoryStore
+
+store = MemoryStore()
+
+mcp = TestServer(name="Test Server", description="Test Server", store=store)
+
+if __name__ == "__main__":
+    mcp.run()

agentr/server.py

@@ -0,0 +1,33 @@
+from abc import ABC, abstractmethod
+from mcp.server.fastmcp import FastMCP
+
+from agentr.applications.zenquotes.app import ZenQuoteApp
+from agentr.store import Store
+
+class Server(FastMCP, ABC):
+    """
+    Server is responsible for managing the applications and the store
+    It also acts as a router for the applications, and exposed to the client
+
+    """
+    def __init__(self, name: str, description: str, store: Store, **kwargs):
+        self.store = store
+        super().__init__(name, description, **kwargs)
+
+
+class TestServer(Server):
+    """
+    Test server for development purposes
+    """
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self.apps_list = ["zenquotes"]
+        self.__load_apps()
+
+    def __load_apps(self):
+        self.apps = []
+        for app in self.apps_list:
+            if app == "zenquotes":
+                app = ZenQuoteApp(store=self.store)
+                tools = app.list_tools()
+                self.add_tool(tools[0])

agentr/store.py

@@ -0,0 +1,35 @@
+
+from abc import ABC, abstractmethod
+
+
+class Store(ABC):
+    @abstractmethod
+    def get(self, key: str):
+        pass
+
+    @abstractmethod
+    def set(self, key: str, value: str):
+        pass
+
+    @abstractmethod
+    def delete(self, key: str):
+        pass
+
+class MemoryStore:
+    """
+    Acts as credential store for the applications.
+    Responsible for storing and retrieving credentials. 
+    Ideally should be a key value store
+    """
+    def __init__(self):
+        self.data = {}
+
+    def get(self, key: str):
+        return self.data.get(key)
+
+    def set(self, key: str, value: str):
+        self.data[key] = value
+
+    def delete(self, key: str):
+        del self.data[key]
+

{agentr-0.1.0.dist-info → agentr-0.1.2.dist-info}/METADATA

@@ -1,8 +1,9 @@
 Metadata-Version: 2.4
 Name: agentr
-Version: 0.1.0
+Version: 0.1.2
 Summary: Add your description here
 Author-email: Manoj Bajaj <manojbajaj95@gmail.com>
-Requires-Python: >=3.13
+Requires-Python: >=3.11
+Requires-Dist: mcp>=1.5.0
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: typer>=0.15.2

agentr-0.1.2.dist-info/RECORD

@@ -0,0 +1,13 @@
+agentr/__init__.py,sha256=LOWhgQayrQV7f5ro4rlBJ_6WevhbWIbjAOHnqP7b_-4,30
+agentr/application.py,sha256=rQ2vomCfZKigdXQLjwCvlTC9_UcnKyux_x9evmQqnjA,1220
+agentr/cli.py,sha256=F16qdCnfgg9UJt7D7VBjZ2-nKvwb7dJ3zezbopNH2YQ,877
+agentr/mcp.py,sha256=kGraScdBgDkCvrzbyg07TTUiIVMXIOF_um7v4g4-4Cs,216
+agentr/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+agentr/server.py,sha256=fweppEwF7L9cRow1zDPHtLYQE-qUdTlICeYUbxHGb2w,990
+agentr/store.py,sha256=u18GL9nDQgkkGm8HFIMz6BiHmy04EDi3vugotbH87ss,689
+agentr/applications/zenquotes/app.py,sha256=4LjYeWdERI8ZMzkajOsDgy9IRTA9plUKem-rW4M03sA,607
+agentr/utils/openapi.py,sha256=yjiPYs-_JsYQ7T3aPh7oimHUCf8pMblIR8IPCaAeNCg,6279
+agentr-0.1.2.dist-info/METADATA,sha256=n6pFbzT-1SjybZX5H81kEjRcpOpfo-HvOkxUMY48PbU,244
+agentr-0.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agentr-0.1.2.dist-info/entry_points.txt,sha256=13fGFeVhgF6_8T-VFiIkNxYO7gDQaUwwTcUNWdvaLQg,42
+agentr-0.1.2.dist-info/RECORD,,

agentr-0.1.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-agentr/__init__.py,sha256=LOWhgQayrQV7f5ro4rlBJ_6WevhbWIbjAOHnqP7b_-4,30
-agentr/cli.py,sha256=6vm9USDbtWBMuvnrBmSHehqGgTsAGpXmNWBxMNWIu-8,776
-agentr/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentr/server.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentr/utils/openapi.py,sha256=yjiPYs-_JsYQ7T3aPh7oimHUCf8pMblIR8IPCaAeNCg,6279
-agentr-0.1.0.dist-info/METADATA,sha256=1juL-iAF7qM0JpG8WdSzMyFreouhD2m2BWyg6z_mScU,218
-agentr-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agentr-0.1.0.dist-info/entry_points.txt,sha256=13fGFeVhgF6_8T-VFiIkNxYO7gDQaUwwTcUNWdvaLQg,42
-agentr-0.1.0.dist-info/RECORD,,