add unique filter

xeol-io · May 6, 2024 · ef715e3 · ef715e3
1 parent b2ee156
commit ef715e3
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/packages/bumpgen-core/src/services/llm/openai.ts b/packages/bumpgen-core/src/services/llm/openai.ts
@@ -1,4 +1,5 @@
 import type { OpenAI } from "openai";
+import { unique } from "radash";
 
 import type { DependencyGraphNode } from "../../models/graph/dependency";
 import type { PlanGraphNode } from "../../models/graph/plan";
@@ -18,9 +19,13 @@ const makePlanNodeMessage = (
   importContext: DependencyGraphNode[],
   bumpedPackage: string,
 ) => {
-  const importMessages = importContext.map((context) => {
-    return context.block;
-  });
+  // We need to do this because we only store nodes for the specific identifiers in an import, so
+  // multiple can be in the same code block.
+  const importMessages = unique(
+    importContext.map((context) => {
+      return context.block;
+    }),
+  );
   return {
     role: "user" as const,
     content: [