Update agents/langgraph.py (#12)

- Update agents/langgraph.py (a4b44edc641ff7076fb5035f7f73290bba6d78e2)


Co-authored-by: sahilgarg <sahil-1-garg@users.noreply.huggingface.co>

agents/langgraph.py

@@ -1,4 +1,4 @@
-from typing import TypedDict, Dict, Any, List, Annotated
+from typing import TypedDict, Dict, Any, List, Annotated, Optional
 import time, uuid, os
 from langgraph.graph import StateGraph, END
 from langchain_core.messages import HumanMessage, AIMessage, BaseMessage
@@ -18,6 +18,8 @@ class FinancialAgentState(TypedDict):
     start_time: float
     end_time: float
     error: str
+    user_api_key: Optional[str]
+    feedback_context: Optional[Dict[str, Any]]
 
 def make_workflow(tool_func):
     def node(state: FinancialAgentState) -> FinancialAgentState:
@@ -65,6 +67,8 @@ def run_workflow(file_path: str, kind: str, **kwargs) -> Dict[str, Any]:
         start_time=0,
         end_time=0,
         error="",
+        user_api_key=None,
+        feedback_context=None,
     )
     # Add feedback_context if provided
     if "feedback_context" in kwargs: