add dockerfile and fix trival bugs (#78)

This commit is contained in:
KevinHuSh
2024-02-28 15:01:12 +08:00
committed by GitHub
parent eb517f3106
commit 39269d2f79
14 changed files with 243 additions and 52 deletions

24
docker/entrypoint.sh Normal file
View File

@@ -0,0 +1,24 @@
#!/bin/bash
python rag/svr/task_broker.py &
function task_exe(){
while [ 1 -eq 1 ];do mpirun -n 2 python rag/svr/task_executor.py ; done
}
function watch_broker(){
while [ 1 -eq 1];do
C=`ps aux|grep "task_broker.py"|grep -v grep|wc -l`;
if [ $C -lt 1 ];then
python rag/svr/task_broker.py &
fi
sleep 5;
done
}
task_exe &
sleep 10;
watch_broker &
python api/ragflow_server.py