retrieval_service/datastore/providers/spanner_postgres.py [656:683]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        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,
                       spanner.cosine_distance(embedding, $1) AS similarity
                    FROM amenities
                ) AS sorted_amenities
                WHERE (1 - similarity) > $2
                ORDER BY similarity
                LIMIT $3
            """

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



retrieval_service/datastore/providers/spanner_postgres.py [992:1017]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        with self.__database.snapshot() as snapshot:
            query = """
                SELECT content
                FROM (
                    SELECT content,  spanner.cosine_distance(embedding, $1) AS similarity
                    FROM policies 
                ) AS sorted_policies
                WHERE (1 - similarity) > $2
                ORDER BY similarity
                LIMIT $3
            """

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