From 133878afdb67901e43d7e06a577f4b0bc5f6755a Mon Sep 17 00:00:00 2001 From: Parker Erickson Date: Tue, 4 Jun 2024 17:40:22 -0500 Subject: [PATCH] fix agent graph when supportai not available --- app/agent/agent_graph.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/app/agent/agent_graph.py b/app/agent/agent_graph.py index 6e5af142..7a54f487 100644 --- a/app/agent/agent_graph.py +++ b/app/agent/agent_graph.py @@ -73,11 +73,14 @@ def route_question(self, state): return "apologize" state["question_retry_count"] += 1 logger.debug_pii(f"request_id={req_id_cv.get()} Routing question: {state['question']}") - source = step.route_question(state['question']) - logger.debug_pii(f"request_id={req_id_cv.get()} Routing question to: {source}") - if source.datasource == "vectorstore" and self.supportai_enabled: - return "supportai_lookup" - elif source.datasource == "functions": + if self.supportai_enabled: + source = step.route_question(state['question']) + logger.debug_pii(f"request_id={req_id_cv.get()} Routing question to: {source}") + if source.datasource == "vectorstore": + return "supportai_lookup" + elif source.datasource == "functions": + return "inquiryai_lookup" + else: return "inquiryai_lookup" def apologize(self, state):