|
@@ -4,12 +4,11 @@
|
|
|
import yaml
|
|
|
import os
|
|
|
|
|
|
-
|
|
|
def load_config():
|
|
|
# Read the YAML configuration file
|
|
|
file_path = "./config.yaml"
|
|
|
with open(file_path, "r") as file:
|
|
|
config = yaml.safe_load(file)
|
|
|
-
|
|
|
- config["api_key"] = os.getenv('OPENAI_API_KEY')
|
|
|
+ # Set the API key from the environment variable
|
|
|
+ config["api_key"] = os.environ["OCTOAI_API_TOKEN"]
|
|
|
return config
|