From b45f3af360ae4cf7c8040d1bff882fa669be092c Mon Sep 17 00:00:00 2001 From: Shivam Sareen Date: Fri, 6 Sep 2024 14:00:42 -0700 Subject: [PATCH] Update resume_yaml_generator.py Updated resume_yaml_generator.py with the correct api key name present in the secrets.yaml file --- resume_yaml_generator.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/resume_yaml_generator.py b/resume_yaml_generator.py index 46982c2..336a23d 100644 --- a/resume_yaml_generator.py +++ b/resume_yaml_generator.py @@ -21,9 +21,13 @@ def get_api_key() -> str: raise FileNotFoundError(f"Secrets file not found at {secrets_path}") secrets = load_yaml(secrets_path) - api_key = secrets.get('openai_api_key') + + if not 'llm_api_key' in secrets: + raise KeyError("No key as llm_api_key in the secret.yaml") + + api_key = secrets.get('llm_api_key') if not api_key: - raise ValueError("OpenAI API key not found in secrets.yaml") + raise ValueError("LLM API key not found in secrets.yaml") return api_key @@ -153,4 +157,4 @@ def main(): print(f"An error occurred: {e}") if __name__ == "__main__": - main() \ No newline at end of file + main()