Codebase improvements

core/config/yaml/models.ts

@@ -49,6 +49,10 @@ async function modelConfigToBaseLLM(
     title: model.name,
     systemMessage,
     promptTemplates: model.promptTemplates,
+    capabilities: {
+      tools: model.capabilities?.includes("tool_use"),
+      uploadImage: model.capabilities?.includes("image_input"),
+    },
   };
 
   if (model.embedOptions?.maxBatchSize) {

core/context/retrieval/repoMapRequest.ts

@@ -2,6 +2,7 @@ import { Chunk, ContinueConfig, IDE, ILLM } from "../..";
 import { getModelByRole } from "../../config/util";
 import generateRepoMap from "../../util/generateRepoMap";
 import { renderChatMessage } from "../../util/messageContent";
+import { localPathToUri } from "../../util/pathToUri";
 
 const SUPPORTED_MODEL_TITLE_FAMILIES = [
   "claude-3",
@@ -72,16 +73,17 @@ This is the question that you should select relevant files for: "${input}"`;
       return [];
     }
 
-    const fileUris = content
+    const filepaths = content
       .split("<results>")[1]
       ?.split("</results>")[0]
       ?.split("\n")
       .filter(Boolean)
       .map((uri) => uri.trim());
 
     const chunks = await Promise.all(
-      fileUris.map(async (uri) => {
-        const content = await ide.readFile(uri);
+      filepaths.map(async (filepath) => {
+        const uri = localPathToUri(filepath);
+        const content = await ide.readFile(filepath);
         const lineCount = content.split("\n").length;
         const chunk: Chunk = {
           digest: uri,

core/indexing/LanceDbIndex.ts

@@ -24,7 +24,6 @@ import {
 } from "./types";
 
 import type * as LanceType from "vectordb";
-import { localPathToUri } from "../util/pathToUri";
 
 interface LanceDbRow {
   uuid: string;
@@ -499,7 +498,7 @@ export class LanceDbIndex implements CodebaseIndex {
     return data.map((d) => {
       return {
         digest: d.cacheKey,
-        filepath: localPathToUri(d.path),
+        filepath: d.path,
         startLine: d.startLine,
         endLine: d.endLine,
         index: 0,

core/llm/llms/stubs/ContinueProxy.ts

@@ -75,6 +75,7 @@ class ContinueProxy extends OpenAI {
         query,
         documents: chunks.map((chunk) => chunk.content),
         model: this.model,
+        ...this.extraBodyProperties(),
       }),
     });
     const data: any = await resp.json();

core/package.json

@@ -46,7 +46,7 @@
     "@aws-sdk/client-sagemaker-runtime": "^3.621.0",
     "@aws-sdk/credential-providers": "^3.620.1",
     "@continuedev/config-types": "^1.0.13",
-    "@continuedev/config-yaml": "^1.0.52",
+    "@continuedev/config-yaml": "^1.0.56",
     "@continuedev/fetch": "^1.0.4",
     "@continuedev/llm-info": "^1.0.2",
     "@continuedev/openai-adapters": "^1.0.10",

core/util/pathToUri.ts

@@ -6,6 +6,10 @@ import * as URI from "uri-js";
 
 // Converts a local path to a file:/// URI
 export function localPathToUri(path: string) {
+  if (path.startsWith("file://")) {
+    console.warn("localPathToUri: path already starts with file://");
+    return path;
+  }
   const url = pathToFileURL(path);
   return URI.normalize(url.toString());
 }

extensions/vscode/package.json

@@ -2,7 +2,7 @@
   "name": "continue",
   "icon": "media/icon.png",
   "author": "Continue Dev, Inc",
-  "version": "1.1.7",
+  "version": "1.1.8",
   "repository": {
     "type": "git",
     "url": "https://github.com/continuedev/continue"