Omkar192002 commited on
Commit
4316268
·
verified ·
1 Parent(s): 6d3e81c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -29,7 +29,6 @@ app = FastAPI(title="E-Bikes Semantic Search API",
29
  # Request and response models
30
  class SearchRequest(BaseModel):
31
  description: str
32
- top_k: int = 3
33
 
34
  class BikeMatch(BaseModel):
35
  id: str
@@ -182,7 +181,7 @@ async def search_ebikes(request: SearchRequest):
182
  # Query Pinecone
183
  results = pinecone_index.query(
184
  vector=query_embedding.tolist(),
185
- top_k=request.top_k,
186
  include_metadata=True
187
  )
188
 
 
29
  # Request and response models
30
  class SearchRequest(BaseModel):
31
  description: str
 
32
 
33
  class BikeMatch(BaseModel):
34
  id: str
 
181
  # Query Pinecone
182
  results = pinecone_index.query(
183
  vector=query_embedding.tolist(),
184
+ top_k=3,
185
  include_metadata=True
186
  )
187