github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-05 02:20:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'dev' of github.com:csunny/DB-GPT into dev

Browse Source
This commit is contained in:
csunny 2023-05-05 22:44:36 +08:00
commit d9f5130db4
2 changed files with 4 additions and 2 deletions
Show Stats 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

2
requirements.txt
View File

@ -51,3 +51,5 @@ 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
pymysql