Artin2009 commited on
Commit
1de578b
1 Parent(s): f716596

Update chain_app.py

Browse files
Files changed (1) hide show
  1. chain_app.py +2 -2
chain_app.py CHANGED
@@ -24,7 +24,7 @@ groq_client = Groq(api_key=groq_api_key)
24
  co = cohere.Client(
25
  api_key=cohere_api_key,
26
  )
27
- NVIDIA_API_KEY = os.environ.get('NVIDIA_API_KEY')
28
 
29
  # API_URL = "https://api-inference.huggingface.co/models/PartAI/TookaBERT-Large"
30
  # headers = {"Authorization": f"Bearer {hf_token}"}
@@ -2330,7 +2330,7 @@ async def main(message: cl.Message):
2330
  # ).send()
2331
  client = OpenAI(
2332
  base_url = "https://integrate.api.nvidia.com/v1",
2333
- api_key = nvidia_api
2334
  )
2335
 
2336
  completion = client.chat.completions.create(
 
24
  co = cohere.Client(
25
  api_key=cohere_api_key,
26
  )
27
+ nvidia_api_key = os.environ.get('NVIDIA_API_KEY')
28
 
29
  # API_URL = "https://api-inference.huggingface.co/models/PartAI/TookaBERT-Large"
30
  # headers = {"Authorization": f"Bearer {hf_token}"}
 
2330
  # ).send()
2331
  client = OpenAI(
2332
  base_url = "https://integrate.api.nvidia.com/v1",
2333
+ api_key = nvidia_api_key
2334
  )
2335
 
2336
  completion = client.chat.completions.create(