| 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 |