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

agentr/server.py

@@ -1,105 +1,105 @@
-from abc import ABC, abstractmethod
-from typing import Literal
-from mcp.server.fastmcp import FastMCP
-from agentr.integration import AgentRIntegration, ApiKeyIntegration
-from agentr.store import EnvironmentStore, MemoryStore
-from pydantic import BaseModel
-from loguru import logger
-
-class StoreConfig(BaseModel):
-    type: Literal["memory", "environment"]
-
-class IntegrationConfig(BaseModel):
-    name: str
-    type: Literal["api_key", "agentr"]
-    credentials: dict | None = None
-    store: StoreConfig | None = None
-
-class AppConfig(BaseModel):
-    name: str
-    integration: IntegrationConfig | None = None
-
-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, **kwargs):
-        super().__init__(name, description, **kwargs)
-
-    @abstractmethod
-    def _load_apps(self):
-        pass
-
-
-class TestServer(Server):
-    """
-    Test server for development purposes
-    """
-    def __init__(self, name: str, description: str, apps_list: list[AppConfig] = [], **kwargs):
-        super().__init__(name, description=description, **kwargs)
-        self.apps_list = [AppConfig.model_validate(app) for app in apps_list]
-        self._load_apps()
-    
-    def _get_store(self, store_config: StoreConfig):
-        if store_config.type == "memory":
-            return MemoryStore()
-        elif store_config.type == "environment":
-            return EnvironmentStore()
-        return None
-
-    def _get_integration(self, integration_config: IntegrationConfig):
-        if integration_config.type == "api_key":
-            store = self._get_store(integration_config.store)
-            integration = ApiKeyIntegration(integration_config.name, store=store)
-            if integration_config.credentials:
-                integration.set_credentials(integration_config.credentials)
-            return integration
-        elif integration_config.type == "agentr":
-            integration = AgentRIntegration(integration_config.name, api_key=integration_config.credentials.get("api_key") if integration_config.credentials else None)
-            return integration
-        return None
-    
-    def _load_app(self, app_config: AppConfig):
-        name = app_config.name
-        if name == "zenquotes":
-            from agentr.applications.zenquotes.app import ZenQuoteApp
-            return ZenQuoteApp()
-        elif name == "tavily":
-            from agentr.applications.tavily.app import TavilyApp
-            integration = self._get_integration(app_config.integration)
-            return TavilyApp(integration=integration)
-        elif name == "github":
-            from agentr.applications.github.app import GithubApp
-            integration = self._get_integration(app_config.integration)
-            return GithubApp(integration=integration)
-        elif name == "google-calendar":
-            from agentr.applications.google_calendar.app import GoogleCalendarApp
-            integration = self._get_integration(app_config.integration)
-            return GoogleCalendarApp(integration=integration)
-        elif name == "gmail":
-            from agentr.applications.google_mail.app import GmailApp
-            integration = self._get_integration(app_config.integration)
-            return GmailApp(integration=integration)
-        elif name == "resend":
-            from agentr.applications.resend.app import ResendApp
-            integration = self._get_integration(app_config.integration)
-            return ResendApp(integration=integration)
-        elif name == "reddit":
-            from agentr.applications.reddit.app import RedditApp
-            integration = self._get_integration(app_config.integration)
-            return RedditApp(integration=integration)
-        else:
-            return None
-            
-    def _load_apps(self):
-        logger.info(f"Loading apps: {self.apps_list}")
-        for app_config in self.apps_list:
-            app = self._load_app(app_config)
-            if app:
-                tools = app.list_tools()
-                for tool in tools:
-                    self.add_tool(tool)
-
-                
+from abc import ABC, abstractmethod
+from typing import Literal
+from mcp.server.fastmcp import FastMCP
+from agentr.integration import AgentRIntegration, ApiKeyIntegration
+from agentr.store import EnvironmentStore, MemoryStore
+from pydantic import BaseModel
+from loguru import logger
+
+class StoreConfig(BaseModel):
+    type: Literal["memory", "environment"]
+
+class IntegrationConfig(BaseModel):
+    name: str
+    type: Literal["api_key", "agentr"]
+    credentials: dict | None = None
+    store: StoreConfig | None = None
+
+class AppConfig(BaseModel):
+    name: str
+    integration: IntegrationConfig | None = None
+
+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, **kwargs):
+        super().__init__(name, description, **kwargs)
+
+    @abstractmethod
+    def _load_apps(self):
+        pass
+
+
+class TestServer(Server):
+    """
+    Test server for development purposes
+    """
+    def __init__(self, name: str, description: str, apps_list: list[AppConfig] = [], **kwargs):
+        super().__init__(name, description=description, **kwargs)
+        self.apps_list = [AppConfig.model_validate(app) for app in apps_list]
+        self._load_apps()
+    
+    def _get_store(self, store_config: StoreConfig):
+        if store_config.type == "memory":
+            return MemoryStore()
+        elif store_config.type == "environment":
+            return EnvironmentStore()
+        return None
+
+    def _get_integration(self, integration_config: IntegrationConfig):
+        if integration_config.type == "api_key":
+            store = self._get_store(integration_config.store)
+            integration = ApiKeyIntegration(integration_config.name, store=store)
+            if integration_config.credentials:
+                integration.set_credentials(integration_config.credentials)
+            return integration
+        elif integration_config.type == "agentr":
+            integration = AgentRIntegration(integration_config.name, api_key=integration_config.credentials.get("api_key") if integration_config.credentials else None)
+            return integration
+        return None
+    
+    def _load_app(self, app_config: AppConfig):
+        name = app_config.name
+        if name == "zenquotes":
+            from agentr.applications.zenquotes.app import ZenQuoteApp
+            return ZenQuoteApp()
+        elif name == "tavily":
+            from agentr.applications.tavily.app import TavilyApp
+            integration = self._get_integration(app_config.integration)
+            return TavilyApp(integration=integration)
+        elif name == "github":
+            from agentr.applications.github.app import GithubApp
+            integration = self._get_integration(app_config.integration)
+            return GithubApp(integration=integration)
+        elif name == "google-calendar":
+            from agentr.applications.google_calendar.app import GoogleCalendarApp
+            integration = self._get_integration(app_config.integration)
+            return GoogleCalendarApp(integration=integration)
+        elif name == "gmail":
+            from agentr.applications.google_mail.app import GmailApp
+            integration = self._get_integration(app_config.integration)
+            return GmailApp(integration=integration)
+        elif name == "resend":
+            from agentr.applications.resend.app import ResendApp
+            integration = self._get_integration(app_config.integration)
+            return ResendApp(integration=integration)
+        elif name == "reddit":
+            from agentr.applications.reddit.app import RedditApp
+            integration = self._get_integration(app_config.integration)
+            return RedditApp(integration=integration)
+        else:
+            return None
+            
+    def _load_apps(self):
+        logger.info(f"Loading apps: {self.apps_list}")
+        for app_config in self.apps_list:
+            app = self._load_app(app_config)
+            if app:
+                tools = app.list_tools()
+                for tool in tools:
+                    self.add_tool(tool)
+
+                

agentr/store.py

@@ -1,71 +1,71 @@
-import os
-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]
-
-
-class EnvironmentStore(Store):
-    """
-    Store that uses environment variables to store credentials.
-    """
-    def __init__(self):
-        pass
-
-    def get(self, key: str):
-        return {"api_key": os.getenv(key)}
-
-    def set(self, key: str, value: str):
-        os.environ[key] = value
-
-    def delete(self, key: str):
-        del os.environ[key]
-
-class RedisStore(Store):
-    """
-    Store that uses a redis database to store credentials.
-    """
-    def __init__(self, host: str, port: int, db: int):
-        import redis
-        self.host = host
-        self.port = port
-        self.db = db
-        self.redis = redis.Redis(host=self.host, port=self.port, db=self.db)
-
-    def get(self, key: str):
-        return self.redis.get(key)
-
-    def set(self, key: str, value: str):
-        self.redis.set(key, value)
-
-    def delete(self, key: str):
+import os
+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]
+
+
+class EnvironmentStore(Store):
+    """
+    Store that uses environment variables to store credentials.
+    """
+    def __init__(self):
+        pass
+
+    def get(self, key: str):
+        return {"api_key": os.getenv(key)}
+
+    def set(self, key: str, value: str):
+        os.environ[key] = value
+
+    def delete(self, key: str):
+        del os.environ[key]
+
+class RedisStore(Store):
+    """
+    Store that uses a redis database to store credentials.
+    """
+    def __init__(self, host: str, port: int, db: int):
+        import redis
+        self.host = host
+        self.port = port
+        self.db = db
+        self.redis = redis.Redis(host=self.host, port=self.port, db=self.db)
+
+    def get(self, key: str):
+        return self.redis.get(key)
+
+    def set(self, key: str, value: str):
+        self.redis.set(key, value)
+
+    def delete(self, key: str):
         self.redis.delete(key)

agentr/test.py

@@ -1,38 +1,38 @@
-from agentr.server import TestServer
-from agentr.store import MemoryStore
-
-store = MemoryStore()
-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",
-        }
-    }
-]
-mcp = TestServer(name="Test Server", description="Test Server", apps_list=apps_list)
-
-async def test():
-    tools = await mcp.list_tools()
-    print(tools)
-    result = await mcp.call_tool("search", {"query": "python"})
-    print(result)
-
-if __name__ == "__main__":
-    import asyncio
+from agentr.server import TestServer
+from agentr.store import MemoryStore
+
+store = MemoryStore()
+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",
+        }
+    }
+]
+mcp = TestServer(name="Test Server", description="Test Server", apps_list=apps_list)
+
+async def test():
+    tools = await mcp.list_tools()
+    print(tools)
+    result = await mcp.call_tool("star_repository", {"repo_full_name": "manojbajaj95/config"})
+    print(result)
+
+if __name__ == "__main__":
+    import asyncio
     asyncio.run(test())