mirror of
https://github.com/langflow-ai/langflow.git
synced 2026-07-24 05:39:16 +08:00
edb3d0cc10
* Refactor API key handling in tests for consistency Centralizes and improves API key retrieval and validation by using shared utility functions from tests/api_keys.py. Updates all test files to use get_openai_api_key and has_api_key, ensuring consistent skipping of tests when the API key is missing, empty, or set to 'dummy'. * Remove unused imports and improve import grouping in tests Eliminated unused 'os' imports and reorganized import statements for clarity across multiple test files. This cleanup reduces clutter and improves code readability without affecting test logic or functionality. * [autofix.ci] apply automated fixes --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
import os
|
|
|
|
import pytest
|
|
|
|
# we need to import tmpdir
|
|
|
|
|
|
def get_required_env_var(var: str) -> str:
|
|
"""Get the value of the specified environment variable.
|
|
|
|
Args:
|
|
var (str): The environment variable to get.
|
|
|
|
Returns:
|
|
str: The value of the environment variable.
|
|
|
|
Raises:
|
|
ValueError: If the environment variable is not set.
|
|
"""
|
|
value = os.getenv(var)
|
|
if not value:
|
|
msg = f"Environment variable {var} is not set"
|
|
raise ValueError(msg)
|
|
if not value.strip():
|
|
msg = f"Environment variable {var} is empty"
|
|
raise ValueError(msg)
|
|
if value == "dummy":
|
|
msg = f"Environment variable {var} is set to dummy"
|
|
raise ValueError(msg)
|
|
return value
|
|
|
|
|
|
def get_openai_api_key() -> str:
|
|
try:
|
|
return get_required_env_var("OPENAI_API_KEY")
|
|
except ValueError:
|
|
pytest.skip("OPENAI_API_KEY is not set")
|
|
|
|
|
|
def get_astradb_application_token() -> str:
|
|
return get_required_env_var("ASTRA_DB_APPLICATION_TOKEN")
|
|
|
|
|
|
def get_astradb_api_endpoint() -> str:
|
|
return get_required_env_var("ASTRA_DB_API_ENDPOINT")
|
|
|
|
|
|
def has_api_key(env_var: str) -> bool:
|
|
"""Return True if the given env var exists and is non-empty after stripping."""
|
|
try:
|
|
bool(get_required_env_var(env_var))
|
|
except ValueError:
|
|
return False
|
|
return True
|