@rubixkube/rubix 0.0.2 → 0.0.3

package/CHANGELOG.md

@@ -0,0 +1,29 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com),
+and this project adheres to [Semantic Versioning](https://semver.org).
+
+## [0.0.3]
+
+### Added
+
+- /models to switch between models
+- Persistent user settings
+- Bugfixes and performance improvements
+
+
+## [0.0.2]
+
+### Added
+
+- Slash commands
+- Session history and auto-resume
+- Streaming workflow event tracing
+
+## [0.0.1]
+
+### Added
+
+- Initial release

package/LICENSE

@@ -0,0 +1,33 @@
+Copyright (c) 2026 Rubixkube Intelligence Pvt. Ltd. All rights reserved.
+
+RUBIXKUBE PROPRIETARY SOFTWARE LICENSE AGREEMENT
+
+IMPORTANT: PLEASE READ THIS LICENSE AGREEMENT CAREFULLY. BY INSTALLING OR USING THE RUBIX CLI ("SOFTWARE"), YOU AGREE TO BE BOUND BY THE TERMS OF THIS AGREEMENT.
+
+RubixKube and Site Reliability Intelligence are trademarks of Rubixkube Intelligence Pvt. Ltd.
+
+1. GRANT OF LICENSE
+Rubixkube Intelligence Pvt. Ltd. ("Company") grants you a non-exclusive, non-transferable, limited license to use the Software solely for your internal business purposes, provided that you have a valid and active account on the RubixKube platform (https://console.rubixkube.ai).
+
+2. RESTRICTIONS
+You shall not, and shall not permit any third party to:
+(a) copy, modify, or create derivative works of the Software;
+(b) redistribute, sell, lease, sublicense, or otherwise transfer the Software to any third party;
+(c) reverse engineer, decompile, disassemble, or attempt to derive the source code of the Software;
+(d) remove or alter any copyright, trademark, or other proprietary notices;
+(e) use the Software in any manner that violates applicable laws or regulations.
+
+3. INTELLECTUAL PROPERTY
+The Software is licensed, not sold. All title, ownership rights, and intellectual property rights in and to the Software (including but not limited to any images, animations, video, audio, music, and text incorporated into the Software) are owned by Rubixkube Intelligence Pvt. Ltd. This Agreement does not grant you any rights to trademarks or service marks of the Company.
+
+4. TERMINATION
+This license is effective until terminated. Your rights under this license will terminate automatically without notice from the Company if you fail to comply with any term(s) of this Agreement or if your RubixKube account is suspended or terminated. Upon termination, you must cease all use of the Software and destroy all copies.
+
+5. DISCLAIMER OF WARRANTIES
+THE SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND. TO THE MAXIMUM EXTENT PERMITTED BY LAW, THE COMPANY DISCLAIMS ALL WARRANTIES, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND NON-INFRINGEMENT.
+
+6. LIMITATION OF LIABILITY
+IN NO EVENT SHALL THE COMPANY BE LIABLE FOR ANY SPECIAL, INCIDENTAL, INDIRECT, OR CONSEQUENTIAL DAMAGES WHATSOEVER (INCLUDING, WITHOUT LIMITATION, DAMAGES FOR LOSS OF BUSINESS PROFITS, BUSINESS INTERRUPTION, OR LOSS OF BUSINESS INFORMATION) ARISING OUT OF THE USE OF OR INABILITY TO USE THE SOFTWARE.
+
+7. CONTACT
+For any questions regarding this license, please contact: connect@rubixkube.ai.

package/README.md

@@ -1,8 +1,14 @@
 # Rubix CLI
 
-Chat with your infrastructure from the terminal. **RubixKube** CLI for Site Reliability Intelligence—predict, prevent, and fix failures with AI.
+[![npm version](https://img.shields.io/npm/v/@rubixkube/rubix.svg)](https://www.npmjs.com/package/@rubixkube/rubix)
+[![npm downloads](https://img.shields.io/npm/dm/@rubixkube/rubix.svg)](https://www.npmjs.com/package/@rubixkube/rubix)
+[![License: Proprietary](https://img.shields.io/badge/License-Proprietary-red.svg)](https://rubixkube.ai)
 
-![Rubix CLI](https://raw.githubusercontent.com/rubixkube-io/rubixkube-ai/main/public/assets/rubix-cli.png)
+Built for SREs and platform engineers who own production reliability. **Rubix** brings RubixKube's Site Reliability Intelligence directly into your terminal — investigate incidents, understand blast radius, and get AI-driven RCA without leaving the command line.
+
+Rubix connects to the RubixKube cloud platform — no local setup beyond a free account.
+
+![Rubix CLI](https://rubixkube.ai/assets/rubix-cli.png)
 
 ## Install
 
@@ -18,24 +24,35 @@ npx @rubixkube/rubix
 
 ## Quick Start
 
-1. Run `rubix` to start the chat UI
-2. On first run, choose **Login** and complete the Auth0 device flow
-3. Start asking questions about your infrastructure
+```bash
+rubix
+```
+
+On first run, you'll be prompted to log in via a browser link — takes 30 seconds. After that, you're in.
+
+Start asking:
+
+```
+> What's causing the latency spike in production?
+> Show me recent incidents for the payments service
+> What changed before this alert fired?
+```
 
-## Features
+## What You Can Do
 
-- **Auth** — Auth0 device code flow; credentials stored in `~/.rubix/auth.json`
-- **Chat** — Real-time streaming to RubixKube; session resume; cluster-aware
-- **Slash commands** — `/login`, `/logout`, `/status`, `/sessions`, `/cluster`, `/new`, `/clear`, `/console`, `/help`, `/exit`
-- **Shortcuts** — `?` help, `Tab` autocomplete, `Ctrl+X` then `H/C/Q` for help/clear/quit
+- **Investigate incidents** — ask natural language questions about your live infrastructure
+- **Get RCA** — AI-generated root cause analysis, right in the terminal
+- **Track multiple clusters** — switch context without leaving the chat
+- **Resume sessions** — pick up where you left off across terminal sessions
+- **Slash commands** — `/new`, `/sessions`, `/status`, `/clear`, `/help`, `/exit`
 
 ## Requirements
 
 - Node.js 18+
-- [RubixKube account](https://console.rubixkube.ai) (free signup)
+- [RubixKube account](https://console.rubixkube.ai) — free to sign up
 
 ## Links
 
-- [RubixKube](https://rubixkube.ai) — Site Reliability Intelligence platform
+- [rubixkube.ai](https://rubixkube.ai) — Site Reliability Intelligence platform
 - [Documentation](https://docs.rubixkube.ai)
 - [Console](https://console.rubixkube.ai)

package/dist/cli.js

@@ -5,6 +5,7 @@ import { VERSION } from "./version.js";
 import { runLoginCommand } from "./commands/login.js";
 import { runLogoutCommand } from "./commands/logout.js";
 import { runSessionsCommand } from "./commands/sessions.js";
+import { runModelCommand } from "./commands/model.js";
 async function main() {
     if (process.argv.length <= 2) {
         await runChatCommand({});
@@ -41,6 +42,12 @@ async function main() {
         .action(async () => {
         await runSessionsCommand();
     });
+    program
+        .command("model [subcommand] [modelId]")
+        .description("Manage AI models")
+        .action(async (subcommand, modelId) => {
+        await runModelCommand({ subcommand, modelId });
+    });
     await program.parseAsync(process.argv);
 }
 main().catch((error) => {

package/dist/commands/login.js

@@ -1,6 +1,27 @@
-import { saveAuthConfig } from "../core/auth-store.js";
-import { authenticateWithDeviceFlow } from "../core/device-auth.js";
+import { loadAuthConfig, saveAuthConfig } from "../core/auth-store.js";
+import { authenticateWithDeviceFlow, isTokenNearExpiry } from "../core/device-auth.js";
+import { refreshAndUpdateAuth } from "../core/rubix-api.js";
 export async function runLoginCommand() {
+    let existing = await loadAuthConfig();
+    if (existing?.isAuthenticated && (existing.idToken || existing.authToken)) {
+        let tokenValid = !isTokenNearExpiry(existing.idToken ?? existing.authToken);
+        if (!tokenValid && existing.refreshToken) {
+            try {
+                console.log("Session token expired, attempting to refresh...");
+                existing = await refreshAndUpdateAuth(existing);
+                await saveAuthConfig(existing);
+                tokenValid = true;
+            }
+            catch (error) {
+                console.log("Failed to refresh token. Need to log in again.");
+            }
+        }
+        if (tokenValid) {
+            console.log(`Already authenticated as ${existing?.userName ?? existing?.userEmail ?? "user"}.`);
+            console.log("Run `rubix logout` first if you want to switch accounts.");
+            return;
+        }
+    }
     console.log("Starting Rubix device authentication...");
     try {
         const authConfig = await authenticateWithDeviceFlow((message) => {

package/dist/commands/model.js

@@ -0,0 +1,84 @@
+import { loadAuthConfig } from "../core/auth-store.js";
+import { listModels, setSessionModel, listSessions } from "../core/rubix-api.js";
+export async function runModelCommand(args) {
+    try {
+        const auth = await loadAuthConfig();
+        if (!auth?.isAuthenticated) {
+            console.error("Not authenticated. Run `rubix login` first.");
+            process.exitCode = 1;
+            return;
+        }
+        const subcommand = args.subcommand?.toLowerCase() ?? "list";
+        switch (subcommand) {
+            case "list": {
+                const models = await listModels(auth);
+                if (models.length === 0) {
+                    console.log("No models available.");
+                    return;
+                }
+                console.log("\nAvailable models:\n");
+                for (const model of models) {
+                    const marker = model.default ? "●" : " ";
+                    const experimental = model.experimental ? " (experimental)" : "";
+                    const thinking = model.thinking_supported ? " · thinking" : "";
+                    console.log(`${marker} ${model.id.padEnd(12)} ${model.display_name}`);
+                    if (model.description) {
+                        console.log(`  ${model.description}`);
+                    }
+                    if (experimental || thinking) {
+                        console.log(`  ${experimental}${thinking}`.trim());
+                    }
+                    console.log();
+                }
+                break;
+            }
+            case "set": {
+                if (!args.modelId) {
+                    console.error("Usage: rubix model set <model_id>");
+                    console.error("Example: rubix model set expert");
+                    process.exitCode = 1;
+                    return;
+                }
+                const sessions = await listSessions(auth, 1, 0);
+                if (sessions.length === 0) {
+                    console.error("No active session. Use `rubix chat` to create one.");
+                    process.exitCode = 1;
+                    return;
+                }
+                const sessionId = sessions[0].id;
+                const result = await setSessionModel(auth, sessionId, args.modelId);
+                console.log(`Switched to: ${result.displayName}`);
+                if (result.thinkingSupported) {
+                    console.log("(This model supports extended thinking)");
+                }
+                break;
+            }
+            case "current": {
+                const sessions = await listSessions(auth, 1, 0);
+                if (sessions.length === 0) {
+                    console.log("No active session.");
+                    process.exitCode = 1;
+                    return;
+                }
+                const models = await listModels(auth);
+                const defaultModel = models.find((m) => m.default);
+                if (defaultModel) {
+                    console.log(`Current model: ${defaultModel.display_name} (${defaultModel.id})`);
+                }
+                else {
+                    console.log("No model information available.");
+                }
+                break;
+            }
+            default:
+                console.error(`Unknown subcommand: ${subcommand}`);
+                console.error("Usage: rubix model [list|set|current]");
+                process.exitCode = 1;
+        }
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        console.error(`Failed: ${message}`);
+        process.exitCode = 1;
+    }
+}

package/dist/core/device-auth.js

@@ -19,6 +19,19 @@ function decodeJwtClaims(token) {
     const decoded = Buffer.from(padded, "base64").toString("utf8");
     return JSON.parse(decoded);
 }
+export function isTokenNearExpiry(token, thresholdMs = 5 * 60 * 1000) {
+    if (!token)
+        return false;
+    try {
+        const claims = decodeJwtClaims(token);
+        if (!claims.exp)
+            return false;
+        return Date.now() + thresholdMs > claims.exp * 1000;
+    }
+    catch {
+        return false;
+    }
+}
 export async function startDeviceAuth() {
     const missing = getMissingEnv(AUTH_REQUIRED_ENV);
     if (missing.length > 0) {
@@ -27,7 +40,7 @@ export async function startDeviceAuth() {
     const { auth0ClientId, auth0Audience } = getConfig();
     const body = new URLSearchParams();
     body.set("client_id", auth0ClientId);
-    body.set("scope", "openid profile email");
+    body.set("scope", "openid profile email offline_access");
     if (auth0Audience) {
         body.set("audience", auth0Audience);
     }
@@ -153,3 +166,28 @@ export async function authenticateWithDeviceFlow(log) {
         timestamp: Date.now(),
     };
 }
+export async function refreshAccessToken(auth) {
+    if (!auth.refreshToken) {
+        throw new Error("No refresh token available. Please login again.");
+    }
+    const body = new URLSearchParams();
+    body.set("grant_type", "refresh_token");
+    body.set("refresh_token", auth.refreshToken);
+    body.set("client_id", getConfig().auth0ClientId);
+    const response = await fetch(`${auth0BaseUrl()}/oauth/token`, {
+        method: "POST",
+        headers: { "Content-Type": "application/x-www-form-urlencoded" },
+        body,
+    });
+    const raw = (await response.json());
+    if (!response.ok || !raw.id_token || !raw.access_token) {
+        throw new Error(`Token refresh failed: ${raw.error ?? response.statusText}`);
+    }
+    return {
+        ...auth,
+        authToken: raw.access_token,
+        idToken: raw.id_token,
+        refreshToken: raw.refresh_token ?? auth.refreshToken,
+        timestamp: Date.now(),
+    };
+}

package/dist/core/rubix-api.js

@@ -1,4 +1,6 @@
 import { getConfig } from "../config/env.js";
+import { refreshAccessToken } from "./device-auth.js";
+import { saveAuthConfig } from "./auth-store.js";
 const DEFAULT_APP_NAME = "SRI Agent";
 export class StreamError extends Error {
     reason;
@@ -21,7 +23,7 @@ function opelBase() {
 }
 function ensureAuth(auth) {
     const token = auth.idToken ?? auth.authToken;
-    const userId = auth.userId ?? auth.userEmail;
+    const userId = auth.userEmail ?? auth.userId;
     if (!token) {
         throw new Error("Missing auth token. Run /login.");
     }
@@ -48,6 +50,38 @@ function headers(auth, includeTenant = true) {
     }
     return out;
 }
+export async function refreshAndUpdateAuth(auth) {
+    try {
+        const refreshed = await refreshAccessToken(auth);
+        await saveAuthConfig(refreshed);
+        return refreshed;
+    }
+    catch (error) {
+        const msg = error instanceof Error ? error.message : String(error);
+        throw new StreamError(`Token refresh failed: ${msg}. Please run /login again.`, {
+            reason: "http_error",
+            status: 401,
+        });
+    }
+}
+async function fetchWithAutoRefresh(auth, url, options) {
+    let response = await fetch(url, options);
+    if (response.status === 401) {
+        try {
+            auth = await refreshAndUpdateAuth(auth);
+            const newHeaders = { ...options.headers };
+            const token = auth.idToken ?? auth.authToken;
+            if (token) {
+                newHeaders.Authorization = `Bearer ${token}`;
+            }
+            response = await fetch(url, { ...options, headers: newHeaders });
+        }
+        catch {
+            return response;
+        }
+    }
+    return response;
+}
 async function parseJsonResponse(response) {
     if (!response.ok) {
         const text = await response.text();
@@ -159,6 +193,19 @@ function parseToolNameFromText(value) {
         return callMatch[1];
     return "";
 }
+export async function listApps(auth) {
+    const url = `${opelBase()}/apps`;
+    const response = await fetchWithAutoRefresh(auth, url, {
+        method: "GET",
+        headers: headers(auth),
+    });
+    if (!response.ok) {
+        const text = await response.text();
+        throw new Error(`Failed to load agents (${response.status}): ${text}`);
+    }
+    const payload = await parseJsonResponse(response);
+    return payload.apps ?? [DEFAULT_APP_NAME];
+}
 function normalizeWorkflowEvent(type, content, details) {
     return {
         id: `${type}-${Date.now()}-${Math.random().toString(16).slice(2)}`,
@@ -171,7 +218,7 @@ function normalizeWorkflowEvent(type, content, details) {
 export async function listSessions(auth, limit = 20, offset = 0) {
     const { userId } = ensureAuth(auth);
     const url = `${opelBase()}/sessions/?user_id=${encodeURIComponent(userId)}&limit=${limit}&offset=${offset}`;
-    const response = await fetch(url, {
+    const response = await fetchWithAutoRefresh(auth, url, {
         method: "GET",
         headers: headers(auth),
     });
@@ -190,7 +237,7 @@ export async function listSessions(auth, limit = 20, offset = 0) {
     }))
         .sort((a, b) => new Date(b.updatedAt).getTime() - new Date(a.updatedAt).getTime());
 }
-export async function createSession(auth, appName = DEFAULT_APP_NAME, clusterId) {
+export async function createSession(auth, appName = DEFAULT_APP_NAME, clusterId, modelId) {
     const { userId, tenantId } = ensureAuth(auth);
     if (!tenantId) {
         throw new Error("Missing tenant ID. Run /login.");
@@ -208,11 +255,12 @@ export async function createSession(auth, appName = DEFAULT_APP_NAME, clusterId)
             session_created_at: new Date().toISOString(),
             client_type: "rubix-cli",
             ...(clusterId ? { cluster_id: clusterId, default_namespace: "default" } : {}),
+            ...(modelId ? { model_id: modelId } : {}),
             ...(auth.userRole ? { user_role: auth.userRole } : {}),
             ...(auth.tenantPlan ? { tenant_plan: auth.tenantPlan } : {}),
         },
     };
-    const response = await fetch(`${opelBase()}/sessions/`, {
+    const response = await fetchWithAutoRefresh(auth, `${opelBase()}/sessions/`, {
         method: "POST",
         headers: headers(auth),
         body: JSON.stringify(payload),
@@ -223,9 +271,9 @@ export async function createSession(auth, appName = DEFAULT_APP_NAME, clusterId)
     }
     return parsed.id;
 }
-export async function updateSessionState(auth, sessionId, state) {
-    const url = `${opelBase()}/sessions/${encodeURIComponent(sessionId)}?app_name=${encodeURIComponent(DEFAULT_APP_NAME)}`;
-    const response = await fetch(url, {
+export async function updateSessionState(auth, sessionId, state, appName = DEFAULT_APP_NAME) {
+    const url = `${opelBase()}/sessions/${encodeURIComponent(sessionId)}?app_name=${encodeURIComponent(appName)}`;
+    const response = await fetchWithAutoRefresh(auth, url, {
         method: "PUT",
         headers: headers(auth),
         body: JSON.stringify({ state }),
@@ -235,17 +283,43 @@ export async function updateSessionState(auth, sessionId, state) {
         throw new Error(`Failed to update session (${response.status}): ${text}`);
     }
 }
-export async function getOrCreateSession(auth, preferredId, clusterId) {
+export async function hasSessionMessages(auth, sessionId) {
+    try {
+        const { userId } = ensureAuth(auth);
+        // Fetch just 1 history event — cheapest possible check.
+        const url = `${opelBase()}/sessions/${encodeURIComponent(sessionId)}/chat-history?user_id=${encodeURIComponent(userId)}&limit=1&offset=0&format=detailed&order_desc=false`;
+        const response = await fetchWithAutoRefresh(auth, url, {
+            method: "GET",
+            headers: headers(auth),
+        });
+        if (!response.ok)
+            return false;
+        const payload = (await response.json());
+        return (payload.chat_history?.length ?? 0) > 0;
+    }
+    catch {
+        // If we can't determine, assume no messages so we reuse rather than spam new sessions.
+        return false;
+    }
+}
+export async function getOrCreateSession(auth, preferredId, clusterId, appName = DEFAULT_APP_NAME) {
     if (preferredId)
         return preferredId;
     const sessions = await listSessions(auth, 50, 0);
     if (sessions.length > 0) {
-        const recent = sessions[0];
+        // Attempt to find the most recent empty session for the same app
+        // Note: since list_sessions merges across apps now, we iterate
+        // until we find one that matches our appName.
+        const recent = sessions.find(s => s.appName === appName) ?? sessions[0];
         // Reuse the session only if it has matching cluster context (or no cluster was requested)
-        if (!clusterId || recent.clusterId === clusterId)
-            return recent.id;
+        // AND it has no messages — otherwise start fresh.
+        if (!clusterId || recent.clusterId === clusterId) {
+            const isEmpty = !(await hasSessionMessages(auth, recent.id));
+            if (isEmpty)
+                return recent.id;
+        }
     }
-    return createSession(auth, undefined, clusterId);
+    return createSession(auth, appName, clusterId);
 }
 const HEALTHY_STATUSES = new Set(["connected", "healthy", "active"]);
 export async function listClusters(auth) {
@@ -257,7 +331,7 @@ export async function listClusters(auth) {
         throw new Error("Missing tenant ID. Run /login.");
     }
     const url = `${authBase.replace(/\/+$/, "")}/clusters/?page=1&page_size=50`;
-    const response = await fetch(url, {
+    const response = await fetchWithAutoRefresh(auth, url, {
         method: "GET",
         headers: {
             Authorization: `Bearer ${token}`,
@@ -284,6 +358,75 @@ export async function listClusters(auth) {
 export function firstHealthyCluster(clusters) {
     return clusters.find((c) => HEALTHY_STATUSES.has(c.status)) ?? clusters[0] ?? null;
 }
+export async function listModels(auth) {
+    const url = `${opelBase()}/models`;
+    const response = await fetchWithAutoRefresh(auth, url, {
+        method: "GET",
+        headers: headers(auth, false),
+    });
+    if (!response.ok) {
+        const text = await response.text();
+        throw new StreamError(`Failed to load models (${response.status}): ${text}`, {
+            status: response.status,
+            reason: "http_error",
+        });
+    }
+    const payload = await parseJsonResponse(response);
+    return (payload.models ?? [])
+        .filter((m) => !!m?.id)
+        .map((m) => ({
+        id: m.id ?? "",
+        model: m.model ?? "",
+        display_name: m.display_name ?? "",
+        description: m.description,
+        thinking_supported: m.thinking_supported ?? false,
+        agent: m.agent ?? "sre_agent",
+        default: m.default ?? false,
+        experimental: m.experimental,
+    }));
+}
+export async function setSessionModel(auth, sessionId, modelId, appName = DEFAULT_APP_NAME) {
+    const { userId } = ensureAuth(auth);
+    const url = `${opelBase()}/sessions/${encodeURIComponent(sessionId)}/model`;
+    const response = await fetchWithAutoRefresh(auth, url, {
+        method: "POST",
+        headers: headers(auth, true),
+        body: JSON.stringify({
+            user_id: userId,
+            model_id: modelId,
+            app_name: appName,
+        }),
+    });
+    if (!response.ok) {
+        const text = await response.text();
+        let errorMsg = `HTTP ${response.status}: ${text || response.statusText}`;
+        try {
+            const errorPayload = JSON.parse(text);
+            const error = errorPayload.detail?.error ?? "Unknown error";
+            const available = errorPayload.detail?.available ?? [];
+            if (available.length > 0) {
+                errorMsg = `${error}. Available: ${available.join(", ")}`;
+            }
+            else {
+                errorMsg = error;
+            }
+        }
+        catch {
+            // Fall back to generic error message
+        }
+        throw new StreamError(errorMsg, {
+            status: response.status,
+            reason: "http_error",
+        });
+    }
+    const parsed = await parseJsonResponse(response);
+    return {
+        modelId: parsed.model_id ?? modelId,
+        model: parsed.model ?? "",
+        displayName: parsed.display_name ?? "",
+        thinkingSupported: parsed.thinking_supported ?? false,
+    };
+}
 function parseParts(content) {
     if (!content)
         return [];
@@ -301,7 +444,7 @@ function parseParts(content) {
 export async function fetchChatHistory(auth, sessionId, limit = 50) {
     const { userId } = ensureAuth(auth);
     const url = `${opelBase()}/sessions/${encodeURIComponent(sessionId)}/chat-history?user_id=${encodeURIComponent(userId)}&limit=${limit}&offset=0&format=detailed&order_desc=false`;
-    const response = await fetch(url, {
+    const response = await fetchWithAutoRefresh(auth, url, {
         method: "GET",
         headers: headers(auth),
     });
@@ -318,15 +461,25 @@ export async function fetchChatHistory(auth, sessionId, limit = 50) {
         let text = "";
         const workflow = [];
         for (const part of parts) {
-            if (part.thought === true)
+            if (part.thought === true) {
+                if (typeof part.text === "string" && part.text.trim()) {
+                    workflow.push({
+                        id: `hist-th-${idx}-${workflow.length}`,
+                        type: "thought",
+                        content: part.text.trim(),
+                        ts,
+                    });
+                }
                 continue;
+            }
             const fc = part.functionCall ?? part.function_call;
             if (fc) {
                 const name = typeof fc.name === "string" ? fc.name : "tool";
+                const argsStr = fc.args && Object.keys(fc.args).length > 0 ? JSON.stringify(fc.args) : "";
                 workflow.push({
                     id: `hist-fc-${idx}-${name}`,
                     type: "function_call",
-                    content: name,
+                    content: argsStr,
                     ts,
                     details: { name, id: fc.id },
                 });
@@ -338,7 +491,7 @@ export async function fetchChatHistory(auth, sessionId, limit = 50) {
                 workflow.push({
                     id: `hist-fr-${idx}-${name}`,
                     type: "function_response",
-                    content: typeof fr.response === "string" ? fr.response : `[${name}]`,
+                    content: typeof fr.response === "string" ? fr.response : (fr.response ? JSON.stringify(fr.response) : `[${name}]`),
                     ts,
                     details: { name, id: fr.id },
                 });
@@ -362,7 +515,7 @@ export async function fetchChatHistory(auth, sessionId, limit = 50) {
 }
 export async function streamChat(input, callbacks = {}) {
     const { userId } = ensureAuth(input.auth);
-    const response = await fetch(`${opelBase()}/chat/${encodeURIComponent(userId)}/session/${encodeURIComponent(input.sessionId)}`, {
+    const response = await fetchWithAutoRefresh(input.auth, `${opelBase()}/chat/${encodeURIComponent(userId)}/session/${encodeURIComponent(input.sessionId)}`, {
         method: "POST",
         headers: headers(input.auth, true),
         signal: input.signal,
@@ -376,6 +529,8 @@ export async function streamChat(input, callbacks = {}) {
             streaming: true,
             minify: true,
             maxTextLen: -1,
+            ...(input.stateDelta ? { stateDelta: input.stateDelta } : {}),
+            ...(input.modelOverride ? { modelOverride: input.modelOverride } : {}),
         }),
     });
     if (!response.ok) {
@@ -453,8 +608,8 @@ export async function streamChat(input, callbacks = {}) {
                 hasWorkflowEvents = true;
                 const name = asText(functionCall.name) || "tool";
                 const args = functionCall.args ?? {};
-                const prettyArgs = Object.keys(args).length > 0 ? `: ${JSON.stringify(args)}` : "";
-                callbacks.onWorkflow?.(normalizeWorkflowEvent("function_call", `${name}${prettyArgs}`, {
+                const argsStr = Object.keys(args).length > 0 ? JSON.stringify(args) : "";
+                callbacks.onWorkflow?.(normalizeWorkflowEvent("function_call", argsStr, {
                     name,
                     id: functionCall.id,
                 }));