diff --git a/agents/core-logic/agent.py b/agents/core-logic/agent.py index dc7054b..4e33098 100644 --- a/agents/core-logic/agent.py +++ b/agents/core-logic/agent.py @@ -31,12 +31,10 @@ PROJECT_ID = os.environ.get("GOOGLE_CLOUD_PROJECT", "propane-will-491900-m5") LOCATION = os.environ.get("GOOGLE_CLOUD_LOCATION", "global") RAG_CORPUS = os.environ.get("RAG_CORPUS", "") -# Light mode — gemini-2.5-flash (GA, global) ORCHESTRATOR_MODEL = os.environ.get("ORCHESTRATOR_MODEL", "gemini-2.5-flash") SUBAGENT_MODEL = os.environ.get("SUBAGENT_MODEL", "gemini-2.5-flash") REASONING_MODEL = os.environ.get("REASONING_MODEL", "gemini-2.5-flash") -# Heavy mode — gemini-2.5-pro (GA, global) HEAVY_ORCHESTRATOR = os.environ.get("HEAVY_ORCHESTRATOR_MODEL", "gemini-2.5-pro") HEAVY_SUBAGENT = os.environ.get("HEAVY_SUBAGENT_MODEL", "gemini-2.5-pro") HEAVY_REASONING = os.environ.get("HEAVY_REASONING_MODEL", "gemini-2.5-flash") @@ -49,7 +47,6 @@ HEAVY_MODE_ALLOWED_USERS = ["opax", "admin"] Mode = Literal["light", "heavy"] APP_NAME = "opax" -# Token logger — feiler stille, stopper aldri agent try: from token_logger import log_token_usage, create_bq_table_if_not_exists except ImportError: @@ -58,7 +55,6 @@ except ImportError: def create_bq_table_if_not_exists(): pass -# Opprett BQ-tabell ved oppstart (idempotent, feiler stille) create_bq_table_if_not_exists() @@ -100,16 +96,16 @@ else: logger.warning("RAG_CORPUS ikke satt — kjører uten RAG") -# —— OPAX-MCP toolset (feiler stille slik at agenten starter selv uten MCP_SECRET) —— -opax_mcp_toolset = None +# —— OPAX-MCP FunctionTools (feiler stille) —— +opax_tools = [] try: import sys, pathlib sys.path.insert(0, str(pathlib.Path(__file__).parent.parent / "tools")) - from mcp_tools import get_opax_mcp_toolset - opax_mcp_toolset = get_opax_mcp_toolset() - logger.info("OPAX-MCP toolset initialisert") + from mcp_tools import get_opax_tools + opax_tools = get_opax_tools() + logger.info(f"OPAX-MCP tools lastet: {[t.name for t in opax_tools]}") except Exception as e: - logger.warning(f"OPAX-MCP toolset ikke tilgjengelig (CI3): {e}") + logger.warning(f"OPAX-MCP tools ikke tilgjengelig (CI3): {e}") def get_models_for_mode(mode: Mode) -> dict: @@ -146,16 +142,16 @@ def build_agent(mode: str = "light") -> Agent: "billing_export (OQ-15) ikke er aktivert ennå — si dette ærlig, ikke gjett tall. " "HITL-grenser: du kan ikke kjøre terraform apply / onboarde kunder, " "selge Vauco OS eller OPAX, eller passere godkjenningsporter uten Chris. " - "Du har tilgang til MCP-verktøy via opax-mcp: push_static (last opp filer til opax.vauco.no), " - "get_build_status (sjekk CI/CD-status), deploy_service (trigger deploy), " - "get_logs (hent Cloud Run-logger). Bruk disse når det er relevant. " + "Du har tilgang til MCP-verktøy: get_build_status (live CI/CD-status), " + "get_logs (Cloud Run-logger), push_static (last opp filer til opax.vauco.no), " + "deploy_service (trigger deploy — kun med Chris sin godkjenning). " + "Bruk alltid get_build_status for spørsmål om CI/CD-status — ikke gjett fra minne. " "Svar på norsk (bokmål) med mindre annet er bedt om." ) tools = [] if rag_tool: tools.append(rag_tool) - if opax_mcp_toolset: - tools.append(opax_mcp_toolset) + tools.extend(opax_tools) return Agent( model=models["orchestrator"], name="opax_agent", @@ -178,7 +174,7 @@ async def _run_async( mode = _normalize_mode(mode) agent = build_agent(mode=mode) models = get_models_for_mode(mode) - module_name = f"jason/{mode}" # CG4: 'jason/light' eller 'jason/heavy' + module_name = f"jason/{mode}" session_service = InMemorySessionService() session = await session_service.create_session( @@ -201,17 +197,16 @@ async def _run_async( input_tokens += getattr(um, "prompt_token_count", 0) or 0 output_tokens += getattr(um, "candidates_token_count", 0) or 0 - # Logg med full CG4-labeling — feiler stille if input_tokens > 0 or output_tokens > 0: log_token_usage( - agent_name=module_name, # bakoverkompatibelt + agent_name=module_name, model_name=models["orchestrator"], input_tokens=input_tokens, output_tokens=output_tokens, request_id=request_id, - module_name=module_name, # CG4: 'jason/light' | 'jason/heavy' - caller_type=caller_type, # CG4: agent|browser|cli|cron|api - session_id=session_id, # CG4: for sesjon-gruppering + module_name=module_name, + caller_type=caller_type, + session_id=session_id, ) return final_text @@ -224,16 +219,6 @@ def run( mode: str = "light", caller_type: str = "agent", ) -> str: - """ - Kjør OPAX-agenten. - - caller_type-verdier: - 'agent' — programmatisk kall fra server/app.py - 'browser' — direkte fra browser-chat (opax.vauco.no) - 'cli' — lokal kjøring fra terminalen - 'cron' — bakgrunnsjobb - 'api' — ekstern REST-kall - """ mode = _normalize_mode(mode) authorize_mode(user_id, mode) return asyncio.run(_run_async(