Spaces:
Sleeping
Sleeping
Fix OpenAIEmbeddings reference in exception handling
Browse files- streamlit_app.py +3 -3
streamlit_app.py
CHANGED
|
@@ -111,15 +111,15 @@ def get_agent_model():
|
|
| 111 |
@st.cache_resource
|
| 112 |
def get_embedding_model():
|
| 113 |
"""Get the embedding model."""
|
|
|
|
|
|
|
|
|
|
| 114 |
try:
|
| 115 |
# First try the standard approach
|
| 116 |
return OpenAIEmbeddings(model="text-embedding-3-small")
|
| 117 |
except Exception as e:
|
| 118 |
if "proxies" in str(e):
|
| 119 |
# If there's a proxies error, create with additional kwargs to override defaults
|
| 120 |
-
import os
|
| 121 |
-
from langchain_openai import OpenAIEmbeddings
|
| 122 |
-
|
| 123 |
return OpenAIEmbeddings(
|
| 124 |
model="text-embedding-3-small",
|
| 125 |
openai_api_key=os.environ.get("OPENAI_API_KEY"),
|
|
|
|
| 111 |
@st.cache_resource
|
| 112 |
def get_embedding_model():
|
| 113 |
"""Get the embedding model."""
|
| 114 |
+
from langchain_openai import OpenAIEmbeddings
|
| 115 |
+
import os
|
| 116 |
+
|
| 117 |
try:
|
| 118 |
# First try the standard approach
|
| 119 |
return OpenAIEmbeddings(model="text-embedding-3-small")
|
| 120 |
except Exception as e:
|
| 121 |
if "proxies" in str(e):
|
| 122 |
# If there's a proxies error, create with additional kwargs to override defaults
|
|
|
|
|
|
|
|
|
|
| 123 |
return OpenAIEmbeddings(
|
| 124 |
model="text-embedding-3-small",
|
| 125 |
openai_api_key=os.environ.get("OPENAI_API_KEY"),
|