Spaces:
Sleeping
Sleeping
| """Metadata search MCP tool implementations.""" | |
| from __future__ import annotations | |
| import json | |
| from metadata_search.search_service import document_ids_from_ranks, search_metadata, search_score_mode | |
| def search_metadata_tool( | |
| query: str, | |
| top_k: int = 10, | |
| year_from: int | None = None, | |
| year_to: int | None = None, | |
| ) -> str: | |
| results = search_metadata( | |
| query, | |
| top_k=top_k, | |
| year_from=year_from, | |
| year_to=year_to, | |
| ) | |
| return json.dumps( | |
| {"query": query, "results": results, "score_mode": search_score_mode()}, | |
| indent=2, | |
| ) | |
| def select_papers_from_search_tool( | |
| query: str, | |
| ranks: list[int], | |
| top_k: int = 10, | |
| year_from: int | None = None, | |
| year_to: int | None = None, | |
| ) -> str: | |
| results = search_metadata( | |
| query, | |
| top_k=top_k, | |
| year_from=year_from, | |
| year_to=year_to, | |
| ) | |
| document_ids = document_ids_from_ranks(results, ranks) | |
| selected = [item for item in results if int(item["rank"]) in ranks] | |
| return json.dumps( | |
| {"query": query, "document_ids": document_ids, "papers": selected}, | |
| indent=2, | |
| ) | |