refactor(agent): extract magic numbers to constants

This commit is contained in:
qzl
2026-03-03 16:28:12 +08:00
parent 9aefb76c9e
commit 7be8669144
+8 -14
View File
@@ -32,6 +32,9 @@ if TYPE_CHECKING:
logger = get_logger("v1.agent.service") logger = get_logger("v1.agent.service")
DEFAULT_RATE_LIMIT = 60
EMPTY_USAGE = {"input_tokens": 0, "output_tokens": 0, "cost": "0"}
class ResumeDecisionResult(BaseModel): class ResumeDecisionResult(BaseModel):
applied: bool applied: bool
@@ -83,8 +86,8 @@ class AgentChatService(BaseService):
await enforce_rate_limit( await enforce_rate_limit(
scope="agent_run", scope="agent_run",
identifier=str(user_id), identifier=str(user_id),
limit=60, limit=DEFAULT_RATE_LIMIT,
window_seconds=60, window_seconds=DEFAULT_RATE_LIMIT,
) )
now = datetime.now(timezone.utc) now = datetime.now(timezone.utc)
@@ -272,26 +275,17 @@ class AgentChatService(BaseService):
self, *, message: str, context: dict[str, object] self, *, message: str, context: dict[str, object]
) -> dict[str, object]: ) -> dict[str, object]:
context["intent"] = "default" context["intent"] = "default"
return { return {"content": message, "usage": EMPTY_USAGE}
"content": message,
"usage": {"input_tokens": 0, "output_tokens": 0, "cost": "0"},
}
async def _execution_stage( async def _execution_stage(
self, *, message: str, context: dict[str, object] self, *, message: str, context: dict[str, object]
) -> dict[str, object]: ) -> dict[str, object]:
return { return {"content": message, "usage": EMPTY_USAGE}
"content": message,
"usage": {"input_tokens": 0, "output_tokens": 0, "cost": "0"},
}
async def _organization_stage( async def _organization_stage(
self, *, message: str, context: dict[str, object] self, *, message: str, context: dict[str, object]
) -> dict[str, object]: ) -> dict[str, object]:
return { return {"content": message, "usage": EMPTY_USAGE}
"content": message,
"usage": {"input_tokens": 0, "output_tokens": 0, "cost": "0"},
}
async def get_state_snapshot(self, session_id: UUID) -> dict | None: async def get_state_snapshot(self, session_id: UUID) -> dict | None:
stmt = select(AgentChatSession).where(AgentChatSession.id == session_id) stmt = select(AgentChatSession).where(AgentChatSession.id == session_id)