Create config.py

config.py

@@ -0,0 +1,11 @@
+import os
+from omegaconf import OmegaConf
+from dotenv import load_dotenv
+load_dotenv(override=True)
+
+def get_agent_config() -> OmegaConf:
+    cfg = OmegaConf.create({
+        'corpus_key': str(os.environ['VECTARA_CORPUS_KEY']),
+        'api_key': str(os.environ['VECTARA_API_KEY'])
+    })
+    return cfg