Fix formatting

dbs/elasticsearch/api/routers/rest.py

@@ -1,12 +1,11 @@
-from elasticsearch import AsyncElasticsearch
-from fastapi import APIRouter, HTTPException, Query, Request
-
 from api.schemas.rest import (
     CountByCountry,
     FullTextSearch,
     TopWinesByCountry,
     TopWinesByProvince,
 )
+from elasticsearch import AsyncElasticsearch
+from fastapi import APIRouter, HTTPException, Query, Request
 
 router = APIRouter()
 

dbs/lancedb/api/main.py

@@ -4,12 +4,11 @@
 
 import lancedb
 from fastapi import FastAPI
+from sentence_transformers import SentenceTransformer
 
 from api.config import Settings
 from api.routers.rest import router
 
-from sentence_transformers import SentenceTransformer
-
 model_type = "sbert"
 
 

dbs/lancedb/api/schemas/rest.py

@@ -1,4 +1,5 @@
 from typing import Optional
+
 from pydantic import BaseModel, ConfigDict
 
 

dbs/lancedb/scripts/bulk_index_sbert.py

@@ -3,7 +3,6 @@
 import sys
 from concurrent.futures import ProcessPoolExecutor, as_completed
 from functools import lru_cache
-from math import e
 from pathlib import Path
 from typing import Any, Iterator
 
@@ -16,10 +15,9 @@
 from tqdm import tqdm
 
 sys.path.insert(1, os.path.realpath(Path(__file__).resolve().parents[1]))
-from sentence_transformers import SentenceTransformer
-
 from api.config import Settings
 from schemas.wine import LanceModelWine, Wine
+from sentence_transformers import SentenceTransformer
 
 load_dotenv()
 # Custom types
@@ -120,18 +118,6 @@ def main(data: list[JsonBlob]) -> None:
         tbl.create_index(metric="cosine", num_partitions=32, num_sub_vectors=96)
 
 
-def query_func() -> None:
-    DB_NAME = "../lancedb"
-    TABLE = "wines"
-    db = lancedb.connect(DB_NAME)
-    tbl = db.open_table(TABLE)
-    print(len(tbl))
-    MODEL = SentenceTransformer(get_settings().embedding_model_checkpoint)
-    query = "tropical fruit and citrus"
-    query_vector = embed_func([query], MODEL)[0]
-    print(tbl.search(query_vector).limit(5).to_df()[["id", "title", "description", "_distance"]])
-
-
 if __name__ == "__main__":
     # fmt: off
     parser = argparse.ArgumentParser("Bulk index database from the wine reviews JSONL data")
@@ -153,5 +139,3 @@ def query_func() -> None:
     data = data[:LIMIT] if LIMIT > 0 else data
     main(data)
     print("Finished execution!")
-
-    query_func()

dbs/meilisearch/api/routers/rest.py

@@ -1,11 +1,10 @@
-from fastapi import APIRouter, HTTPException, Query, Request
-from meilisearch_python_async import Client
-
 from api.schemas.rest import (
     FullTextSearch,
     TopWinesByCountry,
     TopWinesByProvince,
 )
+from fastapi import APIRouter, HTTPException, Query, Request
+from meilisearch_python_async import Client
 
 router = APIRouter()
 

dbs/neo4j/api/routers/rest.py

@@ -1,12 +1,11 @@
-from fastapi import APIRouter, HTTPException, Query, Request
-from neo4j import AsyncManagedTransaction
-
 from api.schemas.rest import (
     FullTextSearch,
     MostWinesByVariety,
     TopWinesByCountry,
     TopWinesByProvince,
 )
+from fastapi import APIRouter, HTTPException, Query, Request
+from neo4j import AsyncManagedTransaction
 
 router = APIRouter()
 

dbs/neo4j/scripts/build_graph.py

@@ -13,7 +13,6 @@
 
 sys.path.insert(1, os.path.realpath(Path(__file__).resolve().parents[1]))
 from api.config import Settings
-
 from schemas.wine import Wine
 
 # Custom types

dbs/qdrant/api/routers/rest.py

@@ -1,8 +1,7 @@
+from api.schemas.rest import CountByCountry, SimilaritySearch
 from fastapi import APIRouter, HTTPException, Query, Request
 from qdrant_client.http import models
 
-from api.schemas.rest import CountByCountry, SimilaritySearch
-
 router = APIRouter()
 
 

dbs/qdrant/scripts/bulk_index_sbert.py

@@ -12,10 +12,9 @@
 from qdrant_client.http import models
 
 sys.path.insert(1, os.path.realpath(Path(__file__).resolve().parents[1]))
-from sentence_transformers import SentenceTransformer
-
 from api.config import Settings
 from schemas.wine import Wine
+from sentence_transformers import SentenceTransformer
 
 load_dotenv()
 # Custom types

dbs/weaviate/api/main.py

@@ -4,12 +4,11 @@
 
 import weaviate
 from fastapi import FastAPI
+from sentence_transformers import SentenceTransformer
 
 from api.config import Settings
 from api.routers import rest
 
-from sentence_transformers import SentenceTransformer
-
 model_type = "sbert"
 
 

dbs/weaviate/api/routers/rest.py

@@ -1,6 +1,5 @@
-from fastapi import APIRouter, HTTPException, Query, Request
-
 from api.schemas.rest import CountByCountry, SimilaritySearch
+from fastapi import APIRouter, HTTPException, Query, Request
 
 router = APIRouter()