rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-01-09 05:12:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: handle camelCase column names in select (#1460)

Browse Source 
Fixes #1385
This commit is contained in:
Will Jones
2024-07-22 12:53:17 -07:00
committed by GitHub
parent 391fa26175
commit 4f601a2d4c
6 changed files with 73 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
nodejs/lancedb/query.ts
View File

@@ -167,20 +167,27 @@ export class QueryBase<NativeQueryType extends NativeQuery | NativeVectorQuery>
select(
columns: string[] | Map<string, string> | Record<string, string> | string,
): this {
let columnTuples: [string, string][];
const selectColumns = (columnArray: string[]) => {
this.doCall((inner: NativeQueryType) => {
inner.selectColumns(columnArray);
});
};
const selectMapping = (columnTuples: [string, string][]) => {
this.doCall((inner: NativeQueryType) => {
inner.select(columnTuples);
});
};
if (typeof columns === "string") {
columns = [columns];
}
if (Array.isArray(columns)) {
columnTuples = columns.map((c) => [c, c]);
selectColumns([columns]);
} else if (Array.isArray(columns)) {
selectColumns(columns);
} else if (columns instanceof Map) {
columnTuples = Array.from(columns.entries());
selectMapping(Array.from(columns.entries()));
} else {
columnTuples = Object.entries(columns);
selectMapping(Object.entries(columns));
}
this.doCall((inner: NativeQueryType) => {
inner.select(columnTuples);
});
return this;
}