review comments

Author: esoteric-ephemera

mp_api/mcp/tools.py

@@ -170,7 +170,13 @@ def fetch(self, idx: str) -> FetchResult:
         metadata: dict[str, str] | None = None
         if (
             len(
-                summary_docs := self.client.materials.summary.search(material_ids=[idx])
+                summary_docs := self.client.materials.summary.search(
+                    material_ids=[idx],
+                    fields=list(
+                        set(MaterialMetadata.model_fields)
+                        & set(self.client.materials.summary.document_model.model_fields)
+                    ),
+                )
             )
             > 0
         ):

tests/mcp/test_tools.py

@@ -20,7 +20,7 @@ def test_core_tools():
     with MPCoreMCP() as mcp_tools:
         search_results = mcp_tools.search("Ga-W")
         fetch_results = mcp_tools.fetch("Ir2 Br6")
-        robo_desc_docs = mcp_tools.client.robocrys.search_docs(
+        robo_desc_docs = mcp_tools.client.materials.robocrys.search_docs(
             material_ids=[*[doc.id for doc in search_results.results], fetch_results.id]
         )