Skip to content

Commit f4ec919

Browse files
committed
fix duplicate tables in listTablesContextProvider
1 parent 825288a commit f4ec919

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

src/aiProviders/continue/listTablesContextProvider.ts

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ import {
1414
getSqlContextItems
1515
} from "../context";
1616
import Configuration from "../../configuration";
17+
import { getContextItems } from "../prompt";
1718

1819
const listDb2Table: ContextProviderDescription = {
1920
title: "list Db2i Tables",
@@ -28,6 +29,7 @@ interface SchemaContextProvider {
2829
}
2930

3031
let providers: SchemaContextProvider[] = []
32+
let refCache: Set<string> = new Set<string>();
3133

3234
class ListDb2iTables implements IContextProvider {
3335
constructor(private schema: string) {
@@ -82,9 +84,13 @@ class ListDb2iTables implements IContextProvider {
8284
}
8385

8486
} else {
85-
const tablesRefs = await getSqlContextItems(extras.fullInput);
86-
for (const table of tablesRefs.items) {
87-
contextItems.push(table);
87+
const tablesRefs = await getContextItems(extras.fullInput);
88+
for (const tableData of tablesRefs.context) {
89+
if (refCache.has(tableData.name)) {
90+
continue;
91+
}
92+
contextItems.push(tableData);
93+
refCache.add(tableData.name);
8894
}
8995
}
9096
return contextItems;

0 commit comments

Comments
 (0)