fix assets_base_url

huggingface · Oct 28, 2024 · 98c3c55 · 98c3c55
1 parent b982014
commit 98c3c55
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/services/rows/src/rows/app.py b/services/rows/src/rows/app.py
@@ -61,7 +61,9 @@ def create_app_with_config(app_config: AppConfig) -> Starlette:
 
     url_signer = get_cloudfront_signer(cloudfront_config=app_config.cloudfront)
     url_preparator = URLPreparator(
-        url_signer=url_signer, hf_endpoint=app_config.common.hf_endpoint, assets_base_url=app_config.assets.base_url
+        url_signer=url_signer,
+        hf_endpoint=app_config.common.hf_endpoint,
+        assets_base_url=app_config.cached_assets.base_url,
     )
     cached_assets_storage_client = StorageClient(
         protocol=app_config.cached_assets.storage_protocol,

diff --git a/services/search/src/search/app.py b/services/search/src/search/app.py
@@ -62,7 +62,9 @@ def create_app_with_config(app_config: AppConfig) -> Starlette:
     queue_resource = QueueMongoResource(database=app_config.queue.mongo_database, host=app_config.queue.mongo_url)
     url_signer = get_cloudfront_signer(cloudfront_config=app_config.cloudfront)
     url_preparator = URLPreparator(
-        url_signer=url_signer, hf_endpoint=app_config.common.hf_endpoint, assets_base_url=app_config.assets.base_url
+        url_signer=url_signer,
+        hf_endpoint=app_config.common.hf_endpoint,
+        assets_base_url=app_config.cached_assets.base_url,
     )
     cached_assets_storage_client = StorageClient(
         protocol=app_config.cached_assets.storage_protocol,