kamkol commited on
Commit
e8d27a7
·
1 Parent(s): 47892f0

Fix OpenAIEmbeddings reference in exception handling

Browse files
Files changed (1) hide show
  1. 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"),