joao-vectara commited on
Commit
c414c6f
·
verified ·
1 Parent(s): 112d607

Update config.py

Browse files

Configuration for Vectara corpus

Files changed (1) hide show
  1. config.py +12 -0
config.py CHANGED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from omegaconf import OmegaConf
3
+ from dotenv import load_dotenv
4
+ load_dotenv(override=True)
5
+
6
+ def get_agent_config() -> OmegaConf:
7
+ cfg = OmegaConf.create({
8
+ 'customer_id': str(os.environ['VECTARA_CUSTOMER_ID']),
9
+ 'corpus_id': str(os.environ['VECTARA_CORPUS_ID']),
10
+ 'api_key': str(os.environ['VECTARA_API_KEY'])
11
+ })
12
+ return cfg