add index_document to search_service.py

src/basic_memory/schemas/search.py

@@ -1,16 +1,29 @@
+"""Search schemas for Basic Memory."""
+
 from typing import Optional, List
 from datetime import datetime
+from enum import Enum
 from pydantic import BaseModel
 
+
+class SearchItemType(str, Enum):
+    """Types of searchable items."""
+    DOCUMENT = "document"
+    ENTITY = "entity"
+
+
 class SearchQuery(BaseModel):
+    """Search query parameters."""
     text: str
-    types: Optional[List[str]] = None
+    types: Optional[List[SearchItemType]] = None
     entity_types: Optional[List[str]] = None
     after_date: Optional[datetime] = None
 
+
 class SearchResult(BaseModel):
+    """Search result item."""
     path_id: str
     file_path: str
-    type: str
+    type: SearchItemType
     score: float
     metadata: dict