(self, text: str, top_k=3)
| 406 | # TODO 做一些打点标记 |
| 407 | |
| 408 | def query_preprocessing(self, text: str, top_k=3): |
| 409 | if self.need_redis: |
| 410 | if isinstance(text, str): |
| 411 | vec = self.embedding_model([text])[0] |
| 412 | res = [] |
| 413 | ans = self.redisCaching.query_cache(query=text, query_vec=np.array(vec), k=top_k) |
| 414 | if ans is not None: |
| 415 | ans = str(ans).split('___') |
| 416 | res.append(ans) |
| 417 | return res |
| 418 | return None |
| 419 | |
| 420 | def batchQuery_preprocessing(self, texts, top_k=3): |
| 421 | if self.need_redis: |
no test coverage detected