mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-14 16:16:48 +00:00
Merge branch 'main' into anydoc-support
Conflicts, and how each was taken: - application/core/settings.py — ours. The renamed OCR_ENABLED / OCR_ATTACHMENTS_ENABLED / OCR_MIN_CHARS_PER_PAGE accept main's DOCLING_OCR_* spellings as AliasChoices, so nothing is dropped. - application/Dockerfile — both. Main's install layers plus the INSTALL_DOCLING build arg. - application/parser/file/constants.py — both imports. - deployment/docker-compose.yaml — both. The INSTALL_DOCLING / INSTALL_TESSERACT build args on backend and worker, and main's -Q docsgpt,parsing,embeddings, which query embedding needs. - tests/conftest.py — theirs. Both sides fixed the same pytest-postgresql 9.0.0 autocommit= breakage; main's spelling is the one already on main. - application/requirements.txt — the comments claimed different reasons torch is in core. Main's is the true one now: it removed sentence-transformers, so docling is torch's only remaining consumer. Two things the merge broke without conflicting: - onnxruntime. This branch moved it out of core into the docling extra; main meanwhile made it the runtime local embeddings execute on (fastembed). Git took the deletion, leaving fastembed with no pinned runtime in a repo that pins everything. Restored to core, and no longer pinned twice from the extra. - The frontend copy of ATTACHMENT_PARSER_EXTENSIONS. The backend list is derived and picked up the anydoc suffixes; the hand-kept frontend mirror did not, so the composer would refuse files the API accepts. tests/parser/file/test_constants.py is what caught it. ruff, pytest (9897 passed), frontend build and docs build all pass. The image build is unverified: no Docker daemon on this machine.
This commit is contained in:
commit
3947c66cda
97 files changed
+8661
-1049
No files matched your search
@@ -250,17 +250,18 @@ configure_vector_store() {
|
||||
configure_embeddings() {
|
||||
echo -e "\n${DEFAULT_FG}${BOLD}Embeddings Configuration${NC}"
|
||||
echo -e "${DEFAULT_FG}Choose your embeddings provider:${NC}"
|
||||
echo -e "${YELLOW}1) HuggingFace (default, local)${NC}"
|
||||
echo -e "${YELLOW}1) Granite multilingual (default, local)${NC}"
|
||||
echo -e "${YELLOW}2) OpenAI Embeddings${NC}"
|
||||
echo -e "${YELLOW}3) Custom Remote Embeddings (OpenAI-compatible API)${NC}"
|
||||
echo -e "${YELLOW}4) all-mpnet-base-v2 (legacy local, English-only)${NC}"
|
||||
echo -e "${YELLOW}b) Back${NC}"
|
||||
echo
|
||||
read -p "$(echo -e "${DEFAULT_FG}Choose option (1-3, or b): ${NC}")" emb_choice
|
||||
read -p "$(echo -e "${DEFAULT_FG}Choose option (1-4, or b): ${NC}")" emb_choice
|
||||
|
||||
case "$emb_choice" in
|
||||
1)
|
||||
echo "EMBEDDINGS_NAME=huggingface_sentence-transformers/all-mpnet-base-v2" >> "$ENV_FILE"
|
||||
echo -e "${GREEN}Embeddings set to HuggingFace (local).${NC}"
|
||||
echo "EMBEDDINGS_NAME=ibm-granite/granite-embedding-311m-multilingual-r2" >> "$ENV_FILE"
|
||||
echo -e "${GREEN}Embeddings set to granite-311m-multilingual-r2 (local).${NC}"
|
||||
;;
|
||||
2)
|
||||
echo "EMBEDDINGS_NAME=openai_text-embedding-ada-002" >> "$ENV_FILE"
|
||||
@@ -277,6 +278,10 @@ configure_embeddings() {
|
||||
[ -n "$emb_key" ] && echo "EMBEDDINGS_KEY=$emb_key" >> "$ENV_FILE"
|
||||
echo -e "${GREEN}Custom remote embeddings configured.${NC}"
|
||||
;;
|
||||
4)
|
||||
echo "EMBEDDINGS_NAME=huggingface_sentence-transformers/all-mpnet-base-v2" >> "$ENV_FILE"
|
||||
echo -e "${GREEN}Embeddings set to all-mpnet-base-v2 (legacy local).${NC}"
|
||||
;;
|
||||
b|B) return ;;
|
||||
*) echo -e "\n${RED}Invalid choice.${NC}" ; sleep 1 ;;
|
||||
esac
|
||||
@@ -541,7 +546,7 @@ serve_local_ollama() {
|
||||
echo "LLM_NAME=$model_name" >> "$ENV_FILE"
|
||||
echo "VITE_API_STREAMING=true" >> "$ENV_FILE"
|
||||
echo "OPENAI_BASE_URL=http://ollama:11434/v1" >> "$ENV_FILE"
|
||||
echo "EMBEDDINGS_NAME=huggingface_sentence-transformers/all-mpnet-base-v2" >> "$ENV_FILE"
|
||||
echo "EMBEDDINGS_NAME=ibm-granite/granite-embedding-311m-multilingual-r2" >> "$ENV_FILE"
|
||||
echo -e "${GREEN}.env file configured for Ollama ($(echo "$docker_compose_file_suffix" | tr '[:lower:]' '[:upper:]')${NC}${GREEN}).${NC}"
|
||||
|
||||
prompt_advanced_settings
|
||||
@@ -654,7 +659,7 @@ connect_local_inference_engine() {
|
||||
echo "LLM_NAME=$model_name" >> "$ENV_FILE"
|
||||
echo "VITE_API_STREAMING=true" >> "$ENV_FILE"
|
||||
echo "OPENAI_BASE_URL=$openai_base_url" >> "$ENV_FILE"
|
||||
echo "EMBEDDINGS_NAME=huggingface_sentence-transformers/all-mpnet-base-v2" >> "$ENV_FILE"
|
||||
echo "EMBEDDINGS_NAME=ibm-granite/granite-embedding-311m-multilingual-r2" >> "$ENV_FILE"
|
||||
echo -e "${GREEN}.env file configured for ${BOLD}${engine_name}${NC}${GREEN} with OpenAI API format.${NC}"
|
||||
echo -e "${YELLOW}Note: MODEL_NAME is set to '${BOLD}$model_name${NC}${YELLOW}'. You can change it later in the .env file.${NC}"
|
||||
|
||||
|
||||
Reference in new issue
Block a user