use vars for batch size and flush interval seconds

This commit is contained in:
Ishaan Jaff 2024-09-11 14:40:58 -07:00
parent 0070741529
commit e681619381
2 changed files with 9 additions and 3 deletions

View file

@ -24,6 +24,9 @@ from litellm.llms.custom_httpx.http_handler import (
httpxSpecialProvider,
)
DEFAULT_BATCH_SIZE = 512
DEFAULT_FLUSH_INTERVAL_SECONDS = 5
class LangsmithInputs(BaseModel):
model: Optional[str] = None
@ -71,11 +74,12 @@ class LangsmithLogger(CustomLogger):
)
_batch_size = (
os.getenv("LANGSMITH_BATCH_SIZE", 100) or litellm.langsmith_batch_size
os.getenv("LANGSMITH_BATCH_SIZE", DEFAULT_BATCH_SIZE)
or litellm.langsmith_batch_size
)
self.batch_size = int(_batch_size)
self.log_queue = []
self.flush_interval = 10 # 5 seconds
self.flush_interval = DEFAULT_FLUSH_INTERVAL_SECONDS # 10 seconds
self.last_flush_time = time.time()
asyncio.create_task(self.periodic_flush())
self.flush_lock = asyncio.Lock()

View file

@ -16,4 +16,6 @@ general_settings:
master_key: sk-1234
litellm_settings:
success_callback: ["langsmith"]
success_callback: ["langsmith", "prometheus"]
service_callback: ["prometheus_system"]
callbacks: ["otel"]