feat: add Halo spinner when creating vector store

- Add Halo spinner when creating vector store to provide visual feedback to the user.
- Also update success message to reflect the number of documents in the vector store.

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "talk-codebase"
-version = "0.1.16"
+version = "0.1.17"
 description = "talk-codebase is a powerful tool for querying and analyzing codebases."
 authors = ["Saryev Rustam <rustam1997@gmail.com>"]
 readme = "README.md"

talk_codebase/llm.py

@@ -2,6 +2,7 @@ import os
 
 import questionary
 import tiktoken
+from halo import Halo
 from langchain import FAISS
 from langchain.callbacks.manager import CallbackManager
 from langchain.chains import ConversationalRetrievalChain
@@ -42,8 +43,10 @@ def create_vector_store(root_dir, openai_api_key):
     if not approve:
         exit(0)
 
+    spinners = Halo(text='Creating vector store', spinner='dots').start()
     embeddings = OpenAIEmbeddings(openai_api_key=openai_api_key)
     db = FAISS.from_documents(texts, embeddings)
+    spinners.succeed(f"Created vector store with {len(docs)} documents")
 
     return db