genai comments

packages/core/src/cache.ts

@@ -27,19 +27,17 @@
         super() // Initialize EventTarget
     }
 
-    /**
-     * Factory method to create or retrieve an existing cache by name.
-     * Sanitizes the name to ensure it is a valid identifier.
-     * @param name - The name of the cache
-     * @returns An instance of JSONLineCache
-     */
-    static byName<K, V>(name: string): JSONLineCache<K, V> {
-        name = name.replace(/[^a-z0-9_]/gi, "_") // Sanitize name
+    static byName<K, V>(
+        name: string,
+        options?: { snapshot?: boolean }
+    ): JSONLineCache<K, V> {
+        const { snapshot } = options || {}
+        name = name.replace(/[^a-z0-9_]/gi, "_")
         const key = "cacheKV." + name
-        if (host.userState[key]) return host.userState[key] // Return if exists
+        if (!snapshot && host.userState[key]) return host.userState[key]
         const r = new JSONLineCache<K, V>(name)
-        host.userState[key] = r
+        if (!snapshot) host.userState[key] = r
         return r
     }
 
     // Get the folder path for the cache storage

packages/core/src/comments.ts

@@ -0,0 +1,22 @@
+import { JSONLineCache } from "./cache"
+import { COMMENTS_CACHE } from "./constants"
+
+export interface CommentKey {
+    uri: string
+    line: number
+    source: string
+}
+
+export function commentsCache(options?: { snapshot?: boolean }) {
+    const cache = JSONLineCache.byName<CommentKey, CommentThread>(
+        COMMENTS_CACHE,
+        options
+    )
+    return cache
+}
+
+export async function commentsForSource(source: string) {
+    const cache = commentsCache()
+    const entries = await cache.entries()
+    return entries.map(({ val }) => val).filter((c) => c.source === source)
+}

packages/core/src/constants.ts

@@ -216,6 +216,8 @@ export const GITHUB_TOKEN = "GITHUB_TOKEN"
 export const AI_REQUESTS_CACHE = "airequests"
 export const CHAT_CACHE = "chat"
 export const GITHUB_PULL_REQUEST_REVIEWS_CACHE = "prr"
+export const GITHUB_PULLREQUEST_REVIEW_COMMENT_LINE_DISTANCE = 5
+export const COMMENTS_CACHE = "comments"
 export const GITHUB_PULL_REQUEST_REVIEW_COMMENT_LINE_DISTANCE = 5
 
 export const PLACEHOLDER_API_BASE = "<custom api base>"

packages/core/src/promptrunner.ts

@@ -22,6 +22,7 @@ import { validateJSONWithSchema } from "./schema"
 import { YAMLParse } from "./yaml"
 import { expandTemplate } from "./expander"
 import { resolveLanguageModel } from "./lm"
+import { commentsCache, commentsForSource } from "./comments"
 
 async function resolveExpansionVars(
     project: Project,
@@ -57,6 +58,7 @@ async function resolveExpansionVars(
             secrets[secret] = value
         } else trace.error(`secret \`${secret}\` not found`)
     }
+    const comments = await commentsForSource(template.id)
     const res: Partial<ExpansionVariables> = {
         dir: ".",
         files,
@@ -67,6 +69,7 @@ async function resolveExpansionVars(
         },
         vars: attrs,
         secrets,
+        comments,
     }
     return res
 }