fix: abbreviation AA

app/api/v2_endpoints.py

@@ -154,7 +154,7 @@ async def handle_v2_query(
         original_query = request.query
         
         # --- EDIT: Call the new, direct replacement function ---
-        normalized_query = query_expansion_service.expand_tph_in_query(original_query)
+        normalized_query = query_expansion_service.replace_abbreviations(original_query)
         
         if original_query != normalized_query:
             logger.info(f"Query expanded from '{original_query}' to '{normalized_query}'")

app/services/query_expansion_service.py

@@ -164,6 +164,41 @@ def expand_tph_in_query(query_text: str) -> str:
     
     return re.sub(pattern, replacement, query_text, flags=re.IGNORECASE)
 
+def replace_abbreviations(query_text: str) -> str:
+    """
+    Expands a predefined set of abbreviations.
+    Specifically, "AA" is only replaced if "Arrangement" is not already in the query.
+    """
+    # 1. Define all replacement rules.
+    replacements = {
+        'tph': 'Payment Hub',
+        'aa': 'Arrangement'
+        # Add other unconditional replacements here.
+    }
+
+    # 2. Check if the word "Arrangement" is already in the query (case-insensitive).
+    #    We use a regex with \b to ensure we match the whole word.
+    if re.search(r'\bArrangement\b', query_text, re.IGNORECASE):
+        # If "Arrangement" is found, we don't want to replace "AA".
+        # So, we remove the 'aa' rule from our dictionary for this run.
+        del replacements['aa']
+
+    # 3. If there are no rules left to apply, return the original query.
+    if not replacements:
+        return query_text
+
+    # 4. Build the regex pattern ONLY with the rules that are still active.
+    pattern = re.compile(
+        r'\b(' + '|'.join(replacements.keys()) + r')\b', 
+        re.IGNORECASE
+    )
+
+    # 5. The replacer function remains the same.
+    def get_replacement(match):
+        return replacements[match.group(0).lower()]
+
+    # 6. Perform the substitution and return the result.
+    return pattern.sub(get_replacement, query_text)
 # async def generate_query_variations(query: str, num_variations: int = 2) -> List[str]:
 #     """
 #     Uses a local T5 model to generate paraphrases of the user's query.