fix(api): tighten agent v2 generated contracts (#36989)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>

api/controllers/console/app/agent_app_feature.py

@@ -9,8 +9,6 @@ persists them onto the app's ``app_model_config`` without touching anything the
 Soul owns.
 """
 
-from typing import Any
-
 from flask_restx import Resource
 from pydantic import BaseModel, Field
 
@@ -25,13 +23,20 @@ from controllers.console.wraps import (
     with_current_user,
 )
 from events.app_event import app_model_config_was_updated
+from libs.helper import dump_response
 from libs.login import login_required
 from models import Account
+from models.agent_config_entities import (
+    AgentFeatureToggleConfig,
+    AgentSensitiveWordAvoidanceFeatureConfig,
+    AgentSuggestedQuestionsAfterAnswerFeatureConfig,
+    AgentTextToSpeechFeatureConfig,
+)
 from models.model import App, AppMode
 from services.agent_app_feature_service import AgentAppFeatureConfigService
 
 
-class AgentAppFeaturesRequest(BaseModel):
+class AgentAppFeaturesPayload(BaseModel):
     """Presentation features configurable on an Agent App.
 
     All fields are optional; an omitted field is reset to its disabled/empty
@@ -42,18 +47,20 @@ class AgentAppFeaturesRequest(BaseModel):
     suggested_questions: list[str] | None = Field(
         default=None, description="Preset questions shown alongside the opener"
     )
-    suggested_questions_after_answer: dict[str, Any] | None = Field(
+    suggested_questions_after_answer: AgentSuggestedQuestionsAfterAnswerFeatureConfig | None = Field(
         default=None, description="Follow-up suggestions config, e.g. {'enabled': true}"
     )
-    speech_to_text: dict[str, Any] | None = Field(default=None, description="Speech-to-text config")
-    text_to_speech: dict[str, Any] | None = Field(default=None, description="Text-to-speech config")
-    retriever_resource: dict[str, Any] | None = Field(
+    speech_to_text: AgentFeatureToggleConfig | None = Field(default=None, description="Speech-to-text config")
+    text_to_speech: AgentTextToSpeechFeatureConfig | None = Field(default=None, description="Text-to-speech config")
+    retriever_resource: AgentFeatureToggleConfig | None = Field(
         default=None, description="Citations / attributions config, e.g. {'enabled': true}"
     )
-    sensitive_word_avoidance: dict[str, Any] | None = Field(default=None, description="Content moderation config")
+    sensitive_word_avoidance: AgentSensitiveWordAvoidanceFeatureConfig | None = Field(
+        default=None, description="Content moderation config"
+    )
 
 
-register_schema_models(console_ns, AgentAppFeaturesRequest)
+register_schema_models(console_ns, AgentAppFeaturesPayload)
 register_response_schema_models(console_ns, SimpleResultResponse)
 
 
@@ -62,7 +69,7 @@ class AgentAppFeatureConfigResource(Resource):
     @console_ns.doc("update_agent_app_features")
     @console_ns.doc(description="Update an Agent App's presentation features (opener, follow-up, citations, ...)")
     @console_ns.doc(params={"app_id": "Application ID"})
-    @console_ns.expect(console_ns.models[AgentAppFeaturesRequest.__name__])
+    @console_ns.expect(console_ns.models[AgentAppFeaturesPayload.__name__])
     @console_ns.response(200, "Features updated successfully", console_ns.models[SimpleResultResponse.__name__])
     @console_ns.response(400, "Invalid configuration")
     @console_ns.response(404, "App not found")
@@ -73,7 +80,7 @@ class AgentAppFeatureConfigResource(Resource):
     @get_app_model(mode=[AppMode.AGENT])
     @with_current_user
     def post(self, current_user: Account, app_model: App):
-        args = AgentAppFeaturesRequest.model_validate(console_ns.payload)
+        args = AgentAppFeaturesPayload.model_validate(console_ns.payload or {})
 
         new_app_model_config = AgentAppFeatureConfigService.update_features(
             app_model=app_model,
@@ -83,4 +90,4 @@ class AgentAppFeatureConfigResource(Resource):
 
         app_model_config_was_updated.send(app_model, app_model_config=new_app_model_config)
 
-        return {"result": "success"}
+        return dump_response(SimpleResultResponse, {"result": "success"})