def search()

in tablestore-python-mcp-server/src/tablestore_mcp_server/tablestore_connector.py [0:0]


    def search(self, query: str, size: int = 20) -> list[Entry]:
        logger.info(f"Search query: {query}, size: {size}")
        # Embed the query
        query_embedding = self._embedding.get_query_embedding(query)

        # Search in tablestore
        search_results = self._store.query(
            query=VectorStoreQuery(
                mode=VectorStoreQueryMode.DEFAULT,
                query_embedding=query_embedding,
                query_str=query,
                similarity_top_k=size,
            ),
            knn_top_k=min(1000, size + 100),
        )
        return [
            Entry(
                content=node.text,
                metadata=node.metadata,
            )
            for node in search_results.nodes
        ]