zgsm-ai
diff --git a/‎apps/cli/README.md‎
Lines changed: 13 additions & 12 deletions b/‎apps/cli/README.md‎
Lines changed: 13 additions & 12 deletions
diff --git a/‎apps/cli/install.sh‎
Lines changed: 10 additions & 10 deletions b/‎apps/cli/install.sh‎
Lines changed: 10 additions & 10 deletions
diff --git a/‎apps/cli/package.json‎
Lines changed: 3 additions & 3 deletions b/‎apps/cli/package.json‎
Lines changed: 3 additions & 3 deletions
diff --git a/‎apps/cli/src/agent/extension-host.ts‎
Lines changed: 8 additions & 2 deletions b/‎apps/cli/src/agent/extension-host.ts‎
Lines changed: 8 additions & 2 deletions
diff --git a/‎apps/cli/src/index.ts‎
Lines changed: 2 additions & 2 deletions b/‎apps/cli/src/index.ts‎
Lines changed: 2 additions & 2 deletions
diff --git a/‎apps/cli/src/lib/utils/provider.ts‎
Lines changed: 4 additions & 0 deletions b/‎apps/cli/src/lib/utils/provider.ts‎
Lines changed: 4 additions & 0 deletions
diff --git a/‎packages/types/src/mcp.ts‎
Lines changed: 56 additions & 0 deletions b/‎packages/types/src/mcp.ts‎
Lines changed: 56 additions & 0 deletions
diff --git a/‎packages/types/src/message.ts‎
Lines changed: 2 additions & 0 deletions b/‎packages/types/src/message.ts‎
Lines changed: 2 additions & 0 deletions
diff --git a/‎packages/types/src/providers/cerebras.ts‎
Lines changed: 0 additions & 11 deletions b/‎packages/types/src/providers/cerebras.ts‎
Lines changed: 0 additions & 11 deletions
diff --git a/‎src/core/costrict/activate.ts‎
Lines changed: 1 addition & 1 deletion b/‎src/core/costrict/activate.ts‎
Lines changed: 1 addition & 1 deletion
@@ -44,7 +44,7 @@ curl -fsSL https://raw.githubusercontent.com/RooCodeInc/Roo-Code/main/apps/cli/i
 ### Uninstalling
 
 ```bash
-rm -rf ~/.roo/cli ~/.local/bin/roo
+rm -rf ~/.roo/cli ~/.local/bin/cos
 ```
 
 ### Development Installation
@@ -71,13 +71,13 @@ By default, the CLI prompts for approval before executing actions:
 ```bash
 export OPENROUTER_API_KEY=sk-or-v1-...
 
-roo ~/Documents/my-project -P "What is this project?"
+cos ~/Documents/my-project -P "What is this project?"
 ```
 
 You can also run without a prompt and enter it interactively in TUI mode:
 
 ```bash
-roo ~/Documents/my-project
+cos ~/Documents/my-project
 ```
 
 In interactive mode:
@@ -92,7 +92,7 @@ In interactive mode:
 For automation and scripts, use `-y` to auto-approve all actions:
 
 ```bash
-roo ~/Documents/my-project -y -P "Refactor the utils.ts file"
+cos ~/Documents/my-project -y -P "Refactor the utils.ts file"
 ```
 
 In non-interactive mode:
@@ -107,20 +107,20 @@ To use Roo Code Cloud features (like the provider proxy), you need to authentica
 
 ```bash
 # Log in to Roo Code Cloud (opens browser)
-roo auth login
+cos auth login
 
 # Check authentication status
-roo auth status
+cos auth status
 
 # Log out
-roo auth logout
+cos auth logout
 ```
 
 The `auth login` command:
 
 1. Opens your browser to authenticate with Roo Code Cloud
 2. Receives a secure token via localhost callback
-3. Stores the token in `~/.config/roo/credentials.json`
+3. Stores the token in `~/.config/cos/credentials.json`
 
 Tokens are valid for 90 days. The CLI will prompt you to re-authenticate when your token expires.
 
@@ -167,16 +167,17 @@ Tokens are valid for 90 days. The CLI will prompt you to re-authenticate when yo
 
 | Command           | Description                        |
 | ----------------- | ---------------------------------- |
-| `roo auth login`  | Authenticate with Roo Code Cloud   |
-| `roo auth logout` | Clear stored authentication token  |
-| `roo auth status` | Show current authentication status |
+| `cos auth login`  | Authenticate with Roo Code Cloud   |
+| `cos auth logout` | Clear stored authentication token  |
+| `cos auth status` | Show current authentication status |
 
 ## Environment Variables
 
 The CLI will look for API keys in environment variables if not provided via `--api-key`:
 
 | Provider      | Environment Variable |
 | ------------- | -------------------- |
+| zgsm          | `COSTRICT_API_KEY`   |
 | anthropic     | `ANTHROPIC_API_KEY`  |
 | openai        | `OPENAI_API_KEY`     |
 | openrouter    | `OPENROUTER_API_KEY` |
@@ -248,7 +249,7 @@ pnpm lint
 To create a new release, execute the /cli-release slash command:
 
 ```bash
-roo ~/Documents/Roo-Code -P "/cli-release" -y
+cos ~/Documents/Roo-Code -P "/cli-release" -y
 ```
 
 The workflow will:
 
@@ -194,7 +194,7 @@ Available at: https://github.com/$REPO/releases"
     fi
 
     # Make executable
-    chmod +x "$INSTALL_DIR/bin/roo"
+    chmod +x "$INSTALL_DIR/bin/cos"
 
     # Also make ripgrep executable if it exists
     if [ -f "$INSTALL_DIR/bin/rg" ]; then
@@ -207,12 +207,12 @@ setup_bin() {
     mkdir -p "$BIN_DIR"
 
     # Remove old symlink if exists
-    if [ -L "$BIN_DIR/roo" ] || [ -f "$BIN_DIR/roo" ]; then
-        rm -f "$BIN_DIR/roo"
+    if [ -L "$BIN_DIR/cos" ] || [ -f "$BIN_DIR/cos" ]; then
+        rm -f "$BIN_DIR/cos"
     fi
 
-    ln -sf "$INSTALL_DIR/bin/roo" "$BIN_DIR/roo"
-    info "Created symlink: $BIN_DIR/roo"
+    ln -sf "$INSTALL_DIR/bin/cos" "$BIN_DIR/cos"
+    info "Created symlink: $BIN_DIR/cos"
 }
 
 # Check if bin dir is in PATH and provide instructions
@@ -257,10 +257,10 @@ check_path() {
 
 # Verify installation
 verify_install() {
-    if [ -x "$BIN_DIR/roo" ]; then
+    if [ -x "$BIN_DIR/cos" ]; then
         info "Verifying installation..."
         # Just check if it runs without error
-        "$BIN_DIR/roo" --version >/dev/null 2>&1 || true
+        "$BIN_DIR/cos" --version >/dev/null 2>&1 || true
     fi
 }
 
@@ -270,15 +270,15 @@ print_success() {
     printf "${GREEN}${BOLD}✓ Roo Code CLI installed successfully!${NC}\n"
     echo ""
     echo "  Installation: $INSTALL_DIR"
-    echo "  Binary: $BIN_DIR/roo"
+    echo "  Binary: $BIN_DIR/cos"
     echo "  Version: $VERSION"
     echo ""
     echo "  ${BOLD}Get started:${NC}"
-    echo "    roo --help"
+    echo "    cos --help"
     echo ""
     echo "  ${BOLD}Example:${NC}"
     echo "    export OPENROUTER_API_KEY=sk-or-v1-..."
-    echo "    roo ~/my-project -P \"What is this project?\""
+    echo "    cos ~/my-project -P \"What is this project?\""
     echo ""
 }
 
 
@@ -6,16 +6,16 @@
 	"type": "module",
 	"main": "dist/index.js",
 	"bin": {
-		"roo": "dist/index.js"
+		"cos": "dist/index.js"
 	},
 	"scripts": {
 		"format": "prettier --write 'src/**/*.ts'",
 		"lint": "eslint src --ext .ts --max-warnings=0",
 		"check-types": "tsc --noEmit",
 		"test": "vitest run",
 		"build": "tsup",
-		"build:extension": "pnpm --filter roo-cline bundle",
-		"build:all": "pnpm --filter roo-cline bundle && tsup",
+		"build:extension": "pnpm --filter zgsm bundle",
+		"build:all": "pnpm --filter zgsm bundle && tsup",
 		"dev": "tsup --watch",
 		"start": "ROO_AUTH_BASE_URL=http://localhost:3000 ROO_SDK_BASE_URL=http://localhost:3001 ROO_CODE_PROVIDER_URL=http://localhost:8080/proxy node dist/index.js",
 		"start:production": "node dist/index.js",
 
@@ -55,6 +55,8 @@ export interface ExtensionHostOptions {
 	provider: SupportedProvider
 	apiKey?: string
 	model: string
+	zgsmAccessToken?: string
+	zgsmModelId?: string
 	workspacePath: string
 	extensionPath: string
 	nonInteractive?: boolean
@@ -151,9 +153,9 @@ export class ExtensionHost extends EventEmitter implements ExtensionHostInterfac
 
 	constructor(options: ExtensionHostOptions) {
 		super()
-
 		this.options = options
 		this.options.integrationTest = true
+		const isZgsm = this?.options?.provider === "zgsm"
 
 		// Initialize client - single source of truth for agent state (including mode).
 		this.client = new ExtensionClient({
@@ -190,7 +192,11 @@ export class ExtensionHost extends EventEmitter implements ExtensionHostInterfac
 			commandExecutionTimeout: 30,
 			browserToolEnabled: false,
 			enableCheckpoints: false,
-			...getProviderSettings(this.options.provider, this.options.apiKey, this.options.model),
+			...getProviderSettings(
+				this.options.provider,
+				isZgsm ? this.options.zgsmAccessToken : this.options.apiKey,
+				isZgsm ? this.options.zgsmModelId : this.options.model,
+			),
 		}
 
 		this.initialSettings = this.options.nonInteractive
 
@@ -7,7 +7,7 @@ import { run, login, logout, status } from "@/commands/index.js"
 const program = new Command()
 
 program
-	.name("roo")
+	.name("cos")
 	.description("Roo Code CLI - starts an interactive session by default, use -p/--print for non-interactive output")
 	.version(VERSION)
 
@@ -20,7 +20,7 @@ program
 	.option("-d, --debug", "Enable debug output (includes detailed debug information)", false)
 	.option("-y, --yes, --dangerously-skip-permissions", "Auto-approve all prompts (use with caution)", false)
 	.option("-k, --api-key <key>", "API key for the LLM provider")
-	.option("--provider <provider>", "API provider (roo, anthropic, openai, openrouter, etc.)")
+	.option("--provider <provider>", "API provider (cos, anthropic, openai, openrouter, etc.)")
 	.option("-m, --model <model>", "Model to use", DEFAULT_FLAGS.model)
 	.option("--mode <mode>", "Mode to start in (code, architect, ask, debug, etc.)", DEFAULT_FLAGS.mode)
 	.option(
 
@@ -29,6 +29,10 @@ export function getProviderSettings(
 	const config: RooCodeSettings = { apiProvider: provider }
 
 	switch (provider) {
+		case "zgsm":
+			if (apiKey) config.zgsmAccessToken = apiKey
+			if (model) config.zgsmModelId = model
+			break
 		case "anthropic":
 			if (apiKey) config.apiKey = apiKey
 			if (model) config.apiModelId = model
 
@@ -1,5 +1,11 @@
 import { z } from "zod"
 
+/**
+ * Maximum number of MCP tools that can be enabled before showing a warning.
+ * LLMs tend to perform poorly when given too many tools to choose from.
+ */
+export const MAX_MCP_TOOLS_THRESHOLD = 60
+
 /**
  * McpServerUse
  */
@@ -136,3 +142,53 @@ export type McpErrorEntry = {
 	timestamp: number
 	level: "error" | "warn" | "info"
 }
+
+/**
+ * Result of counting enabled MCP tools across servers.
+ */
+export interface EnabledMcpToolsCount {
+	/** Number of enabled and connected MCP servers */
+	enabledServerCount: number
+	/** Total number of enabled tools across all enabled servers */
+	enabledToolCount: number
+}
+
+/**
+ * Count the number of enabled MCP tools across all enabled and connected servers.
+ * This is a pure function that can be used in both backend and frontend contexts.
+ *
+ * @param servers - Array of MCP server objects
+ * @returns Object with enabledToolCount and enabledServerCount
+ *
+ * @example
+ * const { enabledToolCount, enabledServerCount } = countEnabledMcpTools(mcpServers)
+ * if (enabledToolCount > MAX_MCP_TOOLS_THRESHOLD) {
+ *   // Show warning
+ * }
+ */
+export function countEnabledMcpTools(servers: McpServer[]): EnabledMcpToolsCount {
+	let serverCount = 0
+	let toolCount = 0
+
+	for (const server of servers) {
+		// Skip disabled servers
+		if (server.disabled) continue
+
+		// Skip servers that are not connected
+		if (server.status !== "connected") continue
+
+		serverCount++
+
+		// Count enabled tools on this server
+		if (server.tools) {
+			for (const tool of server.tools) {
+				// Tool is enabled if enabledForPrompt is undefined (default) or true
+				if (tool.enabledForPrompt !== false) {
+					toolCount++
+				}
+			}
+		}
+	}
+
+	return { enabledToolCount: toolCount, enabledServerCount: serverCount }
+}
@@ -151,6 +151,7 @@ export function isNonBlockingAsk(ask: ClineAsk): ask is NonBlockingAsk {
  * - `condense_context`: Context condensation/summarization has started
  * - `condense_context_error`: Error occurred during context condensation
  * - `codebase_search_result`: Results from searching the codebase
+ * - `too_many_tools_warning`: Warning that too many MCP tools are enabled, which may confuse the LLM
  */
 export const clineSays = [
 	"error",
@@ -184,6 +185,7 @@ export const clineSays = [
 	"sliding_window_truncation",
 	"codebase_search_result",
 	"user_edit_todos",
+	"too_many_tools_warning",
 ] as const
 
 export const clineSaySchema = z.enum(clineSays)
 
@@ -6,17 +6,6 @@ export type CerebrasModelId = keyof typeof cerebrasModels
 export const cerebrasDefaultModelId: CerebrasModelId = "gpt-oss-120b"
 
 export const cerebrasModels = {
-	"zai-glm-4.6": {
-		maxTokens: 16384, // Conservative default to avoid premature rate limiting (Cerebras reserves quota upfront)
-		contextWindow: 131072,
-		supportsImages: false,
-		supportsPromptCache: false,
-		supportsNativeTools: true,
-		defaultToolProtocol: "native",
-		inputPrice: 0,
-		outputPrice: 0,
-		description: "Fast general-purpose model on Cerebras (up to 1,000 tokens/s). To be deprecated soon.",
-	},
 	"zai-glm-4.7": {
 		maxTokens: 16384, // Conservative default to avoid premature rate limiting (Cerebras reserves quota upfront)
 		contextWindow: 131072,
 
@@ -139,7 +139,7 @@ export async function activate(
 					return
 				}
 				provider.log(`Login status detected at plugin startup: valid (${tokens.state})`)
-				writeCostrictAccessToken(tokens.access_token).then(async () => {
+				writeCostrictAccessToken(tokens.access_token, tokens.refresh_token).then(async () => {
 					await zgsmCodebaseIndexManager.initialize()
 					zgsmCodebaseIndexManager.syncToken()
 					workspaceEventMonitor.initialize()
Original file line number	Diff line number	Diff line change
`@@ -139,7 +139,7 @@ export async function activate(`
`139`	`139`	`return`
`140`	`140`	`}`
`141`	`141`	provider.log(`Login status detected at plugin startup: valid (${tokens.state})`)
`142`		`- writeCostrictAccessToken(tokens.access_token).then(async () => {`
	`142`	`+ writeCostrictAccessToken(tokens.access_token, tokens.refresh_token).then(async () => {`
`143`	`143`	`await zgsmCodebaseIndexManager.initialize()`
`144`	`144`	`zgsmCodebaseIndexManager.syncToken()`
`145`	`145`	`workspaceEventMonitor.initialize()`