feat: Generate action plan accounts for user's query and its language (#100)

Author: yoomlam

app/src/app_config.py

@@ -33,7 +33,7 @@ def db_session(self) -> db.Session:
     PROMPT_VERSIONS: dict = {
         "extract_supports": "UHJvbXB0VmVyc2lvbjo0Ng==",
         "generate_referrals": "UHJvbXB0VmVyc2lvbjo0OA==",
-        "generate_action_plan": "UHJvbXB0VmVyc2lvbjo1MQ==",
+        "generate_action_plan": "UHJvbXB0VmVyc2lvbjo1Mg==",
         "crawl_gcta": "UHJvbXB0VmVyc2lvbjozNg==",
         "crawl_indeed": "UHJvbXB0VmVyc2lvbjozNA==",
     }

app/src/pipelines/generate_action_plan/pipeline_wrapper.py

@@ -45,7 +45,8 @@ def setup(self) -> None:
         prompt_template = haystack_utils.get_phoenix_prompt("generate_action_plan")
         pipeline.add_component(
             instance=ChatPromptBuilder(
-                template=prompt_template, required_variables=["resources", "action_plan_json"]
+                template=prompt_template,
+                required_variables=["resources", "action_plan_json", "user_query"],
             ),
             name="prompt_builder",
         )
@@ -57,7 +58,9 @@ def setup(self) -> None:
         self.pipeline = pipeline
 
     # Called for the `generate-action-plan/run` endpoint
-    def run_api(self, resources: list[Resource] | list[dict], user_email: str) -> dict:
+    def run_api(
+        self, resources: list[Resource] | list[dict], user_email: str, user_query: str
+    ) -> dict:
         resource_objects = get_resources(resources)
 
         with using_attributes(user_id=user_email), using_metadata({"user_id": user_email}):
@@ -66,13 +69,13 @@ def run_api(self, resources: list[Resource] | list[dict], user_email: str) -> di
             with tracer.start_as_current_span(  # pylint: disable=not-context-manager,unexpected-keyword-arg
                 self.name, openinference_span_kind="chain"
             ) as span:
-                result = self._run(resource_objects, user_email)
+                result = self._run(resource_objects, user_email, user_query)
                 span.set_input([r.name for r in resource_objects])
                 span.set_output(result["response"])
                 span.set_status(Status(StatusCode.OK))
                 return result
 
-    def _run(self, resource_objects: list[Resource], user_email: str) -> dict:
+    def _run(self, resource_objects: list[Resource], user_email: str, user_query: str) -> dict:
         response = self.pipeline.run(
             {
                 "logger": {
@@ -83,6 +86,7 @@ def _run(self, resource_objects: list[Resource], user_email: str) -> dict:
                 "prompt_builder": {
                     "resources": format_resources(resource_objects),
                     "action_plan_json": action_plan_as_json,
+                    "user_query": user_query,
                 },
                 "llm": {"model": "gpt-5-mini", "reasoning_effort": "low"},
             },

frontend/src/app/[locale]/generate-referrals/page.tsx

@@ -162,7 +162,7 @@ export default function Page() {
 
     try {
       const { actionPlan: plan, errorMessage: planError } =
-        await fetchActionPlan(selectedResources, userEmail);
+        await fetchActionPlan(selectedResources, userEmail, clientDescription);
       setActionPlan(plan);
       if (planError) {
         setErrorMessage(planError);

frontend/src/util/fetchActionPlan.ts

@@ -52,6 +52,7 @@ function fixJsonControlCharacters(jsonString: string): string {
 export async function fetchActionPlan(
   resources: Resource[],
   userEmail: string,
+  userQuery: string,
 ): Promise<{ actionPlan: ActionPlan | null; errorMessage?: string }> {
   const apiDomain = await getApiDomain();
   const url = apiDomain + "generate_action_plan/run";
@@ -69,6 +70,7 @@ export async function fetchActionPlan(
       body: JSON.stringify({
         resources: resources,
         user_email: userEmail,
+        user_query: userQuery,
       }),
       cache: "no-store",
       signal: ac.signal,