diff --git a/.env.example b/.env.example index 7a86880..5e7fcf0 100644 --- a/.env.example +++ b/.env.example @@ -1,3 +1,2 @@ OPENAI_API_KEY=XXXXXXXXXXX -HUGGINGFACE_API_KEY=YYYYYYYYYY ESBMC_AI_CFG_PATH=./config.json \ No newline at end of file diff --git a/esbmc_ai/api_key_collection.py b/esbmc_ai/api_key_collection.py index 9aeec6f..e34d72e 100644 --- a/esbmc_ai/api_key_collection.py +++ b/esbmc_ai/api_key_collection.py @@ -9,4 +9,3 @@ class APIKeyCollection(NamedTuple): """Class that is used to pass keys to AIModels.""" openai: Optional[str] - huggingface: Optional[str] diff --git a/esbmc_ai/config.py b/esbmc_ai/config.py index d99141c..155dc6a 100644 --- a/esbmc_ai/config.py +++ b/esbmc_ai/config.py @@ -205,7 +205,7 @@ def get_env_vars() -> None: if value != None: values[k] = value - keys: list[str] = ["OPENAI_API_KEY", "HUGGINGFACE_API_KEY", "ESBMC_AI_CFG_PATH"] + keys: list[str] = ["OPENAI_API_KEY", "ESBMC_AI_CFG_PATH"] values: dict[str, str] = {} # Load from system env @@ -246,7 +246,6 @@ def get_env_vars() -> None: global api_keys api_keys = APIKeyCollection( openai=str(os.getenv("OPENAI_API_KEY")), - huggingface=str(os.getenv("HUGGINGFACE_API_KEY")), ) global cfg_path