DiamondYin commited on
Commit
83d3d13
·
1 Parent(s): b846d49

Update app_utils.py

Browse files
Files changed (1) hide show
  1. app_utils.py +1 -2
app_utils.py CHANGED
@@ -24,14 +24,13 @@ from tenacity import (
24
  # FUNCTIONS
25
 
26
  # get embeddings
27
- @retry(wait=wait_random_exponential(min=1, max=60), stop=stop_after_attempt(6))
28
  @st.cache_data
29
  def embedding_from_string(input: str, model: str) -> list:
30
  response = openai.Embedding.create(input=input, model=model)
31
  embedding = response["data"][0]["embedding"]
32
  return embedding
33
 
34
-
35
  OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
36
  AWS_ACCESS_KEY_ID = os.getenv('AWS_ACCESS_KEY_ID')
37
  AWS_SECRET_ACCESS_KEY = os.getenv('AWS_SECRET_ACCESS_KEY')
 
24
  # FUNCTIONS
25
 
26
  # get embeddings
27
+ @retry(wait=wait_random_exponential(min=21, max=60), stop=stop_after_attempt(6))
28
  @st.cache_data
29
  def embedding_from_string(input: str, model: str) -> list:
30
  response = openai.Embedding.create(input=input, model=model)
31
  embedding = response["data"][0]["embedding"]
32
  return embedding
33
 
 
34
  OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
35
  AWS_ACCESS_KEY_ID = os.getenv('AWS_ACCESS_KEY_ID')
36
  AWS_SECRET_ACCESS_KEY = os.getenv('AWS_SECRET_ACCESS_KEY')