Wrap asymmetric search model into SearchModels. Test notes search end-to-end

- Wrap asymmetric search model parameters into AsymmetricSearchModel class
- Create wrapper for all search type models. Put notes search model into it
- Test notes search end-to-end from client API layer to results.
  Use model build on test data
This commit is contained in:
Debanjum Singh Solanky
2021-09-29 20:24:27 -07:00
parent cde11a2331
commit e22e0b41e3
4 changed files with 51 additions and 27 deletions

View File

@@ -18,3 +18,15 @@ class SearchSettings():
image_search_enabled: bool = False
class AsymmetricSearchModel():
def __init__(self, entries, corpus_embeddings, bi_encoder, cross_encoder, top_k):
self.entries = entries
self.corpus_embeddings = corpus_embeddings
self.bi_encoder = bi_encoder
self.cross_encoder = cross_encoder
self.top_k = top_k
@dataclass
class SearchModels():
notes_search: AsymmetricSearchModel = None