Handling mongo indexes in firm

api/rpk-api/firm/current_firm/__init__.py

@@ -3,7 +3,7 @@ from typing import Any
 from beanie import PydanticObjectId
 from pydantic import Field
 
-from firm.core.models import CrudDocument
+from firm.core.models import CrudDocument, CrudDocumentConfig
 from firm.core.schemas import Writer, Reader
 from firm.entity.schemas import EntityIndividualCreate, EntityCorporationCreate, EntityRead
 
@@ -59,6 +59,10 @@ class CurrentFirmSchemaUpdate(Writer):
     pass
 
 class Partner(CrudDocument):
+    document_config = CrudDocumentConfig(
+        indexes=["user_id", "entity_id"],
+    )
+
     user_id: PydanticObjectId = Field()
     entity_id: PydanticObjectId = Field()