Update agents/langgraph.py

agents/langgraph.py

@@ -28,6 +28,9 @@ def make_workflow(tool_func):
             # Add feedback_context if available
             if "feedback_context" in state:
                 tool_params["feedback_context"] = state["feedback_context"]
+            # Add user_api_key if available
+            if "user_api_key" in state:
+                tool_params["user_api_key"] = state["user_api_key"]
             # Use .invoke() to avoid deprecation warning
             result = tool_func.invoke(tool_params)
             state["result"] = result
@@ -65,6 +68,9 @@ def run_workflow(file_path: str, kind: str, **kwargs) -> Dict[str, Any]:
     )
     # Add feedback_context if provided
     if "feedback_context" in kwargs:
-        state["feedback_context"] = kwargs["feedback_context"]   
+        state["feedback_context"] = kwargs["feedback_context"]
+    # Add user_api_key if provided
+    if "user_api_key" in kwargs:
+        state["user_api_key"] = kwargs["user_api_key"]
     final = workflows[kind].invoke(state)
     return final