[refactor] rename class

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import pandas as pd
 
-from model.search.vector import RuriVoyagerSearchClient
+from model.search.vector import RuriDuckDBSearchClient
 from model.data.notion_db import fetch_sakurap_corpus
 
 
-def search(search_client: RuriVoyagerSearchClient):
+def search(search_client: RuriDuckDBSearchClient):
     def _search(query: str) -> pd.DataFrame:
         results = search_client.search_top_n(query)
         result = results[0]
@@ -21,7 +21,7 @@ if __name__ == "__main__":
     # Load dataset
     sakurap_df = fetch_sakurap_corpus("./data/sakurap_corpus.csv")
     # Initialize search client
-    search_client = RuriVoyagerSearchClient.from_dataframe(sakurap_df, "content")
+    search_client = RuriDuckDBSearchClient.from_dataframe(sakurap_df, "content")
 
     with gr.Blocks() as search_interface:
         gr.Markdown("""
@@ -49,4 +49,4 @@ if __name__ == "__main__":
 
     # App launch
     search_interface.queue()
-    search_interface.launch(server_name="0.0.0.0")
+    search_interface.launch(server_name="0.0.0.0")

model/search/vector.py

@@ -78,7 +78,7 @@ class RuriEmbedder:
         return embedding
 
 
-class RuriVoyagerSearchClient(BaseSearchClient):
+class RuriDuckDBSearchClient(BaseSearchClient):
     def __init__(self, dataset: pd.DataFrame, target: str,
                  vector_store_name: str,
                  model: RuriEmbedder):