Fix scheme chatbot chain bug

agents/scheme_chatbot.py

@@ -44,16 +44,20 @@ prompt = PromptTemplate(
 )
 
 # --- Build Chain ---
-def get_search_results(query: dict):
-    print(f"---NODE: Searching Schemes for profile: {query['user_profile']}---")
+def get_search_results(input_data):
+    user_profile = input_data.get("user_profile", "") if isinstance(input_data, dict) else input_data
+    print(f"---NODE: Searching Schemes for profile: {user_profile}---")
     try:
-        return scheme_search.invoke(query["user_profile"])
+        return scheme_search.invoke(user_profile)
     except Exception as e:
         print(f"Scheme search failed: {e}")
         return "Search unavailable."
 
+def extract_user_profile(input_data):
+    return input_data.get("user_profile", "") if isinstance(input_data, dict) else input_data
+
 scheme_chatbot = (
-    {"search_results": get_search_results, "user_profile": RunnablePassthrough()}
+    {"search_results": get_search_results, "user_profile": extract_user_profile}
     | prompt
     | llm
     | parser