feat: Add Halo spinner to load_files function

pull/2/head
Saryev Rustam 1 year ago
parent 25b00a4aa0
commit c092d2d0d8

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

@ -5,6 +5,7 @@ import sys
import tiktoken
from git import Repo
from halo import Halo
from langchain import FAISS
from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
@ -41,6 +42,7 @@ class StreamStdOut(StreamingStdOutCallbackHandler):
sys.stdout.flush()
@Halo(text='📂 Loading files', spinner='dots')
def load_files(root_dir):
num_cpus = multiprocessing.cpu_count()
loaded_files = []

Loading…
Cancel
Save