Spaces:
Runtime error
Runtime error
Commit
·
83d3d13
1
Parent(s):
b846d49
Update app_utils.py
Browse files- 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=
|
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')
|