Selaa lähdekoodia

config milvus service

highing666 2 vuotta sitten
vanhempi
commit
3c31d0165e
3 muutettua tiedostoa jossa 6 lisäystä ja 2 poistoa
  1. 2 0
      app/core/config.py
  2. 3 1
      app/core/events.py
  3. 1 1
      docker-compose.yml

+ 2 - 0
app/core/config.py

@@ -53,6 +53,8 @@ class Settings(BaseSettings):
     POSTGRES_DB: str
     SQLALCHEMY_DATABASE_URL: Optional[PostgresDsn] = None
 
+    NEED_MILVUS: str
+
     @validator("SQLALCHEMY_DATABASE_URL", pre=True)
     def assemble_db_connection(cls, v: Optional[str], values: Dict[str, Any]) -> Any:
         if isinstance(v, str):

+ 3 - 1
app/core/events.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 
+import os
 from typing import Callable, Optional
 
 from fastapi import FastAPI
@@ -11,6 +12,7 @@ from app.services.milvus_helpers import get_milvus_cli
 def create_start_app_handler(app: Optional[FastAPI] = None) -> Callable:
     async def start_app() -> None:
         # await regulate_ahu_freq()
-        await get_milvus_cli()
+        if os.getenv("NEED_MILVUS", "yes") == "yes":
+            await get_milvus_cli()
 
     return start_app

+ 1 - 1
docker-compose.yml

@@ -2,7 +2,7 @@ version: "3"
 
 services:
   app:
-      image: registry.persagy.com/sagacloud/saga_algo_api:0.3.39
+      image: registry.persagy.com/sagacloud/saga_algo_api:0.3.40
       container_name: saga_algo_api
       ports:
         - 8002:8002