vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Use pooled Redis client for notification registry in cloud (#4808)

Browse Source
This commit is contained in:
Shuchang Zheng
2026-02-19 09:02:52 -08:00
committed by GitHub
parent c43ec25d22
commit 704c6c8aa8
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
skyvern/forge/forge_app.py
View File

@@ -111,15 +111,14 @@ def create_forge_app() -> ForgeApp:
app.STORAGE = StorageFactory.get_storage() app.STORAGE = StorageFactory.get_storage()
app.CACHE = CacheFactory.get_cache() app.CACHE = CacheFactory.get_cache()
if settings.NOTIFICATION_REGISTRY_TYPE == "redis": if settings.NOTIFICATION_REGISTRY_TYPE == "redis" and settings.NOTIFICATION_REDIS_URL:
from redis.asyncio import from_url as redis_from_url from redis.asyncio import from_url as redis_from_url
from skyvern.forge.sdk.notification.factory import NotificationRegistryFactory from skyvern.forge.sdk.notification.factory import NotificationRegistryFactory
from skyvern.forge.sdk.notification.redis import RedisNotificationRegistry from skyvern.forge.sdk.notification.redis import RedisNotificationRegistry
from skyvern.forge.sdk.redis.factory import RedisClientFactory from skyvern.forge.sdk.redis.factory import RedisClientFactory
redis_url = settings.NOTIFICATION_REDIS_URL or settings.REDIS_URL redis_client = redis_from_url(settings.NOTIFICATION_REDIS_URL, decode_responses=True)
redis_client = redis_from_url(redis_url, decode_responses=True)
RedisClientFactory.set_client(redis_client) RedisClientFactory.set_client(redis_client)
NotificationRegistryFactory.set_registry(RedisNotificationRegistry(redis_client)) NotificationRegistryFactory.set_registry(RedisNotificationRegistry(redis_client))
app.ARTIFACT_MANAGER = ArtifactManager() app.ARTIFACT_MANAGER = ArtifactManager()