@cjhyy/code-shell 0.1.0-alpha.1 → 0.1.0-alpha.3

package/dist/{chunk-ZJV3YSY3.cjs → chunk-BUJ5KYCI.cjs}

@@ -1,6 +1,6 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;
 
-var _chunkDICG4AB7cjs = require('./chunk-DICG4AB7.cjs');
+var _chunkGZBD6QRScjs = require('./chunk-GZBD6QRS.cjs');
 
 
 
@@ -8,12 +8,15 @@ var _chunkDICG4AB7cjs = require('./chunk-DICG4AB7.cjs');
 var _chunkEL2RL5DGcjs = require('./chunk-EL2RL5DG.cjs');
 
 
+var _chunkJHNCK7JPcjs = require('./chunk-JHNCK7JP.cjs');
+
+
 var _chunkIJSHQGMPcjs = require('./chunk-IJSHQGMP.cjs');
 
 // src/llm/providers/anthropic.ts
 _chunkIJSHQGMPcjs.init_cjs_shims.call(void 0, );
 var _sdk = require('@anthropic-ai/sdk'); var _sdk2 = _interopRequireDefault(_sdk);
-var AnthropicClient = (_class = class extends _chunkDICG4AB7cjs.LLMClientBase {
+var AnthropicClient = (_class = class extends _chunkGZBD6QRScjs.LLMClientBase {
   __init() {this._client = null}
   constructor(config) {
     super(config);_class.prototype.__init.call(this);;
@@ -34,10 +37,28 @@ var AnthropicClient = (_class = class extends _chunkDICG4AB7cjs.LLMClientBase {
     return this.withRetry(async () => {
       const messages = this.buildMessages(options.messages);
       const tools = options.tools ? this.convertTools(options.tools) : void 0;
-      if (options.stream && options.onChunk) {
-        return this.streamMessage(options, messages, tools);
+      const span = _chunkJHNCK7JPcjs.logger.span("llm.request", {
+        cat: "llm",
+        provider: this.provider,
+        model: this.model,
+        stream: !!(options.stream && options.onChunk),
+        messageCount: messages.length,
+        toolCount: _nullishCoalesce(_optionalChain([tools, 'optionalAccess', _ => _.length]), () => ( 0))
+      });
+      try {
+        const response = options.stream && options.onChunk ? await this.streamMessage(options, messages, tools) : await this.nonStreamMessage(options, messages, tools);
+        span.end({
+          stopReason: response.stopReason,
+          promptTokens: _optionalChain([response, 'access', _2 => _2.usage, 'optionalAccess', _3 => _3.promptTokens]),
+          completionTokens: _optionalChain([response, 'access', _4 => _4.usage, 'optionalAccess', _5 => _5.completionTokens]),
+          cacheReadTokens: _optionalChain([response, 'access', _6 => _6.usage, 'optionalAccess', _7 => _7.cacheReadTokens]),
+          cacheCreationTokens: _optionalChain([response, 'access', _8 => _8.usage, 'optionalAccess', _9 => _9.cacheCreationTokens])
+        });
+        return response;
+      } catch (err) {
+        span.fail(err);
+        throw err;
       }
-      return this.nonStreamMessage(options, messages, tools);
     });
   }
   async nonStreamMessage(options, messages, tools) {
@@ -54,7 +75,7 @@ var AnthropicClient = (_class = class extends _chunkDICG4AB7cjs.LLMClientBase {
             }
           ],
           messages,
-          ..._optionalChain([tools, 'optionalAccess', _ => _.length]) ? { tools } : {},
+          ..._optionalChain([tools, 'optionalAccess', _10 => _10.length]) ? { tools } : {},
           ...options.temperature !== void 0 ? { temperature: options.temperature } : { temperature: this.temperature }
         },
         { signal: options.signal }
@@ -87,7 +108,7 @@ var AnthropicClient = (_class = class extends _chunkDICG4AB7cjs.LLMClientBase {
             }
           ],
           messages,
-          ..._optionalChain([tools, 'optionalAccess', _2 => _2.length]) ? { tools } : {},
+          ..._optionalChain([tools, 'optionalAccess', _11 => _11.length]) ? { tools } : {},
           ...options.temperature !== void 0 ? { temperature: options.temperature } : { temperature: this.temperature }
         },
         { signal: options.signal }
@@ -96,16 +117,27 @@ var AnthropicClient = (_class = class extends _chunkDICG4AB7cjs.LLMClientBase {
       let currentToolName = "";
       let currentToolId = "";
       let currentToolInput = "";
+      const streamStartedAt = Date.now();
+      let firstByteLogged = false;
       stream.on("text", (text) => {
+        if (!firstByteLogged) {
+          firstByteLogged = true;
+          _chunkJHNCK7JPcjs.logger.debug("llm.first_byte", {
+            cat: "llm",
+            provider: this.provider,
+            model: this.model,
+            ttft_ms: Date.now() - streamStartedAt
+          });
+        }
         currentText += text;
-        _optionalChain([options, 'access', _3 => _3.onChunk, 'optionalCall', _4 => _4({ type: "text", text })]);
+        _optionalChain([options, 'access', _12 => _12.onChunk, 'optionalCall', _13 => _13({ type: "text", text })]);
       });
       stream.on("contentBlock", (block) => {
         if (block.type === "tool_use") {
           currentToolName = block.name;
           currentToolId = block.id;
           currentToolInput = "";
-          _optionalChain([options, 'access', _5 => _5.onChunk, 'optionalCall', _6 => _6({
+          _optionalChain([options, 'access', _14 => _14.onChunk, 'optionalCall', _15 => _15({
             type: "tool_use_start",
             toolCall: { id: block.id, toolName: block.name, args: {} }
           })]);
@@ -113,6 +145,16 @@ var AnthropicClient = (_class = class extends _chunkDICG4AB7cjs.LLMClientBase {
       });
       stream.on("inputJson", (_delta, snapshot) => {
         currentToolInput = JSON.stringify(snapshot);
+        if (currentToolId) {
+          _optionalChain([options, 'access', _16 => _16.onChunk, 'optionalCall', _17 => _17({
+            type: "tool_use_delta",
+            toolCall: {
+              id: currentToolId,
+              toolName: currentToolName,
+              args: _nullishCoalesce(snapshot, () => ( {}))
+            }
+          })]);
+        }
       });
       const finalMessage = await stream.finalMessage();
       const usage = {
@@ -123,7 +165,7 @@ var AnthropicClient = (_class = class extends _chunkDICG4AB7cjs.LLMClientBase {
         cacheCreationTokens: finalMessage.usage.cache_creation_input_tokens
       };
       this.recordUsage(usage);
-      _optionalChain([options, 'access', _7 => _7.onChunk, 'optionalCall', _8 => _8({ type: "stop", stopReason: _nullishCoalesce(finalMessage.stop_reason, () => ( void 0)) })]);
+      _optionalChain([options, 'access', _18 => _18.onChunk, 'optionalCall', _19 => _19({ type: "stop", stopReason: _nullishCoalesce(finalMessage.stop_reason, () => ( void 0)) })]);
       return this.processResponse(finalMessage, usage);
     } catch (err) {
       this.handleApiError(err);
@@ -206,7 +248,7 @@ var AnthropicClient = (_class = class extends _chunkDICG4AB7cjs.LLMClientBase {
       if (err.status === 429) {
         throw new (0, _chunkEL2RL5DGcjs.LLMRateLimitError)("anthropic");
       }
-      if (_optionalChain([err, 'access', _9 => _9.message, 'optionalAccess', _10 => _10.includes, 'call', _11 => _11("prompt is too long")]) || _optionalChain([err, 'access', _12 => _12.message, 'optionalAccess', _13 => _13.includes, 'call', _14 => _14("context_length_exceeded")])) {
+      if (_optionalChain([err, 'access', _20 => _20.message, 'optionalAccess', _21 => _21.includes, 'call', _22 => _22("prompt is too long")]) || _optionalChain([err, 'access', _23 => _23.message, 'optionalAccess', _24 => _24.includes, 'call', _25 => _25("context_length_exceeded")])) {
         throw new (0, _chunkEL2RL5DGcjs.ContextLimitError)("anthropic");
       }
       throw new (0, _chunkEL2RL5DGcjs.LLMError)(`Anthropic API error: ${err.message}`, "anthropic", {

package/dist/{chunk-BMH65OVB.cjs → chunk-C7C7ZZY4.cjs}

@@ -1,6 +1,6 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;
 
-var _chunkIMMTBZ33cjs = require('./chunk-IMMTBZ33.cjs');
+var _chunkJHNCK7JPcjs = require('./chunk-JHNCK7JP.cjs');
 
 
 var _chunkIJSHQGMPcjs = require('./chunk-IJSHQGMP.cjs');
@@ -35,7 +35,7 @@ var MCPManager = (_class = class _MCPManager {
     for (let i = 0; i < results.length; i++) {
       const result = results[i];
       if (result.status === "rejected") {
-        _chunkIMMTBZ33cjs.logger.warn("mcp.connect_failed", {
+        _chunkJHNCK7JPcjs.logger.warn("mcp.connect_failed", {
           server: entries[i][0],
           error: result.reason.message
         });
@@ -47,10 +47,10 @@ var MCPManager = (_class = class _MCPManager {
    */
   async connect(name, config) {
     if (this.connections.has(name)) {
-      _chunkIMMTBZ33cjs.logger.info("mcp.already_connected", { server: name });
+      _chunkJHNCK7JPcjs.logger.info("mcp.already_connected", { server: name });
       return;
     }
-    _chunkIMMTBZ33cjs.logger.info("mcp.connecting", { server: name, transport: _nullishCoalesce(config.transport, () => ( "stdio")) });
+    _chunkJHNCK7JPcjs.logger.info("mcp.connecting", { server: name, transport: _nullishCoalesce(config.transport, () => ( "stdio")) });
     const client = new (0, _indexjs.Client)(
       { name: "code-shell", version: "0.1.0" },
       { capabilities: {} }
@@ -99,7 +99,7 @@ var MCPManager = (_class = class _MCPManager {
     }
     this.connections.set(name, { client, serverName: name, transport });
     await this.discoverTools(name, client);
-    _chunkIMMTBZ33cjs.logger.info("mcp.connected", { server: name });
+    _chunkJHNCK7JPcjs.logger.info("mcp.connected", { server: name });
   }
   /**
    * Discover tools from an MCP server and register them.
@@ -131,7 +131,7 @@ var MCPManager = (_class = class _MCPManager {
         }
         return parts.join("\n") || "(no output)";
       });
-      _chunkIMMTBZ33cjs.logger.info("mcp.tool_registered", { server: serverName, tool: registered.name });
+      _chunkJHNCK7JPcjs.logger.info("mcp.tool_registered", { server: serverName, tool: registered.name });
     }
   }
   /**
@@ -141,9 +141,9 @@ var MCPManager = (_class = class _MCPManager {
     for (const [name, conn] of this.connections) {
       try {
         await conn.client.close();
-        _chunkIMMTBZ33cjs.logger.info("mcp.disconnected", { server: name });
+        _chunkJHNCK7JPcjs.logger.info("mcp.disconnected", { server: name });
       } catch (err) {
-        _chunkIMMTBZ33cjs.logger.warn("mcp.disconnect_error", { server: name, error: err.message });
+        _chunkJHNCK7JPcjs.logger.warn("mcp.disconnect_error", { server: name, error: err.message });
       }
     }
     this.connections.clear();

package/dist/{chunk-KO2FPF7G.js → chunk-F5BS3GUC.js}

@@ -4,6 +4,17 @@ import {
 
 // src/llm/model-pool.ts
 init_esm_shims();
+var BUILTIN_CONTEXT_WINDOWS = [
+  // DeepSeek V4 family — 1M context (api-docs.deepseek.com)
+  [/^deepseek-v4(?:-|$)/i, 1e6],
+  [/(?:^|\/)deepseek-v4(?:-|$)/i, 1e6]
+];
+function lookupBuiltinContextWindow(model) {
+  for (const [re, tokens] of BUILTIN_CONTEXT_WINDOWS) {
+    if (re.test(model)) return tokens;
+  }
+  return void 0;
+}
 var ModelPool = class {
   models = /* @__PURE__ */ new Map();
   activeKey;
@@ -14,7 +25,7 @@ var ModelPool = class {
   constructor(entries, defaultKey) {
     if (entries) {
       for (const e of entries) {
-        this.models.set(e.key, e);
+        this.models.set(e.key, this.withBuiltinDefaults(e));
       }
     }
     if (defaultKey && this.models.has(defaultKey)) {
@@ -25,11 +36,18 @@ var ModelPool = class {
   }
   /** Register a model at runtime. */
   register(entry) {
-    this.models.set(entry.key, entry);
+    this.models.set(entry.key, this.withBuiltinDefaults(entry));
     if (!this.activeKey) {
       this.activeKey = entry.key;
     }
   }
+  /** Fill in known defaults (e.g. context window for DeepSeek V4) when the entry doesn't specify them. */
+  withBuiltinDefaults(entry) {
+    if (entry.maxContextTokens != null) return entry;
+    const builtin = lookupBuiltinContextWindow(entry.model);
+    if (builtin == null) return entry;
+    return { ...entry, maxContextTokens: builtin };
+  }
   /** Switch the active model. Throws if key not found. */
   switch(key) {
     const entry = this.models.get(key);