fiX

app.log

@@ -50483,3 +50483,152 @@ Traceback (most recent call last):
     elif isinstance(message, SystemMessage):
 NameError: name 'SystemMessage' is not defined
 2025-06-06 19:34:07:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:35:43:__main__:INFO: Starting the interface
+2025-06-06 19:35:47:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 19:36:17:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:38:19:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:38:57:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:38:57:__main__:ERROR: Exception occurred
+Traceback (most recent call last):
+  File "/home/user/app/app.py", line 97, in chat_fn
+    async for stream_mode, chunk in graph.astream(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2655, in astream
+    async for _ in runner.atick(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 400, in atick
+    _panic_or_proceed(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 509, in _panic_or_proceed
+    raise exc
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/retry.py", line 136, in arun_with_retry
+    return await task.proc.ainvoke(task.input, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 676, in ainvoke
+    input = await step.ainvoke(input, config, **kwargs)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 440, in ainvoke
+    ret = await self.afunc(*args, **kwargs)
+  File "/home/user/app/graph.py", line 145, in guidance_node
+    elif isinstance(message, SystemMessage):
+NameError: name 'SystemMessage' is not defined
+2025-06-06 19:42:46:__main__:INFO: Starting the interface
+2025-06-06 19:42:50:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 19:42:59:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:42:59:__main__:ERROR: Exception occurred
+Traceback (most recent call last):
+  File "/home/user/app/app.py", line 97, in chat_fn
+    async for stream_mode, chunk in graph.astream(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2655, in astream
+    async for _ in runner.atick(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 400, in atick
+    _panic_or_proceed(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 509, in _panic_or_proceed
+    raise exc
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/retry.py", line 136, in arun_with_retry
+    return await task.proc.ainvoke(task.input, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 676, in ainvoke
+    input = await step.ainvoke(input, config, **kwargs)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 440, in ainvoke
+    ret = await self.afunc(*args, **kwargs)
+  File "/home/user/app/graph.py", line 147, in guidance_node
+    elif isinstance(message, ToolMessage):
+NameError: name 'ToolMessage' is not defined. Did you mean: 'AnyMessage'?
+2025-06-06 19:43:50:__main__:INFO: Starting the interface
+2025-06-06 19:43:55:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 19:43:59:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:44:44:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:45:00:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:45:19:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:48:01:__main__:INFO: Starting the interface
+2025-06-06 19:48:04:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 19:49:31:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:49:31:__main__:ERROR: Exception occurred
+Traceback (most recent call last):
+  File "/home/user/app/app.py", line 97, in chat_fn
+    async for stream_mode, chunk in graph.astream(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2655, in astream
+    async for _ in runner.atick(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 400, in atick
+    _panic_or_proceed(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 509, in _panic_or_proceed
+    raise exc
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/retry.py", line 136, in arun_with_retry
+    return await task.proc.ainvoke(task.input, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 676, in ainvoke
+    input = await step.ainvoke(input, config, **kwargs)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 440, in ainvoke
+    ret = await self.afunc(*args, **kwargs)
+  File "/home/user/app/graph.py", line 137, in guidance_node
+    filtered_messages.append(message)
+NameError: name 'filtered_messages' is not defined
+2025-06-06 19:50:04:__main__:INFO: Starting the interface
+2025-06-06 19:50:08:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 19:52:01:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:52:22:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:54:21:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:55:08:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 19:55:15:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:56:47:__main__:INFO: Starting the interface
+2025-06-06 19:57:07:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:58:40:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 20:25:30:__main__:INFO: Starting the interface
+2025-06-06 20:25:47:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 20:25:47:__main__:ERROR: Exception occurred
+Traceback (most recent call last):
+  File "/home/user/app/app.py", line 97, in chat_fn
+    async for stream_mode, chunk in graph.astream(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2655, in astream
+    async for _ in runner.atick(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 400, in atick
+    _panic_or_proceed(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 509, in _panic_or_proceed
+    raise exc
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/retry.py", line 136, in arun_with_retry
+    return await task.proc.ainvoke(task.input, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 676, in ainvoke
+    input = await step.ainvoke(input, config, **kwargs)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 440, in ainvoke
+    ret = await self.afunc(*args, **kwargs)
+  File "/home/user/app/graph.py", line 172, in guidance_node
+    f"Based on the status, the most logical next stage appears to be: **'{proposed_next_stage}'**.\n\n"
+NameError: name 'proposed_next_stage' is not defined
+2025-06-06 20:26:00:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 20:29:14:__main__:INFO: Starting the interface
+2025-06-06 20:29:45:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 20:29:45:__main__:ERROR: Exception occurred
+Traceback (most recent call last):
+  File "/home/user/app/app.py", line 97, in chat_fn
+    async for stream_mode, chunk in graph.astream(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2655, in astream
+    async for _ in runner.atick(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 400, in atick
+    _panic_or_proceed(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 509, in _panic_or_proceed
+    raise exc
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/retry.py", line 136, in arun_with_retry
+    return await task.proc.ainvoke(task.input, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 676, in ainvoke
+    input = await step.ainvoke(input, config, **kwargs)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 440, in ainvoke
+    ret = await self.afunc(*args, **kwargs)
+  File "/home/user/app/graph.py", line 198, in guidance_node
+    final_prompt = "\n".join([guidance_prompt, state.prompt, ASSISTANT_SYSTEM_PROMPT_BASE])
+NameError: name 'guidance_prompt' is not defined. Did you mean: 'guidance_prompt_text'?
+2025-06-06 20:30:18:__main__:INFO: Starting the interface
+2025-06-06 20:30:39:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 20:30:52:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 20:30:52:__main__:ERROR: Exception occurred
+Traceback (most recent call last):
+  File "/home/user/app/app.py", line 97, in chat_fn
+    async for stream_mode, chunk in graph.astream(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2655, in astream
+    async for _ in runner.atick(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 400, in atick
+    _panic_or_proceed(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 509, in _panic_or_proceed
+    raise exc
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/retry.py", line 136, in arun_with_retry
+    return await task.proc.ainvoke(task.input, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 676, in ainvoke
+    input = await step.ainvoke(input, config, **kwargs)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 440, in ainvoke
+    ret = await self.afunc(*args, **kwargs)
+  File "/home/user/app/graph.py", line 145, in guidance_node
+    elif isinstance(message, SystemMessage):
+NameError: name 'SystemMessage' is not defined

graph.py

@@ -23,6 +23,8 @@ from langgraph.prebuilt import ToolNode
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.types import Command, interrupt
 
+from typing import TypedDict, List, Optional
+
 class State(TypedDict):
     messages: Annotated[list, add_messages]
 
@@ -151,12 +153,9 @@ async def guidance_node(state: GraphProcessingState, config=None):
     print(f"Search Enabled: {state.search_enabled}")
     print(f"Next Stage: {state.next_stage}")
 
-    print(f"Idea Complete: {state.idea_complete}")
+
     print(f"Brainstorming Complete: {state.brainstorming_complete}")
-    print(f"Planning Complete: {state.planning_complete}")
-    print(f"Drawing Complete: {state.drawing_complete}")
-    print(f"Product Searching Complete: {state.product_searching_complete}")
-    print(f"Purchasing Complete: {state.purchasing_complete}")
+    
 
     guidance_node.count = getattr(guidance_node, 'count', 0) + 1
     print('\nGuidance Node called count', guidance_node.count)
@@ -166,19 +165,41 @@ async def guidance_node(state: GraphProcessingState, config=None):
     completed = [stage for stage in stage_order if getattr(state, f"{stage}_complete", False)]
     incomplete = [stage for stage in stage_order if not getattr(state, f"{stage}_complete", False)]
 
+    if not incomplete:
+        print("All stages complete!")
+        # Handle case where all stages are complete
+        # You might want to return a message and end, or set proposed_next_stage to a special value
+        ai_all_complete_msg = AIMessage(content="All DIY project stages are complete!")
+        return {
+            "messages": current_messages + [ai_all_complete_msg],
+            "next_stage": "end_project", # Or None, or a final summary node
+            "pending_approval_stage": None,
+        }
+    else:
+        # THIS LINE DEFINES THE VARIABLE
+        proposed_next_stage = incomplete[0]
+
+    print(f"Proposed next stage: {proposed_next_stage}")
+
     status_summary = f"Completed stages: {completed}\nIncomplete stages: {incomplete}"
 
-    guidance_prompt = (
-        "You are the guiding assistant. Based on the user's input and the current project status, "
-        "help decide the next stage in the DIY process. Ask the user which stage to continue next if needed.\n\n"
-        f"CURRENT STATUS:\n{status_summary}\n\n"
-        "Available stages: brainstorming, planning, drawing, product_searching, purchasing."
+    guidance_prompt_text = (
+        "You are the Guiding Assistant for a DIY project. Your primary responsibility is to determine the next logical step "
+        "and then **obtain the user's explicit approval** before proceeding.\n\n"
+        f"CURRENT PROJECT STATUS:\n{status_summary}\n\n"
+        f"Based on the status, the most logical next stage appears to be: **'{proposed_next_stage}'**.\n\n"
+        "YOUR TASK:\n"
+        f"1. Formulate a clear and concise question for the user, asking if they agree to proceed to the **'{proposed_next_stage}'** stage. For example: 'It looks like '{proposed_next_stage}' is next. Shall we proceed with that?' or 'Are you ready to move on to {proposed_next_stage}?'\n"
+        "2. **You MUST use the 'human_assistance' tool to ask this question.** Do not answer directly. Invoke the tool with your question.\n"
+        "Example of tool usage (though you don't write this, you *call* the tool):\n"
+        "Tool Call: human_assistance(query='The next stage is planning. Do you want to proceed with planning?')\n\n"
+        "Consider the user's most recent message if it provides any preference."
     )
 
     if state.prompt:
-        final_prompt = "\n".join([guidance_prompt, state.prompt, ASSISTANT_SYSTEM_PROMPT_BASE])
+        final_prompt = "\n".join([guidance_prompt_text, state.prompt, ASSISTANT_SYSTEM_PROMPT_BASE])
     else:
-        final_prompt = "\n".join([guidance_prompt, ASSISTANT_SYSTEM_PROMPT_BASE])
+        final_prompt = "\n".join([guidance_prompt_text, ASSISTANT_SYSTEM_PROMPT_BASE])
 
     prompt = ChatPromptTemplate.from_messages(
         [
@@ -410,7 +431,37 @@ async def planning_node(state: GraphProcessingState, config=None):
         "messages": response
     }
 
+def custom_route_after_guidance(state: GraphProcessingState) -> Literal["execute_tools", "proceed_to_next_stage"]:
+    """
+    Checks the last message from the 'guidance_node'.
+    If it's an AIMessage with tool_calls, routes to 'execute_tools'.
+    Otherwise, routes to 'proceed_to_next_stage' (which could be a router itself
+    or directly to the state.next_stage node if it's set).
+    """
+    print("\n--- Custom Route After Guidance Condition ---")
+    messages = state.get("messages", [])
+    if not messages:
+        print("No messages found in state. Defaulting to proceed_to_next_stage.")
+        return "proceed_to_next_stage" # Or handle as an error/specific state
+
+    last_message = messages[-1]
+    print(f"Last message type: {type(last_message)}")
+
+    if isinstance(last_message, AIMessage):
+        if hasattr(last_message, "tool_calls") and last_message.tool_calls:
+            # Ensure tool_calls is not None and not an empty list
+            print(f"AIMessage has tool_calls: {last_message.tool_calls}")
+            return "execute_tools"
+        else:
+            print("AIMessage, but no tool_calls or tool_calls is empty.")
+            # If next_stage was set by guidance_node (e.g. after approval), we'd use that.
+            # For simplicity here, we just assume a generic "proceed"
+            return "proceed_to_next_stage"
+    else:
+        print(f"Last message is not an AIMessage (type: {type(last_message)}). Proceeding to next stage.")
+        return "proceed_to_next_stage"
 
+    print("--- End Custom Route After Guidance Condition ---")
 
 def guidance_routing(state: GraphProcessingState) -> str:
 
@@ -467,8 +518,14 @@ def define_workflow() -> CompiledStateGraph:
     # workflow.add_edge("tools", "brainstorming_node")
 
     workflow.add_conditional_edges(
-        "guidance_node",
-        tools_condition,
+    "guidance_node",  # The source node
+    custom_route_after_guidance,  # Your custom condition function
+        {
+            # "Path name": "Destination node name"
+            "execute_tools": "tools",  # If function returns "execute_tools"
+            "proceed_to_next_stage": "planning_node" # If function returns "proceed_to_next_stage"
+                                                            # Or this could be another router, or END
+        }
     )
     # workflow.add_conditional_edges("guidance_node", guidance_routing)
     # workflow.add_conditional_edges("brainstorming_node", brainstorming_routing)