dzenzzz commited on
Commit
beb4147
·
1 Parent(s): 6e918d9

adds timeout

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. neural_searcher.py +1 -1
app.py CHANGED
@@ -16,8 +16,8 @@ neural_searcher = NeuralSearcher(collection_name=os.getenv('COLLECTION_NAME'))
16
  REQUEST_TIMEOUT_ERROR = 1
17
 
18
  @app.get("/api/search")
19
- async def search(q: str):
20
- data = await neural_searcher.search(text=q)
21
  return data
22
 
23
 
 
16
  REQUEST_TIMEOUT_ERROR = 1
17
 
18
  @app.get("/api/search")
19
+ def search(q: str):
20
+ data = neural_searcher.search(text=q)
21
  return data
22
 
23
 
neural_searcher.py CHANGED
@@ -13,7 +13,7 @@ class NeuralSearcher:
13
  self.late_interaction_model = LateInteractionTextEmbedding(os.getenv('LATE_INTERACTION_MODEL'))
14
  self.qdrant_client = QdrantClient(os.getenv('QDRANT_URL'),api_key=os.getenv('QDRANT_API_KEY'),https=True)
15
 
16
- async def search(self, text: str):
17
 
18
  dense_query = self.dense_model.encode(text).tolist()
19
  sparse_query = next(self.sparse_model.query_embed(text))
 
13
  self.late_interaction_model = LateInteractionTextEmbedding(os.getenv('LATE_INTERACTION_MODEL'))
14
  self.qdrant_client = QdrantClient(os.getenv('QDRANT_URL'),api_key=os.getenv('QDRANT_API_KEY'),https=True)
15
 
16
+ def search(self, text: str):
17
 
18
  dense_query = self.dense_model.encode(text).tolist()
19
  sparse_query = next(self.sparse_model.query_embed(text))