Mehak-098 commited on
Commit
938dfb0
·
verified ·
1 Parent(s): 5bc49fe

Update agents/anomaly/backend/api_chat.py

Browse files
Files changed (1) hide show
  1. agents/anomaly/backend/api_chat.py +3 -0
agents/anomaly/backend/api_chat.py CHANGED
@@ -88,6 +88,8 @@ class AgentState(TypedDict):
88
  final_response: str
89
  check_all: bool
90
  force_retrain: bool
 
 
91
  agent_calls: List[str]
92
 
93
  async def supervisor_node(state: AgentState):
@@ -364,6 +366,7 @@ async def chat_handler(request: ChatRequest):
364
  "asset_name": request.asset_name,
365
  "next_node": "",
366
  "tool_outputs": [],
 
367
  "root_cause_analysis": "",
368
  "recommendations": "",
369
  "final_response": "",
 
88
  final_response: str
89
  check_all: bool
90
  force_retrain: bool
91
+ asset_id: str | None
92
+ asset_name: str | None
93
  agent_calls: List[str]
94
 
95
  async def supervisor_node(state: AgentState):
 
366
  "asset_name": request.asset_name,
367
  "next_node": "",
368
  "tool_outputs": [],
369
+ "agent_calls": [],
370
  "root_cause_analysis": "",
371
  "recommendations": "",
372
  "final_response": "",