Method _extract_query_texts

AskAgent/python/core/baseHandler.py:244–281

Extract just the query text from previous queries. Handles both simple string lists and complex nested structures.

(self, raw_prev_queries)

        return cls(query_params, http_handler)

    def _extract_query_texts(self, raw_prev_queries):
        """
        Extract just the query text from previous queries.
        Handles both simple string lists and complex nested structures.
        """
        if not raw_prev_queries:
            return []

        query_texts = []

        for item in raw_prev_queries:
            if isinstance(item, str):
                # Try to parse as JSON if it looks like JSON
                if item.strip().startswith('[') or item.strip().startswith('{'):
                    try:
                        import json
                        parsed = json.loads(item)
                        # Recursively extract from parsed JSON
                        extracted = self._extract_from_parsed(parsed)
                        query_texts.extend(extracted)
                    except json.JSONDecodeError:
                        # If not JSON, just add the string
                        query_texts.append(item)
                else:
                    query_texts.append(item)
            elif isinstance(item, dict):
                # Extract query from dict structure
                if 'query' in item:
                    if isinstance(item['query'], dict) and 'query' in item['query']:
                        query_texts.append(item['query']['query'])
                    elif isinstance(item['query'], str):
                        query_texts.append(item['query'])
            elif isinstance(item, list):
                # Recursively extract from list
                extracted = self._extract_from_parsed(item)
                query_texts.extend(extracted)

        return query_texts

    def _extract_from_parsed(self, parsed):
        """Helper to extract query texts from parsed JSON structures."""