瀏覽代碼

adding octoAI api key

Hamid Shojanazeri 1 年之前
父節點
當前提交
2580bb14a6
共有 1 個文件被更改,包括 2 次插入3 次删除
  1. 2 3
      tutorials/chatbot/data_pipelines/config.py

+ 2 - 3
tutorials/chatbot/data_pipelines/config.py

@@ -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