kailash 0.3.1__py3-none-any.whl → 0.4.0__py3-none-any.whl

kailash/nodes/ai/a2a.py

@@ -112,8 +112,11 @@ class SharedMemoryPoolNode(Node):
         ... )
     """
 
-    def __init__(self):
-        super().__init__()
+    def __init__(self, name=None, **kwargs):
+        # Accept name parameter and pass all kwargs to parent
+        if name:
+            kwargs["name"] = name
+        super().__init__(**kwargs)
         self.memory_segments = defaultdict(deque)
         self.agent_subscriptions = defaultdict(set)
         self.attention_indices = defaultdict(lambda: defaultdict(list))
@@ -183,6 +186,13 @@ class SharedMemoryPoolNode(Node):
                 required=False,
                 description="Search query for semantic memory search",
             ),
+            "segments": NodeParameter(
+                name="segments",
+                type=list,
+                required=False,
+                default=["general"],
+                description="Memory segments to subscribe to",
+            ),
         }
 
     def run(self, **kwargs) -> Dict[str, Any]:
@@ -545,8 +555,11 @@ class A2AAgentNode(LLMAgentNode):
         >>> assert all("content" in i for i in insights)
     """
 
-    def __init__(self):
-        super().__init__()
+    def __init__(self, name=None, **kwargs):
+        # Accept name parameter and pass all kwargs to parent
+        if name:
+            kwargs["name"] = name
+        super().__init__(**kwargs)
         self.local_memory = deque(maxlen=100)
         self.communication_log = deque(maxlen=50)
 
@@ -1220,8 +1233,11 @@ class A2ACoordinatorNode(CycleAwareNode):
         ... )
     """
 
-    def __init__(self):
-        super().__init__()
+    def __init__(self, name=None, **kwargs):
+        # Accept name parameter and pass all kwargs to parent
+        if name:
+            kwargs["name"] = name
+        super().__init__(**kwargs)
         self.registered_agents = {}
         self.task_queue = deque()
         self.consensus_sessions = {}
@@ -1275,7 +1291,7 @@ class A2ACoordinatorNode(CycleAwareNode):
             ),
         }
 
-    def run(self, context: Dict[str, Any], **kwargs) -> Dict[str, Any]:
+    def run(self, **kwargs) -> Dict[str, Any]:
         """
         Execute coordination action with cycle awareness.
 
@@ -1315,6 +1331,7 @@ class A2ACoordinatorNode(CycleAwareNode):
             ... )
             >>> assert result[\"success\"] == True
         """
+        context = kwargs.get("context", {})
         action = kwargs.get("action")
 
         # Get cycle information using CycleAwareNode helpers

kailash/nodes/ai/ai_providers.py

@@ -399,6 +399,7 @@ class OllamaProvider(UnifiedAIProvider):
             **kwargs: Additional arguments including:
                 model (str): Ollama model name (default: "llama3.1:8b-instruct-q8_0")
                 generation_config (dict): Generation parameters including:
+
                     * temperature, max_tokens, top_p, top_k, repeat_penalty
                     * seed, stop, num_ctx, num_batch, num_thread
                     * tfs_z, typical_p, mirostat, mirostat_tau, mirostat_eta

kailash/nodes/ai/embedding_generator.py

@@ -94,17 +94,17 @@ class EmbeddingGeneratorNode(Node):
         ...     similarity_metric="cosine"
         ... )
 
-        Cached embedding with MCP integration:
-
-        mcp_embedder = EmbeddingGeneratorNode()
-        result = mcp_embedder.run(
-            provider="azure",
-            model="text-embedding-3-small",
-            mcp_resource_uri="data://documents/knowledge_base.json",
-            operation="embed_mcp_resource",
-            cache_ttl=3600,
-            chunk_size=512
-        )
+        Cached embedding with MCP integration::
+
+            mcp_embedder = EmbeddingGeneratorNode()
+            result = mcp_embedder.run(
+                provider="azure",
+                model="text-embedding-3-small",
+                mcp_resource_uri="data://documents/knowledge_base.json",
+                operation="embed_mcp_resource",
+                cache_ttl=3600,
+                chunk_size=512
+            )
     """
 
     def get_parameters(self) -> dict[str, NodeParameter]:

kailash/nodes/ai/intelligent_agent_orchestrator.py

@@ -132,14 +132,29 @@ class IntelligentCacheNode(Node):
         >>> assert "hit_rate" in stats["stats"]
     """
 
-    def __init__(self):
-        super().__init__()
+    def __init__(self, name: str = None, id: str = None, **kwargs):
+        # Set name from parameters
+        if name:
+            self.name = name
+        elif id:
+            self.name = id
+        elif "name" in kwargs:
+            self.name = kwargs.pop("name")
+        elif "id" in kwargs:
+            self.name = kwargs.pop("id")
+        else:
+            self.name = self.__class__.__name__
+
+        # Initialize node attributes
         self.cache = {}
         self.semantic_index = defaultdict(list)
         self.access_patterns = defaultdict(int)
         self.cost_metrics = {}
         self.query_abstractions = {}
 
+        # Call parent constructor
+        super().__init__(name=self.name)
+
     def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "action": NodeParameter(
@@ -482,12 +497,27 @@ class MCPAgentNode(SelfOrganizingAgentNode):
         >>> assert "task" in params
     """
 
-    def __init__(self):
-        super().__init__()
+    def __init__(self, name: str = None, id: str = None, **kwargs):
+        # Set name from parameters
+        if name:
+            self.name = name
+        elif id:
+            self.name = id
+        elif "name" in kwargs:
+            self.name = kwargs.pop("name")
+        elif "id" in kwargs:
+            self.name = kwargs.pop("id")
+        else:
+            self.name = self.__class__.__name__
+
+        # Initialize node attributes
         self.mcp_clients = {}
         self.tool_registry = {}
         self.call_history = deque(maxlen=100)
 
+        # Call parent constructor
+        super().__init__(name=self.name)
+
     def get_parameters(self) -> dict[str, NodeParameter]:
         params = super().get_parameters()
 
@@ -789,8 +819,20 @@ class QueryAnalysisNode(Node):
         >>> assert simple["analysis"]["team_suggestion"]["suggested_size"] >= 1
     """
 
-    def __init__(self):
-        super().__init__()
+    def __init__(self, name: str = None, id: str = None, **kwargs):
+        # Set name from parameters
+        if name:
+            self.name = name
+        elif id:
+            self.name = id
+        elif "name" in kwargs:
+            self.name = kwargs.pop("name")
+        elif "id" in kwargs:
+            self.name = kwargs.pop("id")
+        else:
+            self.name = self.__class__.__name__
+
+        # Initialize node attributes
         self.query_patterns = {
             "data_retrieval": {
                 "keywords": ["what is", "get", "fetch", "retrieve", "show me"],
@@ -824,6 +866,9 @@ class QueryAnalysisNode(Node):
             },
         }
 
+        # Call parent constructor
+        super().__init__(name=self.name)
+
     def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "query": NodeParameter(
@@ -1148,11 +1193,26 @@ class OrchestrationManagerNode(Node):
         >>> assert "max_iterations" in params
     """
 
-    def __init__(self):
-        super().__init__()
+    def __init__(self, name: str = None, id: str = None, **kwargs):
+        # Set name from parameters
+        if name:
+            self.name = name
+        elif id:
+            self.name = id
+        elif "name" in kwargs:
+            self.name = kwargs.pop("name")
+        elif "id" in kwargs:
+            self.name = kwargs.pop("id")
+        else:
+            self.name = self.__class__.__name__
+
+        # Initialize node attributes
         self.session_id = str(uuid.uuid4())
         self.orchestration_history = deque(maxlen=50)
 
+        # Call parent constructor
+        super().__init__(name=self.name)
+
     def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "query": NodeParameter(
@@ -1304,9 +1364,22 @@ class OrchestrationManagerNode(Node):
 
         # Phase 5: Final Processing
         print("📊 Phase 5: Finalizing results...")
+
+        # Handle case where solution_history is empty (time limit reached early)
+        if final_solution:
+            solution_to_use = final_solution
+        elif solution_history:
+            solution_to_use = solution_history[-1]["solution"]
+        else:
+            solution_to_use = {
+                "content": "No solution generated due to time constraints.",
+                "confidence": 0.0,
+                "reasoning": "Time limit reached before any solution could be generated.",
+            }
+
         final_result = self._finalize_results(
             query,
-            final_solution or solution_history[-1]["solution"],
+            solution_to_use,
             solution_history,
             time.time() - start_time,
             infrastructure,
@@ -1825,10 +1898,25 @@ class ConvergenceDetectorNode(Node):
         >>> assert "Diminishing returns" in result2["reason"]
     """
 
-    def __init__(self):
-        super().__init__()
+    def __init__(self, name: str = None, id: str = None, **kwargs):
+        # Set name from parameters
+        if name:
+            self.name = name
+        elif id:
+            self.name = id
+        elif "name" in kwargs:
+            self.name = kwargs.pop("name")
+        elif "id" in kwargs:
+            self.name = kwargs.pop("id")
+        else:
+            self.name = self.__class__.__name__
+
+        # Initialize node attributes
         self.convergence_history = deque(maxlen=100)
 
+        # Call parent constructor
+        super().__init__(name=self.name)
+
     def get_parameters(self) -> dict[str, NodeParameter]:
         return {
             "solution_history": NodeParameter(