Spaces:
Sleeping
Sleeping
david-oplatka
commited on
Commit
•
49799ed
1
Parent(s):
e132805
Fix cfg spacing
Browse files- 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 |
-
|
20 |
-
|
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()
|