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")
|
LOCATION = os.environ.get("GOOGLE_CLOUD_LOCATION", "global")
|
||||||
RAG_CORPUS = os.environ.get("RAG_CORPUS", "")
|
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")
|
ORCHESTRATOR_MODEL = os.environ.get("ORCHESTRATOR_MODEL", "gemini-2.5-flash")
|
||||||
SUBAGENT_MODEL = os.environ.get("SUBAGENT_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")
|
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_ORCHESTRATOR = os.environ.get("HEAVY_ORCHESTRATOR_MODEL", "gemini-2.5-pro")
|
||||||
HEAVY_SUBAGENT = os.environ.get("HEAVY_SUBAGENT_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")
|
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"]
|
Mode = Literal["light", "heavy"]
|
||||||
APP_NAME = "opax"
|
APP_NAME = "opax"
|
||||||
|
|
||||||
# Token logger — feiler stille, stopper aldri agent
|
|
||||||
try:
|
try:
|
||||||
from token_logger import log_token_usage, create_bq_table_if_not_exists
|
from token_logger import log_token_usage, create_bq_table_if_not_exists
|
||||||
except ImportError:
|
except ImportError:
|
||||||
|
|
@ -58,7 +55,6 @@ except ImportError:
|
||||||
def create_bq_table_if_not_exists():
|
def create_bq_table_if_not_exists():
|
||||||
pass
|
pass
|
||||||
|
|
||||||
# Opprett BQ-tabell ved oppstart (idempotent, feiler stille)
|
|
||||||
create_bq_table_if_not_exists()
|
create_bq_table_if_not_exists()
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -100,16 +96,16 @@ else:
|
||||||
logger.warning("RAG_CORPUS ikke satt — kjører uten RAG")
|
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 FunctionTools (feiler stille) ——
|
||||||
opax_mcp_toolset = None
|
opax_tools = []
|
||||||
try:
|
try:
|
||||||
import sys, pathlib
|
import sys, pathlib
|
||||||
sys.path.insert(0, str(pathlib.Path(__file__).parent.parent / "tools"))
|
sys.path.insert(0, str(pathlib.Path(__file__).parent.parent / "tools"))
|
||||||
from mcp_tools import get_opax_mcp_toolset
|
from mcp_tools import get_opax_tools
|
||||||
opax_mcp_toolset = get_opax_mcp_toolset()
|
opax_tools = get_opax_tools()
|
||||||
logger.info("OPAX-MCP toolset initialisert")
|
logger.info(f"OPAX-MCP tools lastet: {[t.name for t in opax_tools]}")
|
||||||
except Exception as e:
|
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:
|
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. "
|
"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, "
|
"HITL-grenser: du kan ikke kjøre terraform apply / onboarde kunder, "
|
||||||
"selge Vauco OS eller OPAX, eller passere godkjenningsporter uten Chris. "
|
"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), "
|
"Du har tilgang til MCP-verktøy: get_build_status (live CI/CD-status), "
|
||||||
"get_build_status (sjekk CI/CD-status), deploy_service (trigger deploy), "
|
"get_logs (Cloud Run-logger), push_static (last opp filer til opax.vauco.no), "
|
||||||
"get_logs (hent Cloud Run-logger). Bruk disse når det er relevant. "
|
"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."
|
"Svar på norsk (bokmål) med mindre annet er bedt om."
|
||||||
)
|
)
|
||||||
tools = []
|
tools = []
|
||||||
if rag_tool:
|
if rag_tool:
|
||||||
tools.append(rag_tool)
|
tools.append(rag_tool)
|
||||||
if opax_mcp_toolset:
|
tools.extend(opax_tools)
|
||||||
tools.append(opax_mcp_toolset)
|
|
||||||
return Agent(
|
return Agent(
|
||||||
model=models["orchestrator"],
|
model=models["orchestrator"],
|
||||||
name="opax_agent",
|
name="opax_agent",
|
||||||
|
|
@ -178,7 +174,7 @@ async def _run_async(
|
||||||
mode = _normalize_mode(mode)
|
mode = _normalize_mode(mode)
|
||||||
agent = build_agent(mode=mode)
|
agent = build_agent(mode=mode)
|
||||||
models = get_models_for_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_service = InMemorySessionService()
|
||||||
session = await session_service.create_session(
|
session = await session_service.create_session(
|
||||||
|
|
@ -201,17 +197,16 @@ async def _run_async(
|
||||||
input_tokens += getattr(um, "prompt_token_count", 0) or 0
|
input_tokens += getattr(um, "prompt_token_count", 0) or 0
|
||||||
output_tokens += getattr(um, "candidates_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:
|
if input_tokens > 0 or output_tokens > 0:
|
||||||
log_token_usage(
|
log_token_usage(
|
||||||
agent_name=module_name, # bakoverkompatibelt
|
agent_name=module_name,
|
||||||
model_name=models["orchestrator"],
|
model_name=models["orchestrator"],
|
||||||
input_tokens=input_tokens,
|
input_tokens=input_tokens,
|
||||||
output_tokens=output_tokens,
|
output_tokens=output_tokens,
|
||||||
request_id=request_id,
|
request_id=request_id,
|
||||||
module_name=module_name, # CG4: 'jason/light' | 'jason/heavy'
|
module_name=module_name,
|
||||||
caller_type=caller_type, # CG4: agent|browser|cli|cron|api
|
caller_type=caller_type,
|
||||||
session_id=session_id, # CG4: for sesjon-gruppering
|
session_id=session_id,
|
||||||
)
|
)
|
||||||
|
|
||||||
return final_text
|
return final_text
|
||||||
|
|
@ -224,16 +219,6 @@ def run(
|
||||||
mode: str = "light",
|
mode: str = "light",
|
||||||
caller_type: str = "agent",
|
caller_type: str = "agent",
|
||||||
) -> str:
|
) -> 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)
|
mode = _normalize_mode(mode)
|
||||||
authorize_mode(user_id, mode)
|
authorize_mode(user_id, mode)
|
||||||
return asyncio.run(_run_async(
|
return asyncio.run(_run_async(
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user