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(
|