File tree Expand file tree Collapse file tree 1 file changed +9
-3
lines changed Expand file tree Collapse file tree 1 file changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -14,6 +14,7 @@ import {
14
14
getSqlContextItems
15
15
} from "../context" ;
16
16
import Configuration from "../../configuration" ;
17
+ import { getContextItems } from "../prompt" ;
17
18
18
19
const listDb2Table : ContextProviderDescription = {
19
20
title : "list Db2i Tables" ,
@@ -28,6 +29,7 @@ interface SchemaContextProvider {
28
29
}
29
30
30
31
let providers : SchemaContextProvider [ ] = [ ]
32
+ let refCache : Set < string > = new Set < string > ( ) ;
31
33
32
34
class ListDb2iTables implements IContextProvider {
33
35
constructor ( private schema : string ) {
@@ -82,9 +84,13 @@ class ListDb2iTables implements IContextProvider {
82
84
}
83
85
84
86
} 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 ) ;
88
94
}
89
95
}
90
96
return contextItems ;
You can’t perform that action at this time.
0 commit comments