refactor: Removes tenant ID check from rate limit logic (#20585)
Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
@@ -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)
|
||||||
|
Reference in New Issue
Block a user