From 2950542c2966dfd520862b380e75b7167fbde878 Mon Sep 17 00:00:00 2001 From: Chris Christiansen Date: Tue, 23 Jun 2026 11:01:24 +0000 Subject: [PATCH] =?UTF-8?q?fix(startup):=20move=20BQ=20table=20init=20to?= =?UTF-8?q?=20lifespan=20event=20=E2=80=94=20fixes=20Cloud=20Run=20startup?= =?UTF-8?q?=20probe=20timeout?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- agents/core-logic/agent.py | 2 -- agents/core-logic/app.py | 2 ++ 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/agents/core-logic/agent.py b/agents/core-logic/agent.py index a81b9f1..2d28ccb 100644 --- a/agents/core-logic/agent.py +++ b/agents/core-logic/agent.py @@ -52,8 +52,6 @@ except ImportError as _e: return flash_model if mode != "heavy" else pro_model class TokenBudgetExceeded(Exception): pass -create_bq_table_if_not_exists() - _OPAX_MCP_URL = os.environ.get("MCP_SERVER_URL", "https://opax-mcp-zjbqp3prqq-uc.a.run.app") _MCP_SECRET = os.environ.get("MCP_SECRET", "") diff --git a/agents/core-logic/app.py b/agents/core-logic/app.py index a128b02..949a98a 100644 --- a/agents/core-logic/app.py +++ b/agents/core-logic/app.py @@ -22,6 +22,7 @@ try: from google.genai.types import Content, Part from agent import root_agent from agents.recommendations_engine import get_recommendations + from token_logger import create_bq_table_if_not_exists except ImportError as e: logger.error(f"Failed to import ADK dependencies: {e}") raise @@ -41,6 +42,7 @@ _budget_store: dict = {} @asynccontextmanager async def lifespan(app: FastAPI): logger.info(f"OSVauco agent '{APP_NAME}' starting up") + create_bq_table_if_not_exists() logger.info(f"Static dir: {STATIC_DIR} (exists={STATIC_DIR.exists()})") yield logger.info(f"OSVauco agent '{APP_NAME}' shutting down")