Spaces:
Running
Running
Update custom_utils.py
Browse files- 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
|