|
@@ -93,6 +93,7 @@ async def get_related_queries(keyword: str):
|
|
|
cached = get_cache(cache_key)
|
|
cached = get_cache(cache_key)
|
|
|
if cached:
|
|
if cached:
|
|
|
return cached
|
|
return cached
|
|
|
|
|
+ entity_info = await resolve_entity(keyword)
|
|
|
related = provider.related_queries(keyword)
|
|
related = provider.related_queries(keyword)
|
|
|
out = related.get(keyword) or related.get(normalize_entity(keyword)) or {}
|
|
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",
|
|
tool="get_related_queries",
|
|
|
keyword=keyword,
|
|
keyword=keyword,
|
|
|
normalized_keyword=normalize_entity(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,
|
|
payload=payload,
|
|
|
)
|
|
)
|
|
|
return payload
|
|
return payload
|
|
@@ -127,6 +128,7 @@ async def get_related_topics(keyword: str):
|
|
|
cached = get_cache(cache_key)
|
|
cached = get_cache(cache_key)
|
|
|
if cached:
|
|
if cached:
|
|
|
return cached
|
|
return cached
|
|
|
|
|
+ entity_info = await resolve_entity(keyword)
|
|
|
try:
|
|
try:
|
|
|
related = provider.related_topics(keyword)
|
|
related = provider.related_topics(keyword)
|
|
|
out = related.get(keyword) or related.get(normalize_entity(keyword)) or {}
|
|
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",
|
|
tool="get_related_topics",
|
|
|
keyword=keyword,
|
|
keyword=keyword,
|
|
|
normalized_keyword=normalize_entity(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,
|
|
payload=payload,
|
|
|
)
|
|
)
|
|
|
return payload
|
|
return payload
|