feat(vscode): add language model tools

packages/vscode/package.json

@@ -434,6 +434,51 @@
         "view": "slidev-slides-tree",
         "contents": "No active slides entry.\n[Choose one](command:slidev.choose-entry)"
       }
+    ],
+    "languageModelTools": [
+      {
+        "name": "slidev_getActiveSlide",
+        "tags": [
+          "slidev"
+        ],
+        "toolReferenceName": "getActiveSlide",
+        "displayName": "Get Active Slide",
+        "modelDescription": "Get the information of the active slide the user is currently focused on in a Slidev presentation.",
+        "userDescription": "Get the information of the active slide in a Slidev presentation.",
+        "canBeReferencedInPrompt": true,
+        "icon": "$(debug-stackframe-active)",
+        "inputSchema": {
+          "type": "object",
+          "properties": {
+          }
+        }
+      },
+      {
+        "name": "slidev_getSlideContent",
+        "tags": [
+          "slidev"
+        ],
+        "toolReferenceName": "getSlideContent",
+        "displayName": "Get Slide Content",
+        "modelDescription": "Get the content of a specific slide in a Slidev presentation by providing the slide number.",
+        "userDescription": "Get the content of a specific slide in a Slidev presentation by providing the slide number.",
+        "canBeReferencedInPrompt": true,
+        "icon": "$(file-code)",
+        "inputSchema": {
+          "type": "object",
+          "properties": {
+            "entrySlidePath": {
+              "type": "string",
+              "description": "The path to the Slidev entry file (e.g., `./slides.md`). Empty string means the active slide entry.",
+              "default": "$ACTIVE_SLIDE_ENTRY"
+            },
+            "slideNo": {
+              "type": "number",
+              "description": "The slide number to retrieve content from. Starts from 1. Hidden slides are not counted."
+            }
+          }
+        }
+      }
     ]
   },
   "scripts": {

packages/vscode/src/index.ts

@@ -1,6 +1,7 @@
 import { defineExtension } from 'reactive-vscode'
 import { useCommands } from './commands'
 import { useLanguageClient } from './languageClient'
+import { useLmTools } from './lmTools'
 import { activeEntry, useProjects } from './projects'
 import { useAnnotations } from './views/annotations'
 import { useFoldings } from './views/foldings'
@@ -26,6 +27,9 @@ const { activate, deactivate } = defineExtension(() => {
   // language server
   const labsInfo = useLanguageClient()
 
+  // language model tools
+  useLmTools()
+
   logger.info('Slidev activated.')
   logger.info(`Entry: ${activeEntry.value}`)
 

packages/vscode/src/lmTools.ts

@@ -0,0 +1,102 @@
+import { slash } from '@antfu/utils'
+import { stringifySlide } from '@slidev/parser/core'
+import { createSingletonComposable, useDisposable } from 'reactive-vscode'
+import { LanguageModelTextPart, LanguageModelToolResult, lm } from 'vscode'
+import { useEditingSlideSource } from './composables/useEditingSlideSource'
+import { useFocusedSlideNo } from './composables/useFocusedSlideNo'
+import { activeEntry, activeProject, projects } from './projects'
+
+export const useLmTools = createSingletonComposable(() => {
+  const focusedSlideNo = useFocusedSlideNo()
+  const editingSlide = useEditingSlideSource()
+
+  registerSimpleTool('slidev_getActiveSlide', () => {
+    const project = activeProject.value
+
+    if (project == null) {
+      throw new Error(`No active slide project found.`)
+    }
+
+    return formatObject({
+      'Entry file': project.entry,
+      'Root directory': project.userRoot,
+      'Preview server port': project.port || 'Not running',
+      'Number of slides': project.data.slides.length,
+      'Focused slide no. in presentation (from 1)': focusedSlideNo.value,
+      'Editing file': editingSlide.markdown.value?.filepath || 'Not editing',
+      'Editing slide index in file (from 0)': editingSlide.index.value,
+    })
+  })
+
+  registerSimpleTool('slidev_getSlideContent', (input: {
+    entrySlidePath: string
+    slideNo: number
+  }) => {
+    const project = resolveProjectFromEntry(input.entrySlidePath)
+    const slide = project.data.slides[input.slideNo - 1]
+
+    if (slide == null) {
+      throw new Error(`No content found for slide number ${input.slideNo} in entry: ${project.entry}. Available slides numbers: 1-${project.data.slides.length}`)
+    }
+
+    return `Content of slide number ${input.slideNo} in entry "${project.entry}" in file "${slide.source.filepath}":\n\n${stringifySlide(slide.source, 1)}`
+  })
+})
+
+function registerSimpleTool<T>(name: string, invoke: (input: T) => string) {
+  useDisposable(lm.registerTool<T>(name, {
+    invoke({ input }) {
+      try {
+        const result = invoke(input)
+        return new LanguageModelToolResult([
+          new LanguageModelTextPart(result),
+        ])
+      }
+      catch (error: any) {
+        return new LanguageModelToolResult([
+          new LanguageModelTextPart(`Error: ${error.message || error.toString()}`),
+        ])
+      }
+    },
+  }))
+}
+
+function resolveProjectFromEntry(entry: string) {
+  if (entry === '' || entry === '$ACTIVE_SLIDE_ENTRY') {
+    if (!activeEntry.value) {
+      throw new Error('No active slide entry found. Please set an active slide entry before using this tool.')
+    }
+    entry = activeEntry.value
+  }
+
+  let project = projects.get(entry)
+  if (!project) {
+    entry = slash(entry)
+    const possibleProjects = [...projects.values()].filter(p => p.entry.includes(entry))
+    if (possibleProjects.length === 0) {
+      throw new Error(`No project found for entry: ${entry}. All entries: ${formatList(projects.keys())}`)
+    }
+    else if (possibleProjects.length > 1) {
+      throw new Error(`Multiple projects found for entry: ${entry}. Please specify the full path. All entries: ${formatList(projects.keys())}`)
+    }
+    else {
+      project = possibleProjects[0]
+    }
+  }
+
+  return project
+}
+
+function formatList(items: Iterable<string>): string {
+  const itemsArray = [...items]
+  if (itemsArray.length === 0) {
+    return 'No items found.'
+  }
+  return itemsArray.map(item => `- ${item}\n`).join('')
+}
+
+function formatObject(obj: Record<string, string | number>): string {
+  return Object.entries(obj)
+    .map(([key, value]) => `- ${key}: ${value}\n`)
+    .join('')
+}