rookie9 commited on
Commit
bd9bdce
·
verified ·
1 Parent(s): 02619a8

Update utils/llm_xiapi.py

Browse files
Files changed (1) hide show
  1. utils/llm_xiapi.py +3 -2
utils/llm_xiapi.py CHANGED
@@ -1,10 +1,11 @@
1
  import requests
2
  import json
3
-
4
  url = "https://api.xi-ai.cn/v1/chat/completions"
 
5
  headers = {
6
  "Content-Type": "application/json",
7
- "Authorization": "sk-iPNe4meYeEmknmgx7c31F953548148F7805555D1Cc0c506e" #your api key
8
  }
9
 
10
  training_info_pri = """
 
1
  import requests
2
  import json
3
+ import os
4
  url = "https://api.xi-ai.cn/v1/chat/completions"
5
+ api_key = os.environ.get("api_key")
6
  headers = {
7
  "Content-Type": "application/json",
8
+ "Authorization": api_key #your api key
9
  }
10
 
11
  training_info_pri = """