github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-10-21 17:09:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

update requirements

Browse Source
This commit is contained in:
csunny
2023-05-05 22:29:33 +08:00
parent 7da910b642
commit d91a0e7d56
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/vector_store/extract_tovec.py
View File

@@ -34,7 +34,7 @@ def knownledge_tovec_st(filename):
text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0) text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
texts = text_splitter(knownledge) texts = text_splitter.split_text(knownledge)
docsearch = Chroma.from_texts(texts, embeddings, metadatas=[{"source": str(i)} for i in range(len(texts))]) docsearch = Chroma.from_texts(texts, embeddings, metadatas=[{"source": str(i)} for i in range(len(texts))])
return docsearch return docsearch

3
requirements.txt
View File

@@ -50,4 +50,5 @@ notebook
gradio==3.24.1 gradio==3.24.1
gradio-client==0.0.8 gradio-client==0.0.8
wandb wandb
fschat=0.1.10 fschat=0.1.10
llama-index=0.5.27