21 lines
642 B
Python
21 lines
642 B
Python
from datetime import timedelta
|
|
from typing import Any, Union
|
|
|
|
from cachetools import TTLCache
|
|
|
|
from skyvern.forge.sdk.cache.base import CACHE_EXPIRE_TIME, MAX_CACHE_ITEM, BaseCache
|
|
|
|
|
|
class LocalCache(BaseCache):
|
|
def __init__(self) -> None:
|
|
self.cache: TTLCache = TTLCache(maxsize=MAX_CACHE_ITEM, ttl=CACHE_EXPIRE_TIME.total_seconds())
|
|
|
|
async def get(self, key: str) -> Any:
|
|
if key not in self.cache:
|
|
return None
|
|
value = self.cache[key]
|
|
return value
|
|
|
|
async def set(self, key: str, value: Any, ex: Union[int, timedelta, None] = CACHE_EXPIRE_TIME) -> None:
|
|
self.cache[key] = value
|