diff --git a/mempalace/mcp_server.py b/mempalace/mcp_server.py index 94cbf99..2169255 100644 --- a/mempalace/mcp_server.py +++ b/mempalace/mcp_server.py @@ -68,7 +68,7 @@ def tool_status(): wings = {} rooms = {} try: - all_meta = col.get(include=["metadatas"])["metadatas"] + all_meta = col.get(include=["metadatas"], limit=10000)["metadatas"] for m in all_meta: w = m.get("wing", "unknown") r = m.get("room", "unknown") @@ -125,7 +125,7 @@ def tool_list_wings(): return _no_palace() wings = {} try: - all_meta = col.get(include=["metadatas"])["metadatas"] + all_meta = col.get(include=["metadatas"], limit=10000)["metadatas"] for m in all_meta: w = m.get("wing", "unknown") wings[w] = wings.get(w, 0) + 1 @@ -140,7 +140,7 @@ def tool_list_rooms(wing: str = None): return _no_palace() rooms = {} try: - kwargs = {"include": ["metadatas"]} + kwargs = {"include": ["metadatas"], "limit": 10000} if wing: kwargs["where"] = {"wing": wing} all_meta = col.get(**kwargs)["metadatas"] @@ -158,7 +158,7 @@ def tool_get_taxonomy(): return _no_palace() taxonomy = {} try: - all_meta = col.get(include=["metadatas"])["metadatas"] + all_meta = col.get(include=["metadatas"], limit=10000)["metadatas"] for m in all_meta: w = m.get("wing", "unknown") r = m.get("room", "unknown") @@ -406,6 +406,7 @@ def tool_diary_read(agent_name: str, last_n: int = 10): results = col.get( where={"$and": [{"wing": wing}, {"room": "diary"}]}, include=["documents", "metadatas"], + limit=10000, ) if not results["ids"]: