updated prompt

src/backend/config/tools.py

@@ -33,7 +33,7 @@ class ToolName(StrEnum):
     Python_Interpreter = "Python_Interpreter"
     Calculator = "Calculator"
     Tavily_Internet_Search = "Internet_Search"
-    MiniMap = "MiniMap"
+    Minimap = "Minimap"
 
 
 ALL_TOOLS = {
@@ -54,12 +54,12 @@ class ToolName(StrEnum):
         category=Category.DataLoader,
         description="Retrieves documents from Wikipedia using LangChain.",
     ),
-    ToolName.MiniMap: ManagedTool(
-        name=ToolName.MiniMap,
+    ToolName.Minimap: ManagedTool(
+        name=ToolName.Minimap,
         implementation=LangChainMinimapRetriever,
         parameter_definitions={
             "query": {
-                "description": "Query for searching the news. It can be a topic, a named entities, or a quoted keyword.",
+                "description": "Query for searching the news. It can be a topic, a named entities, or a quoted keyword. Summarizations should be in the form of a bulleted list where each bullet point is its own core topic.",
                 "type": "str",
                 "required": True,
             }

src/backend/tools/minimap.py

@@ -64,7 +64,7 @@ class MinimapAPIWrapper(BaseModel):
     max_retry: int = 5
 
     # Default values for the parameters
-    top_k_results: int = 15
+    top_k_results: int = 50
     MAX_QUERY_LENGTH: int = 300
     doc_content_chars_max: int = 2000
 
@@ -140,5 +140,9 @@ def is_available(cls) -> bool:
     def call(self, parameters: dict, **kwargs: Any) -> List[Dict[str, Any]]:
         query = parameters.get("query", "")
         results = self.client.run(query)
-        print(results)
+
+        # remap `id` to `document_id`
+        for result in results:
+            result["document_id"] = result.pop("id")
+
         return [dict(result) for result in results]