gamingflexer commited on
Commit
50c110f
1 Parent(s): 72837e0

Add config file with dense models

Browse files
Files changed (1) hide show
  1. src/config.py +19 -0
src/config.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from decouple import config
2
+ import os
3
+
4
+ OPENAI_API_KEY = config('OPENAI_API_KEY', default="")
5
+
6
+ dense_models = {
7
+ 'text-embedding-ada-002': {
8
+ 'source': 'openai',
9
+ 'dimension': 1536,
10
+ 'api_key': True,
11
+ 'metric': 'dotproduct'
12
+ },
13
+ 'multilingual-22-12': {
14
+ 'source': 'cohere',
15
+ 'dimension': 768,
16
+ 'api_key': True,
17
+ 'metric': 'dotproduct'
18
+ }
19
+ }