mirror of
https://github.com/katanemo/plano.git
synced 2026-06-17 15:25:17 +02:00
47 lines
1.3 KiB
Bash
47 lines
1.3 KiB
Bash
#!/bin/bash
|
|
set -e
|
|
|
|
WAIT_FOR_PIDS=()
|
|
|
|
log() {
|
|
timestamp=$(python3 -c 'from datetime import datetime; print(datetime.now().strftime("%Y-%m-%d %H:%M:%S,%f")[:23])')
|
|
message="$*"
|
|
echo "$timestamp - $message"
|
|
}
|
|
|
|
cleanup() {
|
|
log "Caught signal, terminating all user processes ..."
|
|
for PID in "${WAIT_FOR_PIDS[@]}"; do
|
|
if kill $PID 2> /dev/null; then
|
|
log "killed process: $PID"
|
|
fi
|
|
done
|
|
exit 1
|
|
}
|
|
|
|
trap cleanup EXIT
|
|
|
|
# log "Starting input guards filter on port 10500..."
|
|
# uv run python -m rag_agent --host 0.0.0.0 --port 10500 --agent input_guards &
|
|
# WAIT_FOR_PIDS+=($!)
|
|
|
|
|
|
log "Starting query_parser agent on port 10501..."
|
|
uv run python -m rag_agent --host 0.0.0.0 --port 10501 --agent query_rewriter &
|
|
WAIT_FOR_PIDS+=($!)
|
|
|
|
log "Starting context_builder agent on port 10502..."
|
|
uv run python -m rag_agent --host 0.0.0.0 --port 10502 --agent context_builder &
|
|
WAIT_FOR_PIDS+=($!)
|
|
|
|
# log "Starting response_generator agent on port 10400..."
|
|
# uv run python -m rag_agent --host 0.0.0.0 --port 10400 --agent response_generator &
|
|
# WAIT_FOR_PIDS+=($!)
|
|
|
|
log "Starting response_generator agent on port 10505..."
|
|
uv run python -m rag_agent --rest-server --host 0.0.0.0 --rest-port 10505 --agent response_generator &
|
|
WAIT_FOR_PIDS+=($!)
|
|
|
|
for PID in "${WAIT_FOR_PIDS[@]}"; do
|
|
wait "$PID"
|
|
done
|