kailash 0.9.6__py3-none-any.whl → 0.9.8__py3-none-any.whl

kailash/__init__.py

@@ -50,7 +50,7 @@ except ImportError:
 # For backward compatibility
 WorkflowGraph = Workflow
 
-__version__ = "0.9.6"
+__version__ = "0.9.8"
 
 __all__ = [
     # Core workflow components

kailash/nodes/ai/iterative_llm_agent.py

@@ -727,26 +727,38 @@ class IterativeLLMAgentNode(LLMAgentNode):
             elif isinstance(tool, dict):
                 plan["selected_tools"].append(tool.get("name", "unknown"))
 
-        # Create execution steps
+        # Create execution steps based on query and available tools
         if "analyze" in user_query.lower():
+            # For analysis queries, create multi-step plan
             plan["execution_steps"] = [
                 {
                     "step": 1,
                     "action": "gather_data",
-                    "tools": plan["selected_tools"][:1],
+                    "tools": (
+                        plan["selected_tools"][:1] if plan["selected_tools"] else []
+                    ),
                 },
                 {
                     "step": 2,
                     "action": "perform_analysis",
-                    "tools": plan["selected_tools"][1:2],
+                    "tools": (
+                        plan["selected_tools"][1:2]
+                        if len(plan["selected_tools"]) > 1
+                        else []
+                    ),
                 },
                 {
                     "step": 3,
                     "action": "generate_insights",
-                    "tools": plan["selected_tools"][2:3],
+                    "tools": (
+                        plan["selected_tools"][2:3]
+                        if len(plan["selected_tools"]) > 2
+                        else []
+                    ),
                 },
             ]
         else:
+            # For other queries, single step execution
             plan["execution_steps"] = [
                 {"step": 1, "action": "execute_query", "tools": plan["selected_tools"]}
             ]
@@ -959,11 +971,60 @@ class IterativeLLMAgentNode(LLMAgentNode):
                 self.logger.error(f"Tool execution failed for {tool_name}: {e}")
 
         # Combine all tool outputs
-        step_result["output"] = (
-            "\n".join(tool_results)
-            if tool_results
-            else f"No tools executed for action: {action}"
-        )
+        if tool_results:
+            step_result["output"] = "\n".join(tool_results)
+        else:
+            # No tools were executed - fall back to LLM for this action
+            self.logger.info(
+                f"No MCP tools available for action: {action}, using LLM fallback"
+            )
+
+            # Extract user query from kwargs
+            messages = kwargs.get("messages", [])
+            user_query = ""
+            for msg in reversed(messages):
+                if msg.get("role") == "user":
+                    user_query = msg.get("content", "")
+                    break
+
+            # Create a prompt for the LLM to handle this action
+            action_prompt = f"Please {action} for the following request: {user_query}"
+            llm_messages = [
+                {
+                    "role": "system",
+                    "content": kwargs.get(
+                        "system_prompt", "You are a helpful AI assistant."
+                    ),
+                },
+                {"role": "user", "content": action_prompt},
+            ]
+
+            # Use parent's LLM capabilities
+            try:
+                llm_kwargs = {
+                    "provider": kwargs.get("provider", "openai"),
+                    "model": kwargs.get("model", "gpt-4"),
+                    "messages": llm_messages,
+                    "temperature": kwargs.get("temperature", 0.7),
+                    "max_tokens": kwargs.get("max_tokens", 1000),
+                }
+
+                llm_response = super().run(**llm_kwargs)
+
+                if llm_response.get("success") and llm_response.get("response"):
+                    content = llm_response["response"].get("content", "")
+                    step_result["output"] = f"LLM Response for {action}: {content}"
+                    step_result["success"] = True
+                else:
+                    step_result["output"] = (
+                        f"Failed to execute {action}: {llm_response.get('error', 'Unknown error')}"
+                    )
+                    step_result["success"] = False
+            except Exception as e:
+                self.logger.error(f"LLM fallback failed for action {action}: {e}")
+                step_result["output"] = f"Error executing {action}: {str(e)}"
+                step_result["success"] = False
+
         step_result["duration"] = time.time() - start_time
 
         # Mark as failed if no tools executed successfully

kailash/nodes/base.py

@@ -216,17 +216,58 @@ class Node(ABC):
             )
             self.logger = logging.getLogger(f"kailash.nodes.{self.id}")
 
-            # Filter out internal fields from config
-            internal_fields = {
+            # Filter out internal fields from config with comprehensive parameter handling
+            # Get parameter definitions once and cache for both filtering and validation
+            try:
+                if not hasattr(self, "_temp_param_definitions"):
+                    self._temp_param_definitions = self.get_parameters()
+                defined_params = set(self._temp_param_definitions.keys())
+            except Exception as e:
+                # If get_parameters() fails, log but continue with safe defaults
+                self.logger.debug(
+                    f"Could not get parameter definitions during init: {e}"
+                )
+                defined_params = set()
+                self._temp_param_definitions = {}
+
+            # Comprehensive parameter filtering: handle ALL potential conflicts
+            # Fields that are always internal (never user parameters)
+            always_internal = {"metadata"}
+
+            # Fields that can be either internal or user parameters
+            potentially_user_params = {
                 "id",
                 "name",
                 "description",
                 "version",
                 "author",
                 "tags",
-                "metadata",
             }
-            self.config = {k: v for k, v in kwargs.items() if k not in internal_fields}
+
+            # Build dynamic filter list based on user-defined parameters
+            internal_fields = always_internal.copy()
+            for field in potentially_user_params:
+                if field not in defined_params:
+                    # Field is not user-defined, so treat as internal field
+                    internal_fields.add(field)
+                # If field IS user-defined, don't add to internal_fields (preserve it)
+
+            # Also filter any field that starts with metadata prefix or other internal patterns
+            # This handles cases like 'metadata_name', 'metadata_*', etc.
+            def is_internal_field(field_name: str) -> bool:
+                # Check if it's in our explicit internal fields list
+                if field_name in internal_fields:
+                    return True
+                # Check for metadata-related field patterns
+                if field_name.startswith("metadata_"):
+                    return True
+                # Check for other internal patterns
+                if field_name.startswith("_"):  # Private fields
+                    return True
+                return False
+
+            # Apply comprehensive filtering
+            self.config = {k: v for k, v in kwargs.items() if not is_internal_field(k)}
 
             # Parameter resolution cache - initialize before validation
             cache_size = int(
@@ -554,11 +595,23 @@ class Node(ABC):
         return bool(re.match(r"^\$\{[^}]+\}$", value))
 
     def _get_cached_parameters(self) -> dict[str, NodeParameter]:
-        """Get cached parameter definitions.
+        """Get cached parameter definitions with optimal performance.
+
+        Uses parameters cached during initialization to avoid duplicate get_parameters() calls.
 
         Returns:
             Dictionary of parameter definitions, cached for performance
         """
+        # First check if we have parameters cached from initialization
+        if hasattr(self, "_temp_param_definitions") and self._temp_param_definitions:
+            # Use cached parameters from init and clean up temporary cache
+            if self._cached_params is None:
+                self._cached_params = self._temp_param_definitions
+                # Clean up temporary cache to free memory
+                delattr(self, "_temp_param_definitions")
+            return self._cached_params
+
+        # Fallback to original behavior if no cached parameters from init
         if self._cached_params is None:
             try:
                 self._cached_params = self.get_parameters()

{kailash-0.9.6.dist-info → kailash-0.9.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kailash
-Version: 0.9.6
+Version: 0.9.8
 Summary: Python SDK for the Kailash container-node architecture
 Home-page: https://github.com/integrum/kailash-python-sdk
 Author: Integrum

{kailash-0.9.6.dist-info → kailash-0.9.8.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kailash/__init__.py,sha256=4FAW32FbWRGxkLCykiHb-3-1ri0JkYDO_5s0fkjshX0,2771
+kailash/__init__.py,sha256=X2FdWjwIF5LEYMDpF8b6b_KmtkP63DBf3-RNmu7ytnw,2771
 kailash/__main__.py,sha256=vr7TVE5o16V6LsTmRFKG6RDKUXHpIWYdZ6Dok2HkHnI,198
 kailash/access_control.py,sha256=MjKtkoQ2sg1Mgfe7ovGxVwhAbpJKvaepPWr8dxOueMA,26058
 kailash/access_control_abac.py,sha256=FPfa_8PuDP3AxTjdWfiH3ntwWO8NodA0py9W8SE5dno,30263
@@ -140,7 +140,7 @@ kailash/monitoring/__init__.py,sha256=C5WmkNpk_mmAScqMWiCfkUbjhM5W16dsnRnc3Ial-U
 kailash/monitoring/alerts.py,sha256=eKX4ooPw1EicumPuswlR_nU18UgRETWvFg8FzCW5pVU,21416
 kailash/monitoring/metrics.py,sha256=SiAnL3o6K0QaJHgfAuWBa-0pTkW5zymhuPEsj4bgOgM,22022
 kailash/nodes/__init__.py,sha256=p2KSo0dyUBCLClU123qpQ0tyv5S_36PTxosNyW58nyY,1031
-kailash/nodes/base.py,sha256=3KPCp2PDLCPGm4VHSHt8QSONLTX9y3UhQ-3ldQf4oUg,82623
+kailash/nodes/base.py,sha256=GR2E1fWf8j1yMvJic7m2NAih7kjY1NtoDi47hHwoZ40,85437
 kailash/nodes/base_async.py,sha256=whxepCiVplrltfzEQuabmnGCpEV5WgfqwgxbLdCyiDk,8864
 kailash/nodes/base_cycle_aware.py,sha256=Xpze9xZzLepWeLpi9Y3tMn1dm2LVv-omr5TSQuGTtWo,13377
 kailash/nodes/base_with_acl.py,sha256=ZfrkLPgrEBcNbG0LKvtq6glDxyOYOMRw3VXX4vWX6bI,11852
@@ -166,7 +166,7 @@ kailash/nodes/ai/ai_providers.py,sha256=egfiOZzPmZ10d3wBCJ6ST4tRFrrtq0kt1VyCqxVp
 kailash/nodes/ai/embedding_generator.py,sha256=akGCzz7zLRSziqEQCiPwL2qWhRWxuM_1RQh-YtVEddw,31879
 kailash/nodes/ai/hybrid_search.py,sha256=k26uDDP_bwrIpv7Yl7PBCPvWSyQEmTlBjI1IpbgDsO4,35446
 kailash/nodes/ai/intelligent_agent_orchestrator.py,sha256=LvBqMKc64zSxFWVCjbLKKel2QwEzoTeJAEgna7rZw00,83097
-kailash/nodes/ai/iterative_llm_agent.py,sha256=Ed5hOuIrUev7hR5XzAfpfNHRjcZkJYSO_aOZsvvQDkI,98619
+kailash/nodes/ai/iterative_llm_agent.py,sha256=G6pQnvSJcMBxloBvLBletFdiIRZGntNaMaVx2no0igY,101273
 kailash/nodes/ai/llm_agent.py,sha256=NeNJZbV_VOUbULug2LASwyzLyoUO5wi58Bc9sXTubuc,90181
 kailash/nodes/ai/models.py,sha256=wsEeUTuegy87mnLtKgSTg7ggCXvC1n3MsL-iZ4qujHs,16393
 kailash/nodes/ai/self_organizing.py,sha256=B7NwKaBW8OHQBf5b0F9bSs8Wm-5BDJ9IjIkxS9h00mg,62885
@@ -403,9 +403,9 @@ kailash/workflow/templates.py,sha256=XQMAKZXC2dlxgMMQhSEOWAF3hIbe9JJt9j_THchhAm8
 kailash/workflow/type_inference.py,sha256=i1F7Yd_Z3elTXrthsLpqGbOnQBIVVVEjhRpI0HrIjd0,24492
 kailash/workflow/validation.py,sha256=r2zApGiiG8UEn7p5Ji842l8OR1_KftzDkWc7gg0cac0,44675
 kailash/workflow/visualization.py,sha256=nHBW-Ai8QBMZtn2Nf3EE1_aiMGi9S6Ui_BfpA5KbJPU,23187
-kailash-0.9.6.dist-info/licenses/LICENSE,sha256=Axe6g7bTrJkToK9h9j2SpRUKKNaDZDCo2lQ2zPxCE6s,1065
-kailash-0.9.6.dist-info/METADATA,sha256=ITS5v3xASITLatpJV0zFwF24A7gv4afVqPU00beunMQ,22298
-kailash-0.9.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kailash-0.9.6.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
-kailash-0.9.6.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
-kailash-0.9.6.dist-info/RECORD,,
+kailash-0.9.8.dist-info/licenses/LICENSE,sha256=Axe6g7bTrJkToK9h9j2SpRUKKNaDZDCo2lQ2zPxCE6s,1065
+kailash-0.9.8.dist-info/METADATA,sha256=u5ctxqIA1Wv7Vkj-25IgjKvU33wmvjImTjYku2hyFaY,22298
+kailash-0.9.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+kailash-0.9.8.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
+kailash-0.9.8.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
+kailash-0.9.8.dist-info/RECORD,,