kailash 0.9.6__py3-none-any.whl → 0.9.7__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.7"
 
 __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-0.9.6.dist-info → kailash-0.9.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kailash
-Version: 0.9.6
+Version: 0.9.7
 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.7.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kailash/__init__.py,sha256=4FAW32FbWRGxkLCykiHb-3-1ri0JkYDO_5s0fkjshX0,2771
+kailash/__init__.py,sha256=E3eRrjccUAH1875857yUosDh9qAHt-Inu_Q383PARDA,2771
 kailash/__main__.py,sha256=vr7TVE5o16V6LsTmRFKG6RDKUXHpIWYdZ6Dok2HkHnI,198
 kailash/access_control.py,sha256=MjKtkoQ2sg1Mgfe7ovGxVwhAbpJKvaepPWr8dxOueMA,26058
 kailash/access_control_abac.py,sha256=FPfa_8PuDP3AxTjdWfiH3ntwWO8NodA0py9W8SE5dno,30263
@@ -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.7.dist-info/licenses/LICENSE,sha256=Axe6g7bTrJkToK9h9j2SpRUKKNaDZDCo2lQ2zPxCE6s,1065
+kailash-0.9.7.dist-info/METADATA,sha256=yo7vzKt8ikuCZCQBcsJeDPMm8VRW421kg8-U64whL0w,22298
+kailash-0.9.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+kailash-0.9.7.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
+kailash-0.9.7.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
+kailash-0.9.7.dist-info/RECORD,,