UNPKG

@gentrace/pinecone

Version:

Gentrace Pinecone v1 plugin for Node.JS

github.com/gentrace/gentrace-node

gentrace/gentrace-node

132 lines (112 loc) • 3.42 kB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
import { PluginStepRunContext } from "@gentrace/core"; import { Index, QueryOptions, RecordMetadata, UpdateOptions, } from "@pinecone-database/pinecone"; import { FunctionWithPipelineRunId, GentraceParams, ModifiedNamespaceFunction, ModifyFirstParam, ModifySecondParam, PineconePipelineHandler, } from "../pinecone"; class AdvancedPinecone extends PineconePipelineHandler { // @ts-ignore public Index<T extends RecordMetadata = RecordMetadata>(index: string) { return this.index<T>(index); } // @ts-ignore: hack to avoid base class inheritance issues public index<T extends RecordMetadata = RecordMetadata>(index: string) { const apiHandler = super.indexInner<T>(index); type FetchFunctionType = typeof apiHandler.fetch; type ModifiedFetchFunction = FunctionWithPipelineRunId< ModifySecondParam< FetchFunctionType, Omit<GentraceParams, "gentrace"> & { gentrace?: PluginStepRunContext; } > >; type UpdateFunctionType = typeof apiHandler.update; type ModifiedUpdateFunction = FunctionWithPipelineRunId< ModifyFirstParam< UpdateFunctionType, UpdateOptions & Omit<GentraceParams, "gentrace"> & { gentrace?: PluginStepRunContext; } > >; type QueryFunctionType = typeof apiHandler.query; type ModifiedQueryFunction = FunctionWithPipelineRunId< ModifyFirstParam< QueryFunctionType, QueryOptions & Omit<GentraceParams, "gentrace"> & { gentrace?: PluginStepRunContext; } > >; type UpsertFunctionType = typeof apiHandler.upsert; type ModifiedUpsertFunction = FunctionWithPipelineRunId< ModifySecondParam< UpsertFunctionType, Omit<GentraceParams, "gentrace"> & { gentrace?: PluginStepRunContext; } > >; type DeleteOneFunctionType = typeof apiHandler.deleteOne; type ModifiedDeleteOneFunction = FunctionWithPipelineRunId< ModifySecondParam< DeleteOneFunctionType, Omit<GentraceParams, "gentrace"> & { gentrace?: PluginStepRunContext; } > >; type DeleteManyFunctionType = typeof apiHandler.deleteMany; type ModifiedDeleteManyFunction = FunctionWithPipelineRunId< ModifySecondParam< DeleteManyFunctionType, Omit<GentraceParams, "gentrace"> & { gentrace?: PluginStepRunContext; } > >; type DeleteAllFunctionType = typeof apiHandler.deleteAll; type ModifiedDeleteAllFunction = FunctionWithPipelineRunId< ModifyFirstParam< DeleteAllFunctionType, Omit<GentraceParams, "gentrace"> & { gentrace?: PluginStepRunContext; } > >; type ModifiedAdvancedIndex = Omit< Index, | "fetch" | "update" | "query" | "upsert" | "deleteOne" | "deleteMany" | "deleteAll" | "namespace" > & { fetch: ModifiedFetchFunction; update: ModifiedUpdateFunction; query: ModifiedQueryFunction; upsert: ModifiedUpsertFunction; deleteOne: ModifiedDeleteOneFunction; deleteMany: ModifiedDeleteManyFunction; deleteAll: ModifiedDeleteAllFunction; namespace: ModifiedNamespaceFunction; }; return apiHandler as ModifiedAdvancedIndex; } } export { AdvancedPinecone };