Remove some logging/prints

Author: abrookins

agent_memory_server/api.py

@@ -308,12 +308,12 @@ async def put_working_memory(
     if user_id is not None:
         memory.user_id = user_id
 
-    # Validate that all structured memories have id (if any)
-    for mem in memory.memories:
-        if not mem.id:
+    # Validate that all long-term memories have id (if any)
+    for long_term_mem in memory.memories:
+        if not long_term_mem.id:
             raise HTTPException(
                 status_code=400,
-                detail="All memory records in working memory must have an ID",
+                detail="All long-term memory records in working memory must have an ID",
             )
 
     # Handle summarization if needed (before storing) - now token-based
@@ -453,17 +453,13 @@ async def search_long_term_memory(
     # Extract filter objects from the payload
     filters = payload.get_filters()
 
-    print("Long-term search filters: ", filters)
-
     kwargs = {
         "distance_threshold": payload.distance_threshold,
         "limit": payload.limit,
         "offset": payload.offset,
         **filters,
     }
 
-    print("Kwargs: ", kwargs)
-
     kwargs["text"] = payload.text or ""
 
     # Pass text and filter objects to the search function (no redis needed for vectorstore adapter)
@@ -568,8 +564,6 @@ async def memory_prompt(
     redis = await get_redis_conn()
     _messages = []
 
-    print("Received params: ", params)
-
     if params.session:
         # Use token limit for memory prompt, fallback to message count for backward compatibility
         if params.session.model_name or params.session.context_window_max:

agent_memory_server/cli.py

@@ -21,8 +21,12 @@
 from agent_memory_server.utils.redis import ensure_search_index_exists, get_redis_conn
 
 
-configure_logging()
-logger = get_logger(__name__)
+# Don't configure logging at module level - let each command handle it
+def _get_logger():
+    """Get logger instance after ensuring logging is configured."""
+    configure_logging()
+    return get_logger(__name__)
+
 
 VERSION = "0.2.0"
 
@@ -113,7 +117,7 @@ async def setup_and_run():
 
         # Run the MCP server
         if mode == "sse":
-            logger.info(f"Starting MCP server on port {port}\n")
+            _get_logger().info(f"Starting MCP server on port {port}\n")
             await mcp_app.run_sse_async()
         elif mode == "stdio":
             # Try to force all logging to stderr because stdio-mode MCP servers

agent_memory_server/vectorstore_adapter.py

@@ -832,16 +832,12 @@ async def search_memories(
             score_threshold = 1.0 - distance_threshold
             search_kwargs["score_threshold"] = score_threshold
 
-        print("Search kwargs: ", search_kwargs)
-
         search_results = (
             await self.vectorstore.asimilarity_search_with_relevance_scores(
                 **search_kwargs
             )
         )
 
-        print("Search results: ", search_results)
-
         # Convert results to MemoryRecordResult objects
         memory_results = []
         for i, (doc, score) in enumerate(search_results):