feat(fix): add PAGES_LIMIT configuration for ETL services

This commit is contained in:
DESKTOP-RTLN3BA\$punk 2025-12-11 00:29:56 -08:00
parent 0c6a519c37
commit 3207fbf9ea
5 changed files with 11 additions and 2 deletions

View file

@ -78,6 +78,10 @@ STT_SERVICE=local/base
# STT_SERVICE_API_BASE=
# (Optional) Maximum pages limit per user for ETL services (default: `999999999` for unlimited in OSS version)
PAGES_LIMIT=500
FIRECRAWL_API_KEY=fcr-01J0000000000000000000000
# File Parser Service

View file

@ -200,6 +200,9 @@ class Config:
# ETL Service
ETL_SERVICE = os.getenv("ETL_SERVICE")
# Pages limit for ETL services (default to very high number for OSS unlimited usage)
PAGES_LIMIT = int(os.getenv("PAGES_LIMIT", "999999999"))
if ETL_SERVICE == "UNSTRUCTURED":
# Unstructured API Key
UNSTRUCTURED_API_KEY = os.getenv("UNSTRUCTURED_API_KEY")

View file

@ -710,7 +710,7 @@ if config.AUTH_TYPE == "GOOGLE":
)
# Page usage tracking for ETL services
pages_limit = Column(Integer, nullable=False, default=500, server_default="500")
pages_limit = Column(Integer, nullable=False, default=config.PAGES_LIMIT, server_default=str(config.PAGES_LIMIT))
pages_used = Column(Integer, nullable=False, default=0, server_default="0")
else:
@ -731,7 +731,7 @@ else:
)
# Page usage tracking for ETL services
pages_limit = Column(Integer, nullable=False, default=500, server_default="500")
pages_limit = Column(Integer, nullable=False, default=config.PAGES_LIMIT, server_default=str(config.PAGES_LIMIT))
pages_used = Column(Integer, nullable=False, default=0, server_default="0")