Omkar192002 commited on
Commit
72db647
·
verified ·
1 Parent(s): 479a5d1

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -2
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
- filters: dict
 
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.filters.get("product_type"), request.filters.get("category"))
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(