AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-13 22:55:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

refine dockerfile (#1801)

Browse Source 
### What problem does this PR solve?


### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2024-08-05 08:58:22 +08:00 committed by GitHub
parent 216f6495c4
commit 29509ff69d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -10,7 +10,7 @@ ADD ./api ./api
ADD ./conf ./conf ADD ./conf ./conf
ADD ./deepdoc ./deepdoc ADD ./deepdoc ./deepdoc
ADD ./rag ./rag ADD ./rag ./rag
ADD ./graph ./graph ADD ./agent ./agent
ENV PYTHONPATH=/ragflow/ ENV PYTHONPATH=/ragflow/
ENV HF_ENDPOINT=https://hf-mirror.com ENV HF_ENDPOINT=https://hf-mirror.com

2
Dockerfile.arm
View File

@ -21,7 +21,7 @@ ADD ./api ./api
ADD ./conf ./conf ADD ./conf ./conf
ADD ./deepdoc ./deepdoc ADD ./deepdoc ./deepdoc
ADD ./rag ./rag ADD ./rag ./rag
ADD ./graph ./graph ADD ./agent ./agent
ENV PYTHONPATH=/ragflow/ ENV PYTHONPATH=/ragflow/
ENV HF_ENDPOINT=https://hf-mirror.com ENV HF_ENDPOINT=https://hf-mirror.com

2
Dockerfile.cuda
View File

@ -15,7 +15,7 @@ ADD ./api ./api
ADD ./conf ./conf ADD ./conf ./conf
ADD ./deepdoc ./deepdoc ADD ./deepdoc ./deepdoc
ADD ./rag ./rag ADD ./rag ./rag
ADD ./graph ./graph ADD ./agent ./agent
ENV PYTHONPATH=/ragflow/ ENV PYTHONPATH=/ragflow/
ENV HF_ENDPOINT=https://hf-mirror.com ENV HF_ENDPOINT=https://hf-mirror.com

2
Dockerfile.scratch
View File

@ -30,7 +30,7 @@ ADD ./conf ./conf
ADD ./deepdoc ./deepdoc ADD ./deepdoc ./deepdoc
ADD ./rag ./rag ADD ./rag ./rag
ADD ./requirements.txt ./requirements.txt ADD ./requirements.txt ./requirements.txt
ADD ./graph ./graph ADD ./agent ./agent
RUN apt install openmpi-bin openmpi-common libopenmpi-dev RUN apt install openmpi-bin openmpi-common libopenmpi-dev
ENV LD_LIBRARY_PATH /usr/lib/x86_64-linux-gnu/openmpi/lib:$LD_LIBRARY_PATH ENV LD_LIBRARY_PATH /usr/lib/x86_64-linux-gnu/openmpi/lib:$LD_LIBRARY_PATH

2
Dockerfile.scratch.oc9
View File

@ -30,7 +30,7 @@ ADD ./conf ./conf
ADD ./deepdoc ./deepdoc ADD ./deepdoc ./deepdoc
ADD ./rag ./rag ADD ./rag ./rag
ADD ./requirements.txt ./requirements.txt ADD ./requirements.txt ./requirements.txt
ADD ./graph ./graph ADD ./agent ./agent
RUN dnf install -y openmpi openmpi-devel python3-openmpi RUN dnf install -y openmpi openmpi-devel python3-openmpi
ENV C_INCLUDE_PATH /usr/include/openmpi-x86_64:$C_INCLUDE_PATH ENV C_INCLUDE_PATH /usr/include/openmpi-x86_64:$C_INCLUDE_PATH

4
graphrag/leiden.py
View File

@ -25,6 +25,7 @@ from graspologic.partition import hierarchical_leiden
from graspologic.utils import largest_connected_component from graspologic.utils import largest_connected_component
import networkx as nx import networkx as nx
from networkx import is_empty
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
@ -91,13 +92,14 @@ def _compute_leiden_communities(
seed=0xDEADBEEF, seed=0xDEADBEEF,
) -> dict[int, dict[str, int]]: ) -> dict[int, dict[str, int]]:
"""Return Leiden root communities.""" """Return Leiden root communities."""
results: dict[int, dict[str, int]] = {}
if is_empty(graph): return results
if use_lcc: if use_lcc:
graph = stable_largest_connected_component(graph) graph = stable_largest_connected_component(graph)
community_mapping = hierarchical_leiden( community_mapping = hierarchical_leiden(
graph, max_cluster_size=max_cluster_size, random_seed=seed graph, max_cluster_size=max_cluster_size, random_seed=seed
) )
results: dict[int, dict[str, int]] = {}
for partition in community_mapping: for partition in community_mapping:
results[partition.level] = results.get(partition.level, {}) results[partition.level] = results.get(partition.level, {})
results[partition.level][partition.node] = partition.cluster results[partition.level][partition.node] = partition.cluster