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