refactor: Removes tenant ID check from rate limit logic

Signed-off-by: -LAN- <laipz8200@outlook.com>
pull/20585/head
-LAN- 12 months ago
parent e40e9db39a
commit d1b5809dcd
No known key found for this signature in database
GPG Key ID: 6BA0D108DED011FF

@ -86,8 +86,8 @@ class KnowledgeRetrievalNode(LLMNode):
return NodeRunResult( return NodeRunResult(
status=WorkflowNodeExecutionStatus.FAILED, inputs=variables, error="Query is required." status=WorkflowNodeExecutionStatus.FAILED, inputs=variables, error="Query is required."
) )
# TODO(-LAN-): Move this check outside.
# check rate limit # check rate limit
if self.tenant_id:
knowledge_rate_limit = FeatureService.get_knowledge_rate_limit(self.tenant_id) knowledge_rate_limit = FeatureService.get_knowledge_rate_limit(self.tenant_id)
if knowledge_rate_limit.enabled: if knowledge_rate_limit.enabled:
current_time = int(time.time() * 1000) current_time = int(time.time() * 1000)

Loading…
Cancel
Save