Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -37,7 +37,8 @@ def build_filter(pt: Optional[str], cat: Optional[str]) -> dict | None:
|
|
37 |
class SearchRequest(BaseModel):
|
38 |
description: str
|
39 |
top_k: int = 3
|
40 |
-
|
|
|
41 |
|
42 |
class BikeMatch(BaseModel):
|
43 |
id: str
|
@@ -187,7 +188,7 @@ async def search_ebikes(request:SearchRequest):
|
|
187 |
try:
|
188 |
# Create embedding for the query
|
189 |
query_embedding = encoder.encode(request.description)[0]
|
190 |
-
filter_payload = build_filter(request.
|
191 |
|
192 |
# Query Pinecone
|
193 |
results = pinecone_index.query(
|
|
|
37 |
class SearchRequest(BaseModel):
|
38 |
description: str
|
39 |
top_k: int = 3
|
40 |
+
product_type: str
|
41 |
+
category : stry
|
42 |
|
43 |
class BikeMatch(BaseModel):
|
44 |
id: str
|
|
|
188 |
try:
|
189 |
# Create embedding for the query
|
190 |
query_embedding = encoder.encode(request.description)[0]
|
191 |
+
filter_payload = build_filter(request.product_type, request.category)
|
192 |
|
193 |
# Query Pinecone
|
194 |
results = pinecone_index.query(
|