extend ExtendedSearch model

Author: jonhealy1

stac_fastapi/core/stac_fastapi/core/extensions/collections_search.py

@@ -5,6 +5,7 @@
 from fastapi import APIRouter, FastAPI, Request
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
+from stac_pydantic.api.search import ExtendedSearch
 from starlette.responses import Response
 
 from stac_fastapi.api.models import APIRequest
@@ -13,6 +14,12 @@
 from stac_fastapi.types.stac import Collections
 
 
+class CollectionsSearchRequest(ExtendedSearch):
+    """Extended search model for collections with free text search support."""
+
+    q: Optional[Union[str, List[str]]] = None
+
+
 class CollectionsSearchEndpointExtension(ApiExtension):
     """Collections search endpoint extension.
 
@@ -128,10 +135,8 @@ async def collections_search_post_endpoint(
         Returns:
             Collections: Collections object.
         """
-        from stac_pydantic.api.search import ExtendedSearch
-
         # Convert the dict to an ExtendedSearch model
-        search_request = ExtendedSearch.model_validate(body)
+        search_request = CollectionsSearchRequest.model_validate(body)
 
         # Check if fields are present in the body
         if "fields" in body: