Merge pull request #1478 from Agenta-AI/templates-url-in-env-var

Templates url in env var

agenta-backend/agenta_backend/services/templates_manager.py

@@ -14,6 +14,8 @@
 if os.environ["FEATURE_FLAG"] in ["oss", "cloud"]:
     from agenta_backend.services import container_manager
 
+templates_base_url = os.getenv("TEMPLATES_BASE_URL")
+
 from typing import Union
 
 
@@ -118,9 +120,7 @@ async def retrieve_templates_info_from_s3(
             return json.loads(cached_data)
 
     # If not cached, fetch data from Docker Hub and cache it in Redis
-    response = await get_templates_info_from_s3(
-        "https://llm-app-json.s3.eu-central-1.amazonaws.com/llm_info.json"
-    )
+    response = await get_templates_info_from_s3(f"{templates_base_url}/llm_info.json")
 
     # Cache the data in Redis for 60 minutes
     r.set("temp_data", json.dumps(response), ex=900)

docker-compose.gh.yml

@@ -26,6 +26,7 @@ services:
             - DOMAIN_NAME=${DOMAIN_NAME:-http://localhost}
             - CELERY_BROKER_URL=amqp://guest@rabbitmq//
             - CELERY_RESULT_BACKEND=redis://redis:6379/0
+            - TEMPLATES_BASE_URL=https://llm-app-json.s3.eu-central-1.amazonaws.com
         command:
             [
                 "uvicorn",

docker-compose.prod.yml

@@ -22,6 +22,7 @@ services:
       - AGENTA_TEMPLATE_REPO=agentaai/templates_v2
       - CELERY_BROKER_URL=amqp://guest@rabbitmq//
       - CELERY_RESULT_BACKEND=redis://redis:6379/0
+      - TEMPLATES_BASE_URL=https://llm-app-json.s3.eu-central-1.amazonaws.com
     volumes:
       - ./agenta-backend/agenta_backend:/app/agenta_backend
       - ./agenta-backend/tests:/app/tests

docker-compose.test.yml

@@ -27,6 +27,7 @@ services:
             - OPENAI_API_KEY=${OPENAI_API_KEY}
             - AGENTA_TEMPLATE_REPO=agentaai/templates_v2
             - POSTHOG_API_KEY=phc_hmVSxIjTW1REBHXgj2aw4HW9X6CXb6FzerBgP9XenC7
+            - TEMPLATES_BASE_URL=https://llm-app-json.s3.eu-central-1.amazonaws.com
         volumes:
             - ./agenta-backend/agenta_backend:/app/agenta_backend
             - ./agenta-backend/tests:/app/tests

docker-compose.yml

@@ -25,6 +25,7 @@ services:
             - POSTHOG_API_KEY=phc_hmVSxIjTW1REBHXgj2aw4HW9X6CXb6FzerBgP9XenC7
             - CELERY_BROKER_URL=amqp://guest@rabbitmq//
             - CELERY_RESULT_BACKEND=redis://redis:6379/0
+            - TEMPLATES_BASE_URL=https://llm-app-json.s3.eu-central-1.amazonaws.com
         volumes:
             - ./agenta-backend/agenta_backend:/app/agenta_backend
             - ./agenta-backend/tests:/app/tests