fix: Add proper datetime JSON schema format annotations for MCP validation (#312)

Signed-off-by: Claude Code <[email protected]>
Signed-off-by: phernandez <[email protected]>
Co-authored-by: claude[bot] <41898282+claude[bot]@users.noreply.github.com>
Co-authored-by: Paul Hernandez <[email protected]>

Author: phernandez

src/basic_memory/mcp/prompts/utils.py

@@ -103,10 +103,17 @@ def format_prompt_context(context: PromptContext) -> str:
 
                 added_permalinks.add(primary_permalink)
 
-                memory_url = normalize_memory_url(primary_permalink)
+                # Use permalink if available, otherwise use file_path
+                if primary_permalink:
+                    memory_url = normalize_memory_url(primary_permalink)
+                    read_command = f'read_note("{primary_permalink}")'
+                else:
+                    memory_url = f"file://{primary.file_path}"
+                    read_command = f'read_file("{primary.file_path}")'
+
                 section = dedent(f"""
                     --- {memory_url}
-                
+
                     ## {primary.title}
                     - **Type**: {primary.type}
                     """)
@@ -121,8 +128,8 @@ def format_prompt_context(context: PromptContext) -> str:
                         section += f"\n**Excerpt**:\n{content}\n"
 
                 section += dedent(f"""
-    
-                    You can read this document with: `read_note("{primary_permalink}")`
+
+                    You can read this document with: `{read_command}`
                     """)
                 sections.append(section)
 

src/basic_memory/schemas/memory.py

@@ -26,6 +26,7 @@ def validate_memory_url_path(path: str) -> bool:
         >>> validate_memory_url_path("invalid://test")  # Contains protocol
         False
     """
+    # Empty paths are not valid
     if not path or not path.strip():
         return False
 
@@ -68,7 +69,13 @@ def normalize_memory_url(url: str | None) -> str:
         ValueError: Invalid memory URL path: 'memory//test' contains double slashes
     """
     if not url:
-        return ""
+        raise ValueError("Memory URL cannot be empty")
+
+    # Strip whitespace for consistency
+    url = url.strip()
+
+    if not url:
+        raise ValueError("Memory URL cannot be empty or whitespace")
 
     clean_path = url.removeprefix("memory://")
 
@@ -79,8 +86,6 @@ def normalize_memory_url(url: str | None) -> str:
             raise ValueError(f"Invalid memory URL path: '{clean_path}' contains protocol scheme")
         elif "//" in clean_path:
             raise ValueError(f"Invalid memory URL path: '{clean_path}' contains double slashes")
-        elif not clean_path.strip():
-            raise ValueError("Memory URL path cannot be empty or whitespace")
         else:
             raise ValueError(f"Invalid memory URL path: '{clean_path}' contains invalid characters")
 
@@ -123,7 +128,9 @@ class EntitySummary(BaseModel):
     title: str
     content: Optional[str] = None
     file_path: str
-    created_at: datetime
+    created_at: Annotated[
+        datetime, Field(json_schema_extra={"type": "string", "format": "date-time"})
+    ]
 
     @field_serializer("created_at")
     def serialize_created_at(self, dt: datetime) -> str:
@@ -140,7 +147,9 @@ class RelationSummary(BaseModel):
     relation_type: str
     from_entity: Optional[str] = None
     to_entity: Optional[str] = None
-    created_at: datetime
+    created_at: Annotated[
+        datetime, Field(json_schema_extra={"type": "string", "format": "date-time"})
+    ]
 
     @field_serializer("created_at")
     def serialize_created_at(self, dt: datetime) -> str:
@@ -156,7 +165,9 @@ class ObservationSummary(BaseModel):
     permalink: str
     category: str
     content: str
-    created_at: datetime
+    created_at: Annotated[
+        datetime, Field(json_schema_extra={"type": "string", "format": "date-time"})
+    ]
 
     @field_serializer("created_at")
     def serialize_created_at(self, dt: datetime) -> str:
@@ -170,7 +181,9 @@ class MemoryMetadata(BaseModel):
     types: Optional[List[SearchItemType]] = None
     depth: int
     timeframe: Optional[str] = None
-    generated_at: datetime
+    generated_at: Annotated[
+        datetime, Field(json_schema_extra={"type": "string", "format": "date-time"})
+    ]
     primary_count: Optional[int] = None  # Changed field name
     related_count: Optional[int] = None  # Changed field name
     total_results: Optional[int] = None  # For backward compatibility
@@ -235,9 +248,9 @@ class ProjectActivity(BaseModel):
     project_path: str
     activity: GraphContext = Field(description="The actual activity data for this project")
     item_count: int = Field(description="Total items in this project's activity")
-    last_activity: Optional[datetime] = Field(
-        default=None, description="Most recent activity timestamp"
-    )
+    last_activity: Optional[
+        Annotated[datetime, Field(json_schema_extra={"type": "string", "format": "date-time"})]
+    ] = Field(default=None, description="Most recent activity timestamp")
     active_folders: List[str] = Field(default_factory=list, description="Most active folders")
 
     @field_serializer("last_activity")
@@ -253,7 +266,9 @@ class ProjectActivitySummary(BaseModel):
     )
     summary: ActivityStats
     timeframe: str = Field(description="The timeframe used for the query")
-    generated_at: datetime
+    generated_at: Annotated[
+        datetime, Field(json_schema_extra={"type": "string", "format": "date-time"})
+    ]
     guidance: Optional[str] = Field(
         default=None, description="Assistant guidance for project selection and session management"
     )

test-int/mcp/test_build_context_validation.py

@@ -70,7 +70,7 @@ async def test_build_context_empty_urls_fail_validation(mcp_server, app, test_pr
     """Test that empty or whitespace-only URLs fail validation."""
 
     async with Client(mcp_server) as client:
-        # These should fail MinLen validation
+        # These should fail validation
         empty_urls = [
             "",  # Empty string
             "   ",  # Whitespace only
@@ -83,10 +83,9 @@ async def test_build_context_empty_urls_fail_validation(mcp_server, app, test_pr
                 )
 
             error_message = str(exc_info.value)
-            # Should fail with validation error (either MinLen or our custom validation)
+            # Should fail with validation error
             assert (
-                "at least 1" in error_message
-                or "too_short" in error_message
+                "cannot be empty" in error_message
                 or "empty or whitespace" in error_message
                 or "value_error" in error_message
                 or "should be non-empty" in error_message

tests/schemas/test_memory_url.py

@@ -1,5 +1,7 @@
 """Tests for MemoryUrl parsing."""
 
+import pytest
+
 from basic_memory.schemas.memory import memory_url, memory_url_path, normalize_memory_url
 
 
@@ -59,5 +61,6 @@ def test_normalize_memory_url_no_prefix():
 
 
 def test_normalize_memory_url_empty():
-    """Test converting back to string."""
-    assert normalize_memory_url("") == ""
+    """Test that empty string raises ValueError."""
+    with pytest.raises(ValueError, match="cannot be empty"):
+        normalize_memory_url("")

tests/schemas/test_memory_url_validation.py

@@ -113,9 +113,11 @@ def test_valid_normalization(self):
             )
 
     def test_empty_url(self):
-        """Test that empty URLs return empty string."""
-        assert normalize_memory_url(None) == ""
-        assert normalize_memory_url("") == ""
+        """Test that empty URLs raise ValueError."""
+        with pytest.raises(ValueError, match="cannot be empty"):
+            normalize_memory_url(None)
+        with pytest.raises(ValueError, match="cannot be empty"):
+            normalize_memory_url("")
 
     def test_invalid_double_slashes(self):
         """Test that URLs with double slashes raise ValueError."""
@@ -145,14 +147,14 @@ def test_invalid_protocol_schemes(self):
 
     def test_whitespace_only(self):
         """Test that whitespace-only URLs raise ValueError."""
-        invalid_urls = [
+        whitespace_urls = [
             "   ",
             "\t",
             "\n",
             "  \n  ",
         ]
 
-        for url in invalid_urls:
+        for url in whitespace_urls:
             with pytest.raises(ValueError, match="cannot be empty or whitespace"):
                 normalize_memory_url(url)
 
@@ -206,9 +208,9 @@ def test_invalid_urls_fail_validation(self):
             error_msg = str(exc_info.value)
             assert "value_error" in error_msg, f"Should be a value_error for '{url}'"
 
-    def test_empty_string_fails_minlength(self):
-        """Test that empty strings fail MinLen validation."""
-        with pytest.raises(ValidationError, match="at least 1"):
+    def test_empty_string_fails_validation(self):
+        """Test that empty strings fail validation."""
+        with pytest.raises(ValidationError, match="cannot be empty"):
             memory_url.validate_python("")
 
     def test_very_long_urls_fail_maxlength(self):