feat: make maxDepth a config param

ShivTestOrg · Jan 6, 2025 · 6a01303 · 6a01303
1 parent 6e1b819
commit 6a01303
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/handlers/ask-llm.ts b/src/handlers/ask-llm.ts
@@ -9,13 +9,16 @@ import { findGroundTruths } from "./ground-truths/find-ground-truths";
 import { bubbleUpErrorComment, logger } from "../helpers/errors";
 
 export async function askQuestion(context: Context, question: string) {
+  const {
+    config: { maxDepth },
+  } = context;
   if (!question) {
     throw logger.error("No question provided");
   }
   // using any links in comments or issue/pr bodies to fetch more context
   const { specAndBodies, streamlinedComments } = await recursivelyFetchLinkedIssues({
     context,
-    maxDepth: 2,
+    maxDepth,
     owner: context.payload.repository.owner.login,
     repo: context.payload.repository.name,
     issueNum: context.payload.issue.number,

diff --git a/src/types/plugin-input.ts b/src/types/plugin-input.ts
@@ -12,6 +12,7 @@ export const pluginSettingsSchema = T.Object({
   model: T.String({ default: "o1-mini" }),
   openAiBaseUrl: T.Optional(T.String()),
   similarityThreshold: T.Number({ default: 0.9 }),
+  maxDepth: T.Number({ default: 3 }),
   maxTokens: T.Number({ default: 10000 }),
 });