agenticaiframework 1.0.25__tar.gz → 1.0.27__tar.gz

{agenticaiframework-1.0.25 → agenticaiframework-1.0.27}/PKG-INFO

@@ -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.25 → agenticaiframework-1.0.27}/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-1.0.25 → agenticaiframework-1.0.27}/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-1.0.25 → agenticaiframework-1.0.27}/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-1.0.25 → agenticaiframework-1.0.27}/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-1.0.25 → agenticaiframework-1.0.27}/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-1.0.25 → agenticaiframework-1.0.27}/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-1.0.25 → agenticaiframework-1.0.27}/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-1.0.25 → agenticaiframework-1.0.27}/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-1.0.25 → agenticaiframework-1.0.27}/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-1.0.25 → agenticaiframework-1.0.27}/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 → agenticaiframework-1.0.27}/agenticaiframework.egg-info/PKG-INFO

@@ -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.25 → agenticaiframework-1.0.27}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="agenticaiframework",
-    version="1.0.25",
+    version="1.0.27",
     author="Sathishkumar Nagarajan",
     author_email="mail@sathishkumarnagarajan.com",
     description="AgenticAI - A Python SDK for building agentic applications with advanced orchestration, monitoring, and multimodal capabilities.",

{agenticaiframework-1.0.25 → agenticaiframework-1.0.27}/tests/test_agenticai_additional.py

@@ -266,7 +266,7 @@ def test_monitoring_system_log_and_metrics(capsys):
         ms.log_event("evt1")
     except TypeError:
         try:
-            ms.log_event("evt1", {"info": "test"})
+            ms.log_event("evt1", {"info": "test"})  # type: ignore[call-arg]
         except Exception:
             pass
     try: