From 78666c695a458642520800bce6da40d624bb5699 Mon Sep 17 00:00:00 2001 From: luandro Date: Mon, 16 Dec 2024 11:21:02 -0300 Subject: [PATCH] wip: started adding product_support to classifier --- apps/ai_api/eda_ai_api/api/routes/classifier.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/apps/ai_api/eda_ai_api/api/routes/classifier.py b/apps/ai_api/eda_ai_api/api/routes/classifier.py index f281fac..6a0eb51 100644 --- a/apps/ai_api/eda_ai_api/api/routes/classifier.py +++ b/apps/ai_api/eda_ai_api/api/routes/classifier.py @@ -37,12 +37,13 @@ Choose between: - discovery: For finding grant opportunities - proposal: For writing grant proposals +- product_support: For product support - onboarding: For getting help using the system - heartbeat: For checking system health User message: {message} -Return only one word (discovery/proposal/onboarding/heartbeat):""" +Return only one word (discovery/proposal/onboarding/heartbeat/product_support):""" TOPIC_EXTRACTOR_TEMPLATE = """ Extract up to 5 most relevant topics for grant opportunity research from the user message. @@ -161,6 +162,8 @@ def process_decision(decision: str, message: str) -> Dict[str, Any]: .crew() .kickoff() ) + elif decision == "product_support": + return {"redirect": "/product_support"} elif decision == "heartbeat": return {"is_alive": True} elif decision == "onboarding":