feat: Add top_k as endpoint parameter

cognee/api/v1/search/routers/get_search_router.py

@@ -1,3 +1,4 @@
+from typing import Optional
 from uuid import UUID
 from datetime import datetime
 from fastapi import Depends, APIRouter
@@ -12,6 +13,7 @@
 class SearchPayloadDTO(InDTO):
     search_type: SearchType
     query: str
+    top_k: Optional[int] = 10
 
 
 def get_search_router() -> APIRouter:
@@ -39,7 +41,10 @@ async def search(payload: SearchPayloadDTO, user: User = Depends(get_authenticat
 
         try:
             results = await cognee_search(
-                query_text=payload.query, query_type=payload.search_type, user=user
+                query_text=payload.query,
+                query_type=payload.search_type,
+                user=user,
+                top_k=payload.top_k,
             )
 
             return results