Spaces:
Sleeping
Sleeping
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
|