@movk/nuxt-docs 1.6.0 → 1.6.2

package/app/composables/useToolCall.ts

@@ -0,0 +1,9 @@
+export function useToolCall() {
+  const tools: Record<string, string | ((args: any) => string)> = {
+    'list-pages': '列出所有文档页面',
+    'get-page': (args: any) => `检索 ${args?.path || '页面'}`
+  }
+  return {
+    tools
+  }
+}

package/modules/ai-chat/index.ts

@@ -1,3 +1,5 @@
+import { existsSync } from 'node:fs'
+import { join } from 'node:path'
 import { addComponentsDir, addImportsDir, addServerHandler, createResolver, defineNuxtModule } from '@nuxt/kit'
 
 export interface AiChatModuleOptions {
@@ -65,11 +67,28 @@ export default defineNuxtModule<AiChatModuleOptions>({
 
     addImportsDir(resolve('runtime/composables'))
 
+    /**
+     * 检查用户项目中是否存在自定义 handler
+     * '/api/ai-chat' -> 'api/ai-chat'，'/custom' -> 'custom'
+     */
     const routePath = options.apiPath!.replace(/^\//, '')
-    addServerHandler({
-      route: `/${routePath}`,
-      handler: resolve('./runtime/server/api/search')
-    })
+    const extensions = ['ts', 'js', 'mjs']
+
+    const possibleHandlerPaths = extensions.flatMap(ext => [
+      join(nuxt.options.serverDir, `${routePath}.${ext}`),
+      join(nuxt.options.serverDir, routePath, `index.${ext}`)
+    ])
+
+    const hasCustomHandler = possibleHandlerPaths.some(path => existsSync(path))
+
+    if (!hasCustomHandler) {
+      addServerHandler({
+        route: options.apiPath!,
+        handler: resolve('./runtime/server/api/ai-chat')
+      })
+    } else {
+      console.info(`[ai-chat] Using custom handler, skipping default handler registration`)
+    }
   }
 })
 

package/modules/ai-chat/runtime/components/AiChatSlideover.vue

@@ -40,7 +40,17 @@ const components = {
 const { messages, isOpen, pendingMessage, clearPending } = useAIChat()
 const { apiPath } = useRuntimeConfig().public.aiChat
 
-const { getToolLabel } = useTools()
+const { tools } = useToolCall()
+function getToolLabel(toolName: string, args?: any): string {
+  const label = tools[toolName]
+
+  if (!label) {
+    return toolName
+  }
+
+  return typeof label === 'function' ? label(args) : label
+}
+
 const { model } = useModels()
 
 const input = ref('')

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@movk/nuxt-docs",
   "type": "module",
-  "version": "1.6.0",
+  "version": "1.6.2",
   "private": false,
   "description": "Modern Nuxt 4 documentation theme with auto-generated component docs, AI chat assistant, MCP server, and complete developer experience optimization.",
   "author": "YiXuan <mhaibaraai@gmail.com>",

package/modules/ai-chat/runtime/composables/useTools.ts

@@ -1,31 +0,0 @@
-export interface ToolLabelConfig {
-  toolName: string
-  label: string | ((args: any) => string)
-}
-
-export function useTools() {
-  const defaultLabels: Record<string, string | ((args: any) => string)> = {
-    'list-pages': '列出所有文档页面',
-    'get-page': (args: any) => `检索 ${args?.path || '页面'}`
-  }
-
-  /**
-   * 获取工具的显示标签
-   * @param toolName - 工具名称
-   * @param args - 工具参数
-   * @returns 工具的显示标签
-   */
-  function getToolLabel(toolName: string, args?: any): string {
-    const label = defaultLabels[toolName]
-
-    if (!label) {
-      return toolName
-    }
-
-    return typeof label === 'function' ? label(args) : label
-  }
-
-  return {
-    getToolLabel
-  }
-}