From ec92f3e1468a6c9991926614c27c090a1cc5ca84 Mon Sep 17 00:00:00 2001
From: Debanjum Singh Solanky <debanjum@gmail.com>
Date: Sun, 15 Aug 2021 19:09:50 -0700
Subject: [PATCH] Move different search types into search_types directory

---
 main.py                                         | 14 ++++++++------
 search_types/__init__.py                        |  0
 asymmetric.py => search_types/asymmetric.py     |  0
 image-search.py => search_types/image-search.py |  0
 symmetric.py => search_types/symmetric.py       |  0
 5 files changed, 8 insertions(+), 6 deletions(-)
 create mode 100644 search_types/__init__.py
 rename asymmetric.py => search_types/asymmetric.py (100%)
 rename image-search.py => search_types/image-search.py (100%)
 rename symmetric.py => search_types/symmetric.py (100%)

diff --git a/main.py b/main.py
index f25b40a1..200c56b3 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,8 @@
 from typing import Optional
 from fastapi import FastAPI
-from asymmetric import *
+from search_types import asymmetric
+import argparse
+import pathlib
 import uvicorn
 
 app = FastAPI()
@@ -8,7 +10,7 @@ app = FastAPI()
 def create_search_notes(corpus_embeddings, entries, bi_encoder, cross_encoder, top_k):
     "Closure to create search_notes method from initialized model, entries and embeddings"
     def search_notes(query):
-        return query_notes(
+        return asymmetric.query_notes(
             query,
             corpus_embeddings,
             entries,
@@ -33,7 +35,7 @@ def search(q: str, n: Optional[int] = 5, t: Optional[str] = 'notes'):
         hits = search_notes(user_query)
 
         # collate and return results
-        return collate_results(hits, entries, results_count)
+        return asymmetric.collate_results(hits, entries, results_count)
 
     else:
         return {}
@@ -48,13 +50,13 @@ if __name__ == '__main__':
     args = parser.parse_args()
 
     # Initialize Model
-    bi_encoder, cross_encoder, top_k = initialize_model()
+    bi_encoder, cross_encoder, top_k = asymmetric.initialize_model()
 
     # Extract Entries
-    entries = extract_entries(args.jsonl_file, args.verbose)
+    entries = asymmetric.extract_entries(args.jsonl_file, args.verbose)
 
     # Compute or Load Embeddings
-    corpus_embeddings = compute_embeddings(entries, bi_encoder, args.embeddings_file, args.verbose)
+    corpus_embeddings = asymmetric.compute_embeddings(entries, bi_encoder, args.embeddings_file, args.verbose)
 
     # Generate search_notes method from initialized model, entries and embeddings
     search_notes = create_search_notes(corpus_embeddings, entries, bi_encoder, cross_encoder, top_k)
diff --git a/search_types/__init__.py b/search_types/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/asymmetric.py b/search_types/asymmetric.py
similarity index 100%
rename from asymmetric.py
rename to search_types/asymmetric.py
diff --git a/image-search.py b/search_types/image-search.py
similarity index 100%
rename from image-search.py
rename to search_types/image-search.py
diff --git a/symmetric.py b/search_types/symmetric.py
similarity index 100%
rename from symmetric.py
rename to search_types/symmetric.py