fix: fix review comments

api/routes/telephony.py

@@ -153,6 +153,7 @@ async def initiate_call(
                 "telephony_configuration_id": telephony_configuration_id,
             },
             use_draft=True,
+            organization_id=user.selected_organization_id,
         )
         workflow_run_id = workflow_run.id
     else:

api/routes/workflow.py

@@ -1081,7 +1081,12 @@ async def create_workflow_run(
         user: The user to create the workflow run for
     """
     run = await db_client.create_workflow_run(
-        request.name, workflow_id, request.mode, user.id, use_draft=True
+        request.name,
+        workflow_id,
+        request.mode,
+        user.id,
+        use_draft=True,
+        organization_id=user.selected_organization_id,
     )
     return {
         "id": run.id,

api/routes/workflow_text_chat.py

@@ -10,6 +10,7 @@ from api.db import db_client
 from api.db.models import UserModel, WorkflowRunTextSessionModel
 from api.enums import WorkflowRunMode
 from api.services.auth.depends import get_user
+from api.services.quota_service import check_dograh_quota
 from api.services.workflow.text_chat_session_service import (
     TextChatPendingTurnLostError,
     TextChatSessionExecutionError,
@@ -95,16 +96,27 @@ def _revision_conflict_detail(e: Any) -> dict[str, Any]:
     }
 
 
+def _require_selected_organization_id(user: UserModel) -> int:
+    if user.selected_organization_id is None:
+        raise HTTPException(status_code=403, detail="Organization context is required")
+    return user.selected_organization_id
+
+
+async def _ensure_text_chat_quota(user: UserModel, workflow_id: int) -> None:
+    quota_result = await check_dograh_quota(user, workflow_id=workflow_id)
+    if not quota_result.has_quota:
+        raise HTTPException(status_code=402, detail=quota_result.error_message)
+
+
 async def _load_text_session_or_404(
     workflow_id: int,
     run_id: int,
     user: UserModel,
 ) -> WorkflowRunTextSessionModel:
     set_current_run_id(run_id)
-    if user.selected_organization_id is None:
-        raise HTTPException(status_code=403, detail="Organization context is required")
+    organization_id = _require_selected_organization_id(user)
     text_session = await db_client.get_workflow_run_text_session(
-        run_id, organization_id=user.selected_organization_id
+        run_id, organization_id=organization_id
     )
     if not text_session or not text_session.workflow_run:
         raise HTTPException(status_code=404, detail="Text chat session not found")
@@ -148,6 +160,9 @@ async def create_text_chat_session(
     request: CreateTextChatSessionRequest,
     user: UserModel = Depends(get_user),
 ) -> WorkflowRunTextSessionResponse:
+    organization_id = _require_selected_organization_id(user)
+    await _ensure_text_chat_quota(user, workflow_id)
+
     session_name = request.name or f"WR-TEXT-{uuid4().hex[:6].upper()}"
     try:
         workflow_run = await db_client.create_workflow_run(
@@ -157,6 +172,7 @@ async def create_text_chat_session(
             user_id=user.id,
             initial_context=request.initial_context,
             use_draft=True,
+            organization_id=organization_id,
         )
     except ValueError as e:
         raise HTTPException(status_code=404, detail=str(e))
@@ -221,6 +237,8 @@ async def append_text_chat_message(
     user: UserModel = Depends(get_user),
 ) -> WorkflowRunTextSessionResponse:
     text_session = await _load_text_session_or_404(workflow_id, run_id, user)
+    await _ensure_text_chat_quota(user, workflow_id)
+
     try:
         text_session = await append_text_chat_user_message(
             run_id=run_id,