gamingflexer
Add config file with dense models
50c110f
raw
history blame
407 Bytes
from decouple import config
import os
OPENAI_API_KEY = config('OPENAI_API_KEY', default="")
dense_models = {
'text-embedding-ada-002': {
'source': 'openai',
'dimension': 1536,
'api_key': True,
'metric': 'dotproduct'
},
'multilingual-22-12': {
'source': 'cohere',
'dimension': 768,
'api_key': True,
'metric': 'dotproduct'
}
}