Skip to content

Commit eaff911

Browse files
authored
Merge pull request #759 from Portkey-AI/feat/qdrant
Add Qdrant as a provider to the gateway
2 parents 14f1011 + e941d41 commit eaff911

File tree

3 files changed

+29
-0
lines changed

3 files changed

+29
-0
lines changed

src/providers/index.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ import { UpstageConfig } from './upstage';
4747
import { LAMBDA } from '../globals';
4848
import { LambdaProviderConfig } from './lambda';
4949
import { DashScopeConfig } from './dashscope';
50+
import QdrantConfig from './qdrant';
5051

5152
const Providers: { [key: string]: ProviderConfigs } = {
5253
openai: OpenAIConfig,
@@ -94,6 +95,7 @@ const Providers: { [key: string]: ProviderConfigs } = {
9495
upstage: UpstageConfig,
9596
[LAMBDA]: LambdaProviderConfig,
9697
dashscope: DashScopeConfig,
98+
qdrant: QdrantConfig,
9799
};
98100

99101
export default Providers;

src/providers/qdrant/api.ts

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
import { ProviderAPIConfig } from '../types';
2+
3+
const QdrantAPIConfig: ProviderAPIConfig = {
4+
getBaseURL: ({ providerOptions }) => {
5+
return providerOptions.customHost || '';
6+
},
7+
headers: ({ providerOptions }) => {
8+
return { 'api-key': `Bearer ${providerOptions.apiKey}` };
9+
},
10+
getEndpoint: ({ fn }) => {
11+
switch (fn) {
12+
default:
13+
return '';
14+
}
15+
},
16+
};
17+
18+
export default QdrantAPIConfig;

src/providers/qdrant/index.ts

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
import { ProviderConfigs } from '../types';
2+
import QdrantAPIConfig from './api';
3+
4+
const QdrantConfig: ProviderConfigs = {
5+
api: QdrantAPIConfig,
6+
responseTransforms: {},
7+
};
8+
9+
export default QdrantConfig;

0 commit comments

Comments
 (0)