bstraehle commited on
Commit
8451a7d
·
verified ·
1 Parent(s): 25d324c

Update custom_utils.py

Browse files
Files changed (1) hide show
  1. custom_utils.py +0 -16
custom_utils.py CHANGED
@@ -91,7 +91,6 @@ def process_records(data_frame):
91
 
92
  def vector_search(openai_api_key, user_query, db, collection, additional_stages=[], vector_index="vector_index"):
93
  query_embedding = get_text_embedding(openai_api_key, user_query)
94
- query_embedding2 = get_image_embedding(openai_api_key, user_query)
95
 
96
  if query_embedding is None:
97
  return "Invalid query or embedding generation failed."
@@ -141,21 +140,6 @@ def get_text_embedding(openai_api_key, text):
141
  input=text,
142
  model="text-embedding-3-small", dimensions=1536).data[0].embedding
143
  return embedding
144
- except Exception as e:
145
- print(f"Error in get_embedding: {e}")
146
- return None
147
-
148
- def get_image_embedding(openai_api_key, text):
149
- if not text or not isinstance(text, str):
150
- return None
151
-
152
- openai.api_key = openai_api_key
153
-
154
- try:
155
- embedding = openai.embeddings.create(
156
- input=text,
157
- model="clip-vit-base-patch32", dimensions=512).data[0].embedding
158
- return embedding
159
  except Exception as e:
160
  print(f"Error in get_embedding: {e}")
161
  return None
 
91
 
92
  def vector_search(openai_api_key, user_query, db, collection, additional_stages=[], vector_index="vector_index"):
93
  query_embedding = get_text_embedding(openai_api_key, user_query)
 
94
 
95
  if query_embedding is None:
96
  return "Invalid query or embedding generation failed."
 
140
  input=text,
141
  model="text-embedding-3-small", dimensions=1536).data[0].embedding
142
  return embedding
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
143
  except Exception as e:
144
  print(f"Error in get_embedding: {e}")
145
  return None