fix(CI3): bruk get_opax_tools() FunctionTools i stedet for MCPToolset
This commit is contained in:
parent
d2a93979a4
commit
e5b4e2faa5
|
|
@ -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(
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user