fix bug

Canner · Sep 14, 2024 · d39c41b · d39c41b
1 parent 40cc2a6
commit d39c41b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/wren-ai-service/src/__main__.py b/wren-ai-service/src/__main__.py
@@ -38,7 +38,7 @@ async def lifespan(app: FastAPI):
     providers = init_providers(
         engine_config=EngineConfig(provider=os.getenv("ENGINE", "wren_ui"))
     )
-    create_service_container(
+    app.state.service_container = create_service_container(
         app,
         *providers,
         should_force_deploy=bool(os.getenv("SHOULD_FORCE_DEPLOY", "")),
@@ -64,7 +64,7 @@ async def lifespan(app: FastAPI):
             "ttl": int(os.getenv("QUERY_CACHE_TTL") or 120),
         },
     )
-    create_service_metadata(app, *providers)
+    app.state.service_metadata = create_service_metadata(app, *providers)
     init_langfuse()
 
     yield

diff --git a/wren-ai-service/src/globals.py b/wren-ai-service/src/globals.py
@@ -69,7 +69,7 @@ def create_service_container(
             dataset_name="view_questions", recreate_index=True
         )
 
-    app.state.service_container = ServiceContainer(
+    return ServiceContainer(
         semantics_preparation_service=SemanticsPreparationService(
             pipelines={
                 "indexing": indexing.Indexing(
@@ -185,7 +185,7 @@ def create_service_metadata(
     embedder_provider: EmbedderProvider,
     *_,
     pyproject_path: str = "pyproject.toml",
-):
+) -> ServiceMetadata:
     def _get_version_from_pyproject() -> str:
         with open(pyproject_path, "r") as f:
             pyproject = toml.load(f)
@@ -201,7 +201,7 @@ def _get_version_from_pyproject() -> str:
 
     logger.info(f"Service version: {service_version}")
 
-    app.state.service_metadata = ServiceMetadata(models_metadata, service_version)
+    return ServiceMetadata(models_metadata, service_version)
 
 
 # Create a dependency that will be used to access the ServiceMetadata