Skip to content

Commit

Permalink
fix: Upgrade spacy model (eosphoros-ai#1511)
Browse files Browse the repository at this point in the history
  • Loading branch information
fangyinc authored May 11, 2024
1 parent 8eb64d7 commit bc9ce3c
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion dbgpt/storage/vector_store/chroma_store.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ def __init__(self, vector_store_config: ChromaVectorConfig) -> None:
# client_settings=chroma_settings,
client=client,
collection_metadata=collection_metadata,
) # type: ignore
) # type: ignore

def similar_search(
self, text, topk, filters: Optional[MetadataFilters] = None
Expand Down
6 changes: 3 additions & 3 deletions docker/base/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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;) \
Expand Down

0 comments on commit bc9ce3c

Please sign in to comment.