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
]