github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-18 07:30:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Upgrade spacy model (#1511)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-05-11 15:41:06 +08:00
committed by GitHub
parent 8eb64d7311
commit bc9ce3c2ae
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dbgpt/storage/vector_store/chroma_store.py
View File

@@ -82,7 +82,7 @@ class ChromaStore(VectorStoreBase):
# client_settings=chroma_settings,
client=client,
collection_metadata=collection_metadata,
) # type: ignore
) # type: ignore
def similar_search(
self, text, topk, filters: Optional[MetadataFilters] = None

6
docker/base/Dockerfile
View File

@@ -36,9 +36,9 @@ RUN pip3 install --upgrade pip -i $PIP_INDEX_URL \
RUN (if [ "${LANGUAGE}" = "zh" ]; \
# language is zh, download zh_core_web_sm from github
then wget https://github.com/explosion/spacy-models/releases/download/zh_core_web_sm-3.5.0/zh_core_web_sm-3.5.0-py3-none-any.whl -O /tmp/zh_core_web_sm-3.5.0-py3-none-any.whl \
&& pip3 install /tmp/zh_core_web_sm-3.5.0-py3-none-any.whl -i $PIP_INDEX_URL \
&& rm /tmp/zh_core_web_sm-3.5.0-py3-none-any.whl; \
then wget https://github.com/explosion/spacy-models/releases/download/zh_core_web_sm-3.7.0/zh_core_web_sm-3.7.0-py3-none-any.whl -O /tmp/zh_core_web_sm-3.7.0-py3-none-any.whl \
&& pip3 install /tmp/zh_core_web_sm-3.7.0-py3-none-any.whl -i $PIP_INDEX_URL \
&& rm /tmp/zh_core_web_sm-3.7.0-py3-none-any.whl; \
# not zh, download directly
else python3 -m spacy download zh_core_web_sm; \
fi;) \