retrieval_service/datastore/providers/spanner_gsql.py [655:682]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        with self.__database.snapshot() as snapshot:
            # Spread SQL query for readability
            query = """
                SELECT name, description, location, terminal, category, hour
                FROM (
                    SELECT name, description, location, terminal, category, hour,
                       COSINE_DISTANCE(embedding, @query_embedding) AS similarity
                    FROM amenities
                ) AS sorted_amenities
                WHERE (1 - similarity) > @similarity_threshold
                ORDER BY similarity
                LIMIT @top_k
            """

            # Execute SQL query with parameters
            results = snapshot.execute_sql(
                sql=query,
                params={
                    "query_embedding": query_embedding,
                    "similarity_threshold": similarity_threshold,
                    "top_k": top_k,
                },
                param_types={
                    "query_embedding": param_types.Array(param_types.FLOAT64),
                    "similarity_threshold": param_types.FLOAT64,
                    "top_k": param_types.INT64,
                },
            )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



retrieval_service/datastore/providers/spanner_gsql.py [968:993]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        with self.__database.snapshot() as snapshot:
            query = """
                SELECT content
                FROM (
                    SELECT content,  COSINE_DISTANCE(embedding, @query_embedding) AS similarity
                    FROM policies 
                ) AS sorted_policies
                WHERE (1 - similarity) > @similarity_threshold
                ORDER BY similarity
                LIMIT @top_k
            """

            # Execute SQL query with parameters
            results = snapshot.execute_sql(
                sql=query,
                params={
                    "query_embedding": query_embedding,
                    "similarity_threshold": similarity_threshold,
                    "top_k": top_k,
                },
                param_types={
                    "query_embedding": param_types.Array(param_types.FLOAT64),
                    "similarity_threshold": param_types.FLOAT64,
                    "top_k": param_types.INT64,
                },
            )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -