stiff

app.log

@@ -50632,3 +50632,168 @@ Traceback (most recent call last):
   File "/home/user/app/graph.py", line 145, in guidance_node
     elif isinstance(message, SystemMessage):
 NameError: name 'SystemMessage' is not defined
+2025-06-06 21:03:31:__main__:INFO: Starting the interface
+2025-06-06 21:03:37:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 21:03:42:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 21:03:42:__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, SystemMessage):
+NameError: name 'SystemMessage' is not defined
+2025-06-06 21:03:50:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 21:03:53:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 21:04:10:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 21:04:26:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 21:05:15:__main__:INFO: Starting the interface
+2025-06-06 21:05:20:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 21:05:25:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 21:05:25:__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 149, in guidance_node
+    elif isinstance(message, ToolMessage):
+NameError: name 'ToolMessage' is not defined. Did you mean: 'AnyMessage'?
+2025-06-06 21:05:56:__main__:INFO: Starting the interface
+2025-06-06 21:06:01:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 21:06:17:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 21:06:17:__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 678, in ainvoke
+    input = await step.ainvoke(input, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 440, in ainvoke
+    ret = await self.afunc(*args, **kwargs)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/graph/branch.py", line 196, in _aroute
+    result = await self.path.ainvoke(value, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 433, in ainvoke
+    ret = await coro
+  File "/usr/local/lib/python3.10/site-packages/langchain_core/runnables/config.py", line 616, in run_in_executor
+    return await asyncio.get_running_loop().run_in_executor(
+  File "/usr/local/lib/python3.10/concurrent/futures/thread.py", line 58, in run
+    result = self.fn(*self.args, **self.kwargs)
+  File "/usr/local/lib/python3.10/site-packages/langchain_core/runnables/config.py", line 607, in wrapper
+    return func(*args, **kwargs)
+  File "/home/user/app/graph.py", line 442, in custom_route_after_guidance
+    messages = state.get("messages", [])
+  File "/usr/local/lib/python3.10/site-packages/pydantic/main.py", line 891, in __getattr__
+    raise AttributeError(f'{type(self).__name__!r} object has no attribute {item!r}')
+AttributeError: 'GraphProcessingState' object has no attribute 'get'
+2025-06-06 21:09:02:__main__:INFO: Starting the interface
+2025-06-06 21:09:07:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 21:09:46:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 21:09:46:__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 678, in ainvoke
+    input = await step.ainvoke(input, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 440, in ainvoke
+    ret = await self.afunc(*args, **kwargs)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/graph/branch.py", line 196, in _aroute
+    result = await self.path.ainvoke(value, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 433, in ainvoke
+    ret = await coro
+  File "/usr/local/lib/python3.10/site-packages/langchain_core/runnables/config.py", line 616, in run_in_executor
+    return await asyncio.get_running_loop().run_in_executor(
+  File "/usr/local/lib/python3.10/concurrent/futures/thread.py", line 58, in run
+    result = self.fn(*self.args, **self.kwargs)
+  File "/usr/local/lib/python3.10/site-packages/langchain_core/runnables/config.py", line 607, in wrapper
+    return func(*args, **kwargs)
+  File "/home/user/app/graph.py", line 443, in custom_route_after_guidance
+    if not messages:
+NameError: name 'messages' is not defined. Did you mean: 'essages'?
+2025-06-06 21:10:22:__main__:INFO: Starting the interface
+2025-06-06 21:10:28:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 21:11:30:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 21:11:30:__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 428, in planning_node
+    response = await chain.ainvoke({"messages": state.messages}, config=config)
+  File "/usr/local/lib/python3.10/site-packages/langchain_core/runnables/base.py", line 3089, in ainvoke
+    input_ = await coro_with_context(part(), context, create_task=True)
+  File "/usr/local/lib/python3.10/site-packages/langchain_core/runnables/base.py", line 5444, in ainvoke
+    return await self.bound.ainvoke(
+  File "/usr/local/lib/python3.10/site-packages/langchain_core/language_models/chat_models.py", line 394, in ainvoke
+    llm_result = await self.agenerate_prompt(
+  File "/usr/local/lib/python3.10/site-packages/langchain_core/language_models/chat_models.py", line 968, in agenerate_prompt
+    return await self.agenerate(
+  File "/usr/local/lib/python3.10/site-packages/langchain_core/language_models/chat_models.py", line 926, in agenerate
+    raise exceptions[0]
+  File "/usr/local/lib/python3.10/site-packages/langchain_core/language_models/chat_models.py", line 1083, in _agenerate_with_cache
+    async for chunk in self._astream(messages, stop=stop, **kwargs):
+  File "/usr/local/lib/python3.10/site-packages/langchain_anthropic/chat_models.py", line 1394, in _astream
+    _handle_anthropic_bad_request(e)
+  File "/usr/local/lib/python3.10/site-packages/langchain_anthropic/chat_models.py", line 1374, in _astream
+    stream = await self._acreate(payload)
+  File "/usr/local/lib/python3.10/site-packages/langchain_anthropic/chat_models.py", line 1322, in _acreate
+    return await self._async_client.messages.create(**payload)
+  File "/usr/local/lib/python3.10/site-packages/anthropic/resources/messages/messages.py", line 2229, in create
+    return await self._post(
+  File "/usr/local/lib/python3.10/site-packages/anthropic/_base_client.py", line 1819, in post
+    return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls)
+  File "/usr/local/lib/python3.10/site-packages/anthropic/_base_client.py", line 1626, in request
+    raise self._make_status_error_from_response(err.response) from None
+anthropic.BadRequestError: Error code: 400 - {'type': 'error', 'error': {'type': 'invalid_request_error', 'message': 'messages.0.content.1: unexpected `tool_use_id` found in `tool_result` blocks: toolu_01MvUi4ZEnFZ2cN39bLVinQ2. Each `tool_result` block must have a corresponding `tool_use` block in the previous message.'}}
+2025-06-06 21:18:33:__main__:INFO: Starting the interface
+2025-06-06 21:18:38:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 21:18:43:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 21:27:34:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 21:27:48:__main__:INFO: Prompt: You are a helpful assistant.

graph.py

@@ -8,7 +8,7 @@ import uuid
 
 
 import aiohttp
-from langchain_core.messages import AIMessage, HumanMessage, AnyMessage, ToolCall
+from langchain_core.messages import AIMessage, HumanMessage, AnyMessage, ToolCall, SystemMessage, ToolMessage
 
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.tools import tool
@@ -23,7 +23,7 @@ from langgraph.prebuilt import ToolNode
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.types import Command, interrupt
 
-from typing import TypedDict, List, Optional
+from typing import TypedDict, List, Optional, Literal
 
 class State(TypedDict):
     messages: Annotated[list, add_messages]
@@ -129,121 +129,146 @@ class GraphProcessingState(BaseModel):
 
 
 
-async def guidance_node(state: GraphProcessingState, config=None):
-    print("\n--- Guidance Node (Debug via print) ---")
+# async def guidance_node(state: GraphProcessingState, config=None):
+#     print("\n--- Guidance Node (Debug via print) ---")
 
-    print(f"Prompt: {state.prompt}")
-    for message in state.messages:
-        if isinstance(message, HumanMessage):
-            print(f"Human: {message.content}")
-        elif isinstance(message, AIMessage):
-            if message.content:
-                if isinstance(message.content, list):
-                    texts = [item.get('text', '') for item in message.content if isinstance(item, dict) and 'text' in item]
-                    if texts:
-                        print(f"AI: {' '.join(texts)}")
-                elif isinstance(message.content, str):
-                    print(f"AI: {message.content}")
-        elif isinstance(message, SystemMessage):
-            print(f"System: {message.content}")
-        elif isinstance(message, ToolMessage):
-            print(f"Tool: {message.content}")
+#     print(f"Prompt: {state.prompt}")
+#     for message in state.messages:
+#         if isinstance(message, HumanMessage):
+#             print(f"Human: {message.content}")
+#         elif isinstance(message, AIMessage):
+#             if message.content:
+#                 if isinstance(message.content, list):
+#                     texts = [item.get('text', '') for item in message.content if isinstance(item, dict) and 'text' in item]
+#                     if texts:
+#                         print(f"AI: {' '.join(texts)}")
+#                 elif isinstance(message.content, str):
+#                     print(f"AI: {message.content}")
+#         elif isinstance(message, SystemMessage):
+#             print(f"System: {message.content}")
+#         elif isinstance(message, ToolMessage):
+#             print(f"Tool: {message.content}")
 
-    print(f"Tools Enabled: {state.tools_enabled}")
-    print(f"Search Enabled: {state.search_enabled}")
-    print(f"Next Stage: {state.next_stage}")
+#     print(f"Tools Enabled: {state.tools_enabled}")
+#     print(f"Search Enabled: {state.search_enabled}")
+#     print(f"Next Stage: {state.next_stage}")
 
 
-    print(f"Brainstorming Complete: {state.brainstorming_complete}")
+#     print(f"Brainstorming Complete: {state.brainstorming_complete}")
     
 
-    guidance_node.count = getattr(guidance_node, 'count', 0) + 1
-    print('\nGuidance Node called count', guidance_node.count)
-    print("\n--- End Guidance Node Debug ---")
+#     guidance_node.count = getattr(guidance_node, 'count', 0) + 1
+#     print('\nGuidance Node called count', guidance_node.count)
+#     print("\n--- End Guidance Node Debug ---")
+
+#     stage_order = ["brainstorming", "planning", "drawing", "product_searching", "purchasing"]
+#     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_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_text, state.prompt, ASSISTANT_SYSTEM_PROMPT_BASE])
+#     else:
+#         final_prompt = "\n".join([guidance_prompt_text, ASSISTANT_SYSTEM_PROMPT_BASE])
+
+#     prompt = ChatPromptTemplate.from_messages(
+#         [
+#             ("system", final_prompt),
+#             MessagesPlaceholder(variable_name="messages"),
+#         ]
+#     )
+
+#     assistant_model = model.bind_tools([human_assistance])
 
+#     chain = prompt | assistant_model
+
+#     try:
+#         response = await chain.ainvoke({"messages": state.messages}, config=config)
+
+#         for msg in response:
+#             if isinstance(msg, HumanMessage):
+#                 print("Human:", msg.content)
+#             elif isinstance(msg, AIMessage):
+#                 if isinstance(msg.content, list):
+#                     ai_texts = [part.get("text", "") for part in msg.content if isinstance(part, dict)]
+#                     print("AI:", " ".join(ai_texts))
+#                 else:
+#                     print("AI:", msg.content)
+
+#         # Check for tool calls in the response
+#         if hasattr(response, "tool_calls"):
+#             for tool_call in response.tool_calls:
+#                 tool_name = tool_call['name']
+#                 if tool_name == "human_assistance":
+#                     query = tool_call['args']['query']
+#                     print(f"Human input needed: {query}")
+#                     # Handle human assistance tool call
+#                     # You can pause execution and wait for user input here
+
+#         return {
+#             "messages": [response],
+#             "next_stage": incomplete[0] if incomplete else "brainstorming"
+#         }
+#     except Exception as e:
+#         print(f"Error in guidance node: {e}")
+#         return {
+#             "messages": [AIMessage(content="Error in guidance node.")],
+#             "next_stage": "brainstorming"
+#         }
+
+async def guidance_node(state: GraphProcessingState, config=None):
+    # Define the order of stages
     stage_order = ["brainstorming", "planning", "drawing", "product_searching", "purchasing"]
+    
+    # Identify completed and incomplete stages
     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)]
-
+    
+    # Determine the next stage
     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!")
+        # All stages are complete
         return {
-            "messages": current_messages + [ai_all_complete_msg],
-            "next_stage": "end_project", # Or None, or a final summary node
+            "messages": [AIMessage(content="All DIY project stages are complete!")],
+            "next_stage": "end_project",
             "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_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_text, state.prompt, ASSISTANT_SYSTEM_PROMPT_BASE])
-    else:
-        final_prompt = "\n".join([guidance_prompt_text, ASSISTANT_SYSTEM_PROMPT_BASE])
-
-    prompt = ChatPromptTemplate.from_messages(
-        [
-            ("system", final_prompt),
-            MessagesPlaceholder(variable_name="messages"),
-        ]
-    )
-
-    assistant_model = model.bind_tools([human_assistance])
-
-    chain = prompt | assistant_model
-
-    try:
-        response = await chain.ainvoke({"messages": state.messages}, config=config)
-
-        for msg in response:
-            if isinstance(msg, HumanMessage):
-                print("Human:", msg.content)
-            elif isinstance(msg, AIMessage):
-                if isinstance(msg.content, list):
-                    ai_texts = [part.get("text", "") for part in msg.content if isinstance(part, dict)]
-                    print("AI:", " ".join(ai_texts))
-                else:
-                    print("AI:", msg.content)
-
-        # Check for tool calls in the response
-        if hasattr(response, "tool_calls"):
-            for tool_call in response.tool_calls:
-                tool_name = tool_call['name']
-                if tool_name == "human_assistance":
-                    query = tool_call['args']['query']
-                    print(f"Human input needed: {query}")
-                    # Handle human assistance tool call
-                    # You can pause execution and wait for user input here
-
-        return {
-            "messages": [response],
-            "next_stage": incomplete[0] if incomplete else "brainstorming"
-        }
-    except Exception as e:
-        print(f"Error in guidance node: {e}")
+        # Set the next stage to the first incomplete stage
+        next_stage = incomplete[0]
         return {
-            "messages": [AIMessage(content="Error in guidance node.")],
-            "next_stage": "brainstorming"
+            "messages": [],
+            "next_stage": next_stage,
+            "pending_approval_stage": None,
         }
 
 async def brainstorming_node(state: GraphProcessingState, config=None):
@@ -439,7 +464,7 @@ def custom_route_after_guidance(state: GraphProcessingState) -> Literal["execute
     or directly to the state.next_stage node if it's set).
     """
     print("\n--- Custom Route After Guidance Condition ---")
-    messages = state.get("messages", [])
+    messages = state['messages'] if 'messages' in state else []
     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
@@ -515,18 +540,24 @@ def define_workflow() -> CompiledStateGraph:
     # workflow.add_edge("planning_node", "guidance_node")
     # workflow.add_edge("brainstorming_node", "guidance_node")
     # workflow.add_edge("brainstorming_node", "tools")
-    # workflow.add_edge("tools", "brainstorming_node")
-
+    workflow.add_edge("tools", "guidance_node")
     workflow.add_conditional_edges(
-    "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
-        }
+        "guidance_node",
+        guidance_routing,
+            {
+                "brainstorming_node" : "brainstorming_node",
+            }
     )
+    # workflow.add_conditional_edges(
+    # "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)