AliInamdar commited on
Commit
454c0a1
·
verified ·
1 Parent(s): 3d3cb20

Update utils/helper.py

Browse files
Files changed (1) hide show
  1. utils/helper.py +9 -1
utils/helper.py CHANGED
@@ -8,8 +8,16 @@ os.system("python3 -m pip install --no-cache-dir together")
8
 
9
  from together import Together
10
 
 
 
 
 
 
 
 
 
 
11
 
12
- client = Together(api_key=os.environ["f2f419e9e636fe7b6cd1bc7ac2e74ed72610efed52f89b48aef68ae88e12d65a"])
13
 
14
 
15
  def call_llama(prompt: str) -> str:
 
8
 
9
  from together import Together
10
 
11
+ import os
12
+
13
+ api_key = os.environ.get("f2f419e9e636fe7b6cd1bc7ac2e74ed72610efed52f89b48aef68ae88e12d65a")
14
+ if not api_key:
15
+ raise ValueError("API_KEY environment variable is not set.")
16
+
17
+ client = Together(api_key=api_key)
18
+
19
+
20
 
 
21
 
22
 
23
  def call_llama(prompt: str) -> str: