david-oplatka commited on
Commit
49799ed
1 Parent(s): e132805

Fix cfg spacing

Browse files
Files changed (1) hide show
  1. test_agent.py +2 -3
test_agent.py CHANGED
@@ -16,9 +16,8 @@ class TestAgentResponses(unittest.TestCase):
16
 
17
  cfg = OmegaConf.create({
18
  'customer_id': str(os.environ['VECTARA_CUSTOMER_ID']),
19
- 'corpus_ids': str(os.environ['VECTARA_CORPUS_IDS']).split(','),
20
- 'api_keys': str(os.environ['VECTARA_API_KEYS']).split(','),
21
- 'examples': os.environ.get('QUERY_EXAMPLES', None)
22
  })
23
 
24
  setup_db()
 
16
 
17
  cfg = OmegaConf.create({
18
  'customer_id': str(os.environ['VECTARA_CUSTOMER_ID']),
19
+ 'corpus_ids': str(os.environ['VECTARA_CORPUS_IDS']).split(','),
20
+ 'api_keys': str(os.environ['VECTARA_API_KEYS']).split(',')
 
21
  })
22
 
23
  setup_db()