agenticaiframework 1.0.25__py3-none-any.whl → 1.0.27__py3-none-any.whl

agenticaiframework/communication.py

@@ -10,6 +10,11 @@ class CommunicationManager:
         self.protocols[name] = handler_fn
         self._log(f"Registered communication protocol '{name}'")
 
+    def register_handler(self, handler_fn: Callable[[Any], Any], name: str = None):
+        """Alternative method for registering handlers - alias for register_protocol"""
+        protocol_name = name or f"handler_{len(self.protocols)}"
+        self.register_protocol(protocol_name, handler_fn)
+
     def send(self, protocol: str, data: Any):
         if protocol in self.protocols:
             try:
@@ -23,5 +28,17 @@ class CommunicationManager:
     def list_protocols(self):
         return list(self.protocols.keys())
 
+    def send_message(self, message: Any, protocol: str = None):
+        """Send a message using the first available protocol or specified protocol"""
+        if protocol:
+            return self.send(protocol, message)
+        elif self.protocols:
+            # Use the first available protocol
+            first_protocol = next(iter(self.protocols))
+            return self.send(first_protocol, message)
+        else:
+            self._log("No protocols available to send message")
+            return None
+
     def _log(self, message: str):
         print(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] [CommunicationManager] {message}")

agenticaiframework/guardrails.py

@@ -41,5 +41,13 @@ class GuardrailManager:
                 return False
         return True
 
+    def validate(self, guardrail_name: str, data: Any) -> bool:
+        """Validate data against a specific guardrail by name"""
+        for guardrail in self.guardrails.values():
+            if guardrail.name == guardrail_name:
+                return guardrail.validate(data)
+        self._log(f"Guardrail '{guardrail_name}' not found")
+        return False
+
     def _log(self, message: str):
         print(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] [GuardrailManager] {message}")

agenticaiframework/hub.py

@@ -9,6 +9,7 @@ class Hub:
         self.tools: Dict[str, Any] = {}
         self.guardrails: Dict[str, Any] = {}
         self.llms: Dict[str, Any] = {}
+        self.services: Dict[str, Any] = {}
 
     def register(self, category: str, name: str, item: Any):
         if hasattr(self, category):
@@ -17,6 +18,15 @@ class Hub:
         else:
             self._log(f"Invalid category '{category}'")
 
+    def register_service(self, name: str, service: Any):
+        """Register a service in the hub"""
+        self.services[name] = service
+        self._log(f"Registered service '{name}'")
+
+    def get_service(self, name: str) -> Any:
+        """Get a service by name"""
+        return self.services.get(name)
+
     def get(self, category: str, name: str) -> Any:
         if hasattr(self, category):
             return getattr(self, category).get(name)

agenticaiframework/knowledge.py

@@ -6,11 +6,17 @@ class KnowledgeRetriever:
     def __init__(self):
         self.sources: Dict[str, Callable[[str], List[Dict[str, Any]]]] = {}
         self.cache: Dict[str, Any] = {}
+        self.knowledge_base: Dict[str, str] = {}  # Simple in-memory knowledge store
 
     def register_source(self, name: str, retrieval_fn: Callable[[str], List[Dict[str, Any]]]):
         self.sources[name] = retrieval_fn
         self._log(f"Registered knowledge source '{name}'")
 
+    def add_knowledge(self, key: str, content: str):
+        """Add knowledge to the internal knowledge base"""
+        self.knowledge_base[key] = content
+        self._log(f"Added knowledge for key '{key}'")
+
     def retrieve(self, query: str, use_cache: bool = True) -> List[Dict[str, Any]]:
         if use_cache and query in self.cache:
             self._log(f"Cache hit for query '{query}'")

agenticaiframework/mcp_tools.py

@@ -44,5 +44,14 @@ class MCPToolManager:
             self._log(f"Tool with ID {tool_id} not found")
             return None
 
+    def execute_tool_by_name(self, tool_name: str, *args, **kwargs):
+        """Execute a tool by its name instead of ID"""
+        for tool in self.tools.values():
+            if tool.name == tool_name:
+                self._log(f"Executing MCP tool '{tool.name}'")
+                return tool.execute(*args, **kwargs)
+        self._log(f"Tool with name '{tool_name}' not found")
+        return None
+
     def _log(self, message: str):
         print(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] [MCPToolManager] {message}")

agenticaiframework/memory.py

@@ -12,6 +12,15 @@ class MemoryManager:
         self.short_term[key] = value
         self._log(f"Stored short-term memory: {key}")
 
+    def store(self, key: str, value: Any, memory_type: str = "short_term"):
+        """Generic store method that defaults to short-term memory"""
+        if memory_type == "long_term":
+            self.store_long_term(key, value)
+        elif memory_type == "external":
+            self.store_external(key, value)
+        else:
+            self.store_short_term(key, value)
+
     def store_long_term(self, key: str, value: Any):
         self.long_term[key] = value
         self._log(f"Stored long-term memory: {key}")

agenticaiframework/monitoring.py

@@ -31,5 +31,9 @@ class MonitoringSystem:
     def get_logs(self) -> List[str]:
         return self.logs
 
+    def get_metrics(self) -> Dict[str, Any]:
+        """Get all recorded metrics"""
+        return self.metrics.copy()
+
     def _log(self, message: str):
         self.log_message(f"[MonitoringSystem] {message}")

agenticaiframework/processes.py

@@ -13,6 +13,10 @@ class Process:
         self.tasks.append((task_callable, args, kwargs))
         self._log(f"Added task {task_callable.__name__}")
 
+    def add_step(self, step_callable: Callable, *args, **kwargs):
+        """Alias for add_task - add a step to the process"""
+        self.add_task(step_callable, *args, **kwargs)
+
     def execute(self):
         self.status = "running"
         self._log(f"Executing process '{self.name}' with strategy '{self.strategy}'")

agenticaiframework/prompts.py

@@ -18,9 +18,21 @@ class PromptManager:
     def __init__(self):
         self.prompts: Dict[str, Prompt] = {}
 
-    def register_prompt(self, prompt: Prompt):
-        self.prompts[prompt.id] = prompt
-        self._log(f"Registered prompt with ID {prompt.id}")
+    def register_prompt(self, prompt_or_name, prompt_obj=None):
+        if isinstance(prompt_or_name, Prompt):
+            # Original behavior: register a Prompt object
+            prompt = prompt_or_name
+            self.prompts[prompt.id] = prompt
+            self._log(f"Registered prompt with ID {prompt.id}")
+        elif isinstance(prompt_or_name, str) and prompt_obj is not None:
+            # New behavior: register with a name and Prompt object
+            prompt = prompt_obj
+            prompt.metadata = prompt.metadata or {}
+            prompt.metadata['name'] = prompt_or_name
+            self.prompts[prompt.id] = prompt
+            self._log(f"Registered prompt '{prompt_or_name}' with ID {prompt.id}")
+        else:
+            self._log("Invalid arguments for register_prompt")
 
     def get_prompt(self, prompt_id: str) -> Prompt:
         return self.prompts.get(prompt_id)

agenticaiframework/tasks.py

@@ -55,5 +55,20 @@ class TaskManager:
             results[task_id] = task.run()
         return results
 
+    def execute_task(self, task_name_or_id: str):
+        """Execute a task by name or ID"""
+        # Try to find by ID first
+        task = self.get_task(task_name_or_id)
+        if task:
+            return task.run()
+        
+        # If not found by ID, try to find by name
+        for task in self.tasks.values():
+            if task.name == task_name_or_id:
+                return task.run()
+        
+        self._log(f"Task '{task_name_or_id}' not found")
+        return None
+
     def _log(self, message: str):
         print(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] [TaskManager] {message}")

{agenticaiframework-1.0.25.dist-info → agenticaiframework-1.0.27.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agenticaiframework
-Version: 1.0.25
+Version: 1.0.27
 Summary: AgenticAI - A Python SDK for building agentic applications with advanced orchestration, monitoring, and multimodal capabilities.
 Home-page: https://github.com/isathish/AgenticAI
 Author: Sathishkumar Nagarajan

agenticaiframework-1.0.27.dist-info/RECORD

@@ -0,0 +1,20 @@
+agenticaiframework/__init__.py,sha256=Bk5u9D1FnUkZXtxvkQudCNQTSrcYfNGcjAq8NVzzcp8,1026
+agenticaiframework/agents.py,sha256=gSdEzgUeme9jrKZVaLrBwSM5S--DTVsR3GMNt3XSBc0,1896
+agenticaiframework/communication.py,sha256=eyqy4-vu3D8_6jHPmE3SgIrEgrlHZhAggKwZlAI0nXs,1710
+agenticaiframework/configurations.py,sha256=BFWyMWSc8dWrXGGE9twntQZrTyn2L_jUiH1MeSQgdVM,1155
+agenticaiframework/evaluation.py,sha256=aDsCmcYq84RLAXfWmydt6q0-GGyuYniO0L4A9AdE6qE,1090
+agenticaiframework/guardrails.py,sha256=qCNb1mmGkAj2S7lDuTp00AP-PIoqCWUkrVR8rHhrdDc,1914
+agenticaiframework/hub.py,sha256=CXal8SYqmu5k9291gPaBbkuqgb4EcXVpR47b21zNKS4,1585
+agenticaiframework/knowledge.py,sha256=gcPAYEC_3iCErsPsIpzSJficzjLWN0OQsnofJ28fRvc,1596
+agenticaiframework/llms.py,sha256=BlW386KDKqtc7iyeUmrQR_-zpIkn2aIKIr6-AK4oJrY,1217
+agenticaiframework/mcp_tools.py,sha256=QoquKZKLtbt9aongt8aiYwZj4PIqTKyrZo2JhZ7qXHI,1935
+agenticaiframework/memory.py,sha256=GFUmDQEuKwVlKnxhAzj4ymtUyLAKkUc0zD1WlA0sQbQ,1750
+agenticaiframework/monitoring.py,sha256=Fn_ajaFWIRucKffFHc4gzxmmFKUJYL1E2DA_8MQYBbo,1252
+agenticaiframework/processes.py,sha256=A4HPi9dZAAY2HaXmgDUp8Wggmy8E4iD2BRq-RLLvGh0,2085
+agenticaiframework/prompts.py,sha256=UwMUVAcI-GIGrpmJXjsk5_KeOR8x6VcgmW3DgatIw3A,2046
+agenticaiframework/tasks.py,sha256=IY1HHSzYK10jXS72Ll-1lROeQRmpWG2a_ovgYU05uws,2336
+agenticaiframework-1.0.27.dist-info/licenses/LICENSE,sha256=fKYjpVuxF75tWQhzLkN90tDG4knIjDZKUMiPJbKWxIw,1079
+agenticaiframework-1.0.27.dist-info/METADATA,sha256=8CpjWa03_MUR1V2DB7h8_IWhCl9O2AHSbvLcs2Ke9Jc,5917
+agenticaiframework-1.0.27.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agenticaiframework-1.0.27.dist-info/top_level.txt,sha256=q30EgNdX3TPF02Aq6mVwI0LUBpgfH12X7pwBIwm0mPo,19
+agenticaiframework-1.0.27.dist-info/RECORD,,

agenticaiframework-1.0.25.dist-info/RECORD

@@ -1,20 +0,0 @@
-agenticaiframework/__init__.py,sha256=Bk5u9D1FnUkZXtxvkQudCNQTSrcYfNGcjAq8NVzzcp8,1026
-agenticaiframework/agents.py,sha256=gSdEzgUeme9jrKZVaLrBwSM5S--DTVsR3GMNt3XSBc0,1896
-agenticaiframework/communication.py,sha256=aWhXWCGu5vSq4MeSm2AoVYKEnM5IUqqux99YTzZRGWI,906
-agenticaiframework/configurations.py,sha256=BFWyMWSc8dWrXGGE9twntQZrTyn2L_jUiH1MeSQgdVM,1155
-agenticaiframework/evaluation.py,sha256=aDsCmcYq84RLAXfWmydt6q0-GGyuYniO0L4A9AdE6qE,1090
-agenticaiframework/guardrails.py,sha256=JhUqYU-sn2R2ycjbkVHkuv9DHtfTICnKnWJhgPy9ZZE,1554
-agenticaiframework/hub.py,sha256=ezwChjQlJofNFlsuGwxp2Cz_iM2CIn-vVGMQfscUg4Y,1231
-agenticaiframework/knowledge.py,sha256=Lqw-c4BJpnwz0C_GCID7J2MyzkNKr9b7i94n1TIqboY,1301
-agenticaiframework/llms.py,sha256=BlW386KDKqtc7iyeUmrQR_-zpIkn2aIKIr6-AK4oJrY,1217
-agenticaiframework/mcp_tools.py,sha256=VSXbepC-YVYF6cHi33KDiGx9_vm5zepmQYF5h1r-kyU,1533
-agenticaiframework/memory.py,sha256=5GrvUvuyN9g2haROpYmkUwDICw0q0eConQv-dtV15FE,1375
-agenticaiframework/monitoring.py,sha256=1xE0HrSUhTcV48nu8M0tPssOEbIX7DpMlCxPBlafM9I,1132
-agenticaiframework/processes.py,sha256=oxuR4r5CKt7oIfJwAqKNe_TN_j1HNxvJfEsGL1oBMFA,1903
-agenticaiframework/prompts.py,sha256=f0Biar0GvQ5bXB2jqUkSy-5FG468qBU6jJEVLAsLXRs,1398
-agenticaiframework/tasks.py,sha256=hJKrFYJNlqkyGwl6qVYjXfZ7aeBRC9xGKEaWobF_KA4,1850
-agenticaiframework-1.0.25.dist-info/licenses/LICENSE,sha256=fKYjpVuxF75tWQhzLkN90tDG4knIjDZKUMiPJbKWxIw,1079
-agenticaiframework-1.0.25.dist-info/METADATA,sha256=9B0BuA2vbdYeH2zHKjYSdFCwo1YhgyT79yCUGy7TxXY,5917
-agenticaiframework-1.0.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agenticaiframework-1.0.25.dist-info/top_level.txt,sha256=q30EgNdX3TPF02Aq6mVwI0LUBpgfH12X7pwBIwm0mPo,19
-agenticaiframework-1.0.25.dist-info/RECORD,,