Add a source range prompt containing the current file content range (#8129)

* Add a source range prompt containing the current file content range

To help @greg-kcio with not needing to stuff everything into the LLM
context window.

* Fix lints

Author: franknoirot

src/components/ModelingSidebar/ModelingPanes/MlEphantConversationPane.tsx

@@ -76,7 +76,10 @@ export const MlEphantConversationPane = (props: {
       type: MlEphantManagerTransitions.PromptEditModel,
       prompt: requestedPrompt,
       projectForPromptOutput: props.theProject,
-      fileSelectedDuringPrompting: props.loaderFile,
+      fileSelectedDuringPrompting: {
+        entry: props.loaderFile,
+        content: props.codeManager.code,
+      },
       projectFiles,
       selections: props.contextModeling.selectionRanges,
       artifactGraph: props.kclManager.artifactGraph,

src/lib/promptToEdit.tsx

@@ -79,10 +79,14 @@ export function constructMultiFileIterationRequestWithPromptHelpers({
   projectFiles,
   artifactGraph,
   projectName,
+  currentFile,
   kclVersion,
 }: ConstructRequestArgs): PromptToEditRequest {
   const kclFilesMap: KclFileMetaMap = {}
   const files: KittyCadLibFile[] = []
+  const currentFileMeta = projectFiles.find(
+    (f) => f.type !== 'other' && f.absPath === currentFile.entry?.path
+  )
   projectFiles.forEach((file) => {
     let data: Blob
     if (file.type === 'other') {
@@ -98,12 +102,23 @@ export function constructMultiFileIterationRequestWithPromptHelpers({
     })
   })
 
+  // Way to patch in supplying the currently-opened file without updating the API.
+  // TODO: update the API to support currently-opened files as other parts of the payload
+  const currentFilePrompt: Models['SourceRangePrompt_type'] = {
+    prompt: 'This is the active file',
+    range: convertAppRangeToApiRange(
+      [0, currentFile.content.length, 0],
+      currentFile.content
+    ),
+    file: currentFileMeta?.relPath,
+  }
+
   // If no selection, use whole file
   if (selections === null) {
     return {
       body: {
         prompt,
-        source_ranges: [],
+        source_ranges: [currentFilePrompt],
         project_name:
           projectName !== '' && projectName !== 'browser'
             ? projectName
@@ -245,6 +260,8 @@ See later source ranges for more context. about the sweep`,
       }
       return prompts
     })
+  // Push the current file prompt alongside the selection-based prompts
+  ranges.push(currentFilePrompt)
   let payload = {
     body: {
       prompt,

src/lib/promptToEditTypes.ts

@@ -3,6 +3,7 @@ import type { ArtifactGraph } from '@src/lang/wasm'
 import type { Selections } from '@src/lib/selections'
 import type { FileMeta } from '@src/lib/types'
 import type { File as KittyCadLibFile } from '@kittycad/lib/dist/types/src/models'
+import type { FileEntry } from '@src/lib/project'
 
 export type KclFileMetaMap = {
   [execStateFileNamesIndex: number]: Extract<FileMeta, { type: 'kcl' }>
@@ -29,6 +30,7 @@ export interface ConstructRequestArgs {
   selections: Selections | null
   projectFiles: FileMeta[]
   projectName: string
+  currentFile: { entry?: FileEntry; content: string }
   artifactGraph: ArtifactGraph
   kclVersion: string
 }

src/machines/mlEphantManagerMachine.ts

@@ -81,7 +81,7 @@ export type MlEphantManagerEvents =
       type: MlEphantManagerTransitions.PromptEditModel
       projectForPromptOutput: Project
       prompt: string
-      fileSelectedDuringPrompting?: FileEntry
+      fileSelectedDuringPrompting: { entry?: FileEntry; content: string }
       projectFiles: FileMeta[]
       selections: Selections
       artifactGraph: ArtifactGraph
@@ -380,6 +380,7 @@ export const mlEphantManagerMachine = setup({
             projectFiles: event.projectFiles,
             artifactGraph: event.artifactGraph,
             projectName: event.projectForPromptOutput.name,
+            currentFile: event.fileSelectedDuringPrompting,
             kclVersion: getKclVersion(),
           }
         )
@@ -406,7 +407,7 @@ export const mlEphantManagerMachine = setup({
         promptsBelongingToConversation.push(result.id)
         promptsMeta.set(result.id, {
           type: PromptType.Edit,
-          targetFile: args.input.event.fileSelectedDuringPrompting,
+          targetFile: args.input.event.fileSelectedDuringPrompting.entry,
           project: args.input.event.projectForPromptOutput,
         })