MCPcopy Index your code
hub / github.com/CL-lau/SQL-GPT / batchQuery_preprocessing

Method batchQuery_preprocessing

embedding/embeddingHelper.py:420–432  ·  view source on GitHub ↗
(self, texts, top_k=3)

Source from the content-addressed store, hash-verified

418 return None
419
420 def batchQuery_preprocessing(self, texts, top_k=3):
421 if self.need_redis:
422 if isinstance(texts, list):
423 res = []
424 for text in texts:
425 vec = self.embedding_model([text])[0]
426 ans = self.redisCaching.query_cache(query=text, query_vec=np.array(vec), k=top_k)
427 if ans is not None:
428 ans = str(ans).split('___')
429 res.append(ans)
430 if len(res) == len(texts):
431 return res
432 return None
433
434 def query_embedding_preprocessing(self, text, thresholds=.9):
435 if self.need_redis:

Callers 1

batchQueryMethod · 0.95

Calls 2

query_cacheMethod · 0.80
splitMethod · 0.80

Tested by

no test coverage detected