mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 18:54:30 +00:00
add MAX_SIZE_IN_MEMORY_QUEUE
This commit is contained in:
parent
3374c54ba2
commit
363fb0c46f
1 changed files with 2 additions and 0 deletions
|
@ -4,6 +4,7 @@ Base class for in memory buffer for database transactions
|
|||
import asyncio
|
||||
|
||||
from litellm._logging import verbose_proxy_logger
|
||||
from litellm.constants import MAX_SIZE_IN_MEMORY_QUEUE
|
||||
|
||||
|
||||
class BaseUpdateQueue:
|
||||
|
@ -11,6 +12,7 @@ class BaseUpdateQueue:
|
|||
|
||||
def __init__(self):
|
||||
self.update_queue = asyncio.Queue()
|
||||
self.MAX_SIZE_IN_MEMORY_QUEUE = MAX_SIZE_IN_MEMORY_QUEUE
|
||||
|
||||
async def add_update(self, update):
|
||||
"""Enqueue an update."""
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue