diff --git a/services/comms/atlasbot-deployment.yaml b/services/comms/atlasbot-deployment.yaml index d24cba2..f4e7f7d 100644 --- a/services/comms/atlasbot-deployment.yaml +++ b/services/comms/atlasbot-deployment.yaml @@ -16,7 +16,7 @@ spec: labels: app: atlasbot annotations: - checksum/atlasbot-configmap: manual-atlasbot-43 + checksum/atlasbot-configmap: manual-atlasbot-44 vault.hashicorp.com/agent-inject: "true" vault.hashicorp.com/role: "comms" vault.hashicorp.com/agent-inject-secret-turn-secret: "kv/data/atlas/comms/turn-shared-secret" diff --git a/services/comms/scripts/atlasbot/bot.py b/services/comms/scripts/atlasbot/bot.py index bee72e9..4316fe0 100644 --- a/services/comms/scripts/atlasbot/bot.py +++ b/services/comms/scripts/atlasbot/bot.py @@ -1868,9 +1868,12 @@ class _AtlasbotHandler(BaseHTTPRequestHandler): workloads=workloads, ) fallback = "I don't have enough data to answer that." + llm_prompt = cleaned + if cluster_query: + llm_prompt = f"Atlas cluster question: {cleaned}" answer = ollama_reply( ("http", "internal"), - cleaned, + llm_prompt, context=context, fallback=fallback, use_history=False, @@ -2313,11 +2316,14 @@ def sync_loop(token: str, room_id: str): context = (context + "\n\n" + extra).strip() if context else extra fallback = "I don't have enough data to answer that." + llm_prompt = cleaned_body + if cluster_query: + llm_prompt = f"Atlas cluster question: {cleaned_body}" reply = ollama_reply_with_thinking( token, rid, hist_key, - cleaned_body, + llm_prompt, context=context, fallback=fallback, use_history=cluster_query,