Skip to content

Commit c5a5eb2

Browse files
committed
Refactor listTablesContextProvider to remove unused reference cache and improve context item retrieval
1 parent f4ec919 commit c5a5eb2

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

src/aiProviders/continue/listTablesContextProvider.ts

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

1919
const listDb2Table: ContextProviderDescription = {
2020
title: "list Db2i Tables",
@@ -29,7 +29,6 @@ interface SchemaContextProvider {
2929
}
3030

3131
let providers: SchemaContextProvider[] = []
32-
let refCache: Set<string> = new Set<string>();
3332

3433
class ListDb2iTables implements IContextProvider {
3534
constructor(private schema: string) {
@@ -84,13 +83,9 @@ class ListDb2iTables implements IContextProvider {
8483
}
8584

8685
} else {
87-
const tablesRefs = await getContextItems(extras.fullInput);
86+
const tablesRefs = await getContextItems(query);
8887
for (const tableData of tablesRefs.context) {
89-
if (refCache.has(tableData.name)) {
90-
continue;
91-
}
9288
contextItems.push(tableData);
93-
refCache.add(tableData.name);
9489
}
9590
}
9691
return contextItems;

0 commit comments

Comments
 (0)