dsh-custom-api 1.0.2 → 1.0.3

package/lib/index.js

@@ -82,8 +82,9 @@ const GEMINI_LEVELS = Object.freeze({
 });
 
 function geminiEfforts(id) {
-  const norm = String(id).toLowerCase();
-  const documented = GEMINI_LEVELS[norm] || ['low', 'medium', 'high'];
+  const norm = normalizeModelId(id);
+  const tail = String(id).toLowerCase().split('/').pop() || '';
+  const documented = GEMINI_LEVELS[tail] || GEMINI_LEVELS[norm] || ['low', 'medium', 'high'];
   const efforts = { off: null };
   for (const level of documented) efforts[level] = level;
   if (!efforts.medium) efforts.medium = efforts.high || 'medium';
@@ -146,6 +147,16 @@ function buildModelsDevIndex(data) {
   isLoaded = true;
 }
 
+// Synchronously load cache on module import if exists
+try {
+  if (fs.existsSync(CACHE_PATH)) {
+    const content = fs.readFileSync(CACHE_PATH, 'utf8');
+    buildModelsDevIndex(JSON.parse(content));
+  }
+} catch {
+  // Ignore sync read error, will be retried in initModelsDevDatabase
+}
+
 async function initModelsDevDatabase(logger) {
   try {
     if (fs.existsSync(CACHE_PATH)) {
@@ -329,11 +340,22 @@ function determineReasoningEfforts(id, model, dev, forceThinking) {
   let isThinking = forceThinking;
   if (!isThinking) {
     if (typeof model?.supports_thinking === 'boolean') isThinking = model.supports_thinking;
+    else if (model?.reasoningEfforts && typeof model.reasoningEfforts === 'object') isThinking = true;
     else if (dev?.reasoning === true) isThinking = true;
-    else if (THINKING_MODEL_ID.test(String(id))) isThinking = true;
     else {
-      const desc = typeof model?.description === 'string' ? model.description : '';
-      isThinking = THINKING_DESCRIPTION.test(desc);
+      const s = String(id).toLowerCase();
+      isThinking = (
+        s.includes('r1') || s.includes('o1') || s.includes('o3') || s.includes('o4') ||
+        s.includes('reason') || s.includes('think') || s.includes('thought') ||
+        s.includes('qwq') || s.includes('qvq') ||
+        s.includes('gemini-2.5') || s.includes('gemini-3') || s.includes('gemini-4') ||
+        (s.includes('claude') && (s.includes('3-7') || s.includes('3.7') || s.includes('sonnet') || s.includes('opus') || s.includes('4') || s.includes('5'))) ||
+        s.includes('gpt-5') || s.includes('gpt-6') ||
+        s.includes('grok-4') || s.includes('grok-3') ||
+        s.includes('glm-5') || s.includes('deepseek-v4') ||
+        s.includes('qwen-3') || s.includes('qwen3') ||
+        THINKING_DESCRIPTION.test(typeof model?.description === 'string' ? model.description : '')
+      );
     }
   }
 
@@ -389,7 +411,7 @@ function isImageGenerationEndpoint(id, dev) {
 /**
  * Giải quyết và gán năng lực hoàn chỉnh cho một model
  */
-function resolveCapabilities(rawModel, opts = {}) {
+export function resolveCapabilities(rawModel, opts = {}) {
   const id = typeof rawModel === 'string' ? rawModel : rawModel.id;
   const dev = lookupModelsDev(id);
   const apiMeta = extractFromApiMetadata(rawModel);
@@ -976,7 +998,7 @@ export function apply(ctx) {
                 maxTokens: Math.min(m.maxTokens || cap.modelEntry.maxTokens, m.contextWindow || cap.modelEntry.contextWindow),
               };
 
-              if (cap.hasThinking && (!merged.reasoningEfforts || Object.keys(merged.reasoningEfforts).length <= 2)) {
+              if (cap.hasThinking) {
                 merged.reasoningEfforts = cap.modelEntry.reasoningEfforts;
                 if (p.api === 'openai-completions') {
                   const s = (m.id || '').toLowerCase();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-custom-api",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "DeepSeek Harness plugin: Auto-discovery, Vision, Reasoning/Thinking Effort & Tool Calls for Custom Model APIs",
   "type": "module",
   "main": "lib/index.js",