def generate_query()

in datastore/providers/analyticdb_datastore.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

        def generate_query(query: QueryWithEmbedding) -> Tuple[str, List[Any]]:
            embedding = "[" + ", ".join(str(x) for x in query.embedding) + "]"
            q = f"""
                SELECT
                    id,
                    content,
                    source,
                    source_id,
                    document_id,
                    url,
                    created_at,
                    author,
                    embedding,
                    l2_distance(embedding,array{embedding}::real[]) AS similarity
                FROM
                    {self.collection_name}
            """
            where_clause, params = generate_where_clause(query.filter)
            q += where_clause
            q += f"ORDER BY embedding <-> array{embedding}::real[] LIMIT {query.top_k};"
            return q, params