浏览代码

Store entity metadata in related topic snapshots

Lukas Goldschmidt 1 月之前
父节点
当前提交
9f5b5a791b
共有 1 个文件被更改,包括 6 次插入4 次删除
  1. 6 4
      trends_mcp/mcp_server_fastmcp.py

+ 6 - 4
trends_mcp/mcp_server_fastmcp.py

@@ -93,6 +93,7 @@ async def get_related_queries(keyword: str):
     cached = get_cache(cache_key)
     if cached:
         return cached
+    entity_info = await resolve_entity(keyword)
     related = provider.related_queries(keyword)
     out = related.get(keyword) or related.get(normalize_entity(keyword)) or {}
 
@@ -114,8 +115,8 @@ async def get_related_queries(keyword: str):
         tool="get_related_queries",
         keyword=keyword,
         normalized_keyword=normalize_entity(keyword),
-        mid=None,
-        canonical_label=None,
+        mid=entity_info.get("mid"),
+        canonical_label=entity_info.get("canonical_label"),
         payload=payload,
     )
     return payload
@@ -127,6 +128,7 @@ async def get_related_topics(keyword: str):
     cached = get_cache(cache_key)
     if cached:
         return cached
+    entity_info = await resolve_entity(keyword)
     try:
         related = provider.related_topics(keyword)
         out = related.get(keyword) or related.get(normalize_entity(keyword)) or {}
@@ -153,8 +155,8 @@ async def get_related_topics(keyword: str):
         tool="get_related_topics",
         keyword=keyword,
         normalized_keyword=normalize_entity(keyword),
-        mid=None,
-        canonical_label=None,
+        mid=entity_info.get("mid"),
+        canonical_label=entity_info.get("canonical_label"),
         payload=payload,
     )
     return payload