ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter3/retrieval-pipeline/start_all_services.sh

@@ -0,0 +1,125 @@
+#!/bin/bash
+
+# Start all services for the retrieval pipeline
+
+echo "========================================="
+echo "Starting Retrieval Pipeline Services"
+echo "========================================="
+
+# Colors for output
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+NC='\033[0m' # No Color
+
+# Function to check if port is in use
+check_port() {
+    if lsof -Pi :$1 -sTCP:LISTEN -t >/dev/null ; then
+        return 0
+    else
+        return 1
+    fi
+}
+
+# Kill existing services on ports
+echo -e "${YELLOW}Checking for existing services...${NC}"
+for port in 4240 4241 4242; do
+    if check_port $port; then
+        echo -e "${YELLOW}Killing existing service on port $port${NC}"
+        lsof -ti:$port | xargs kill -9 2>/dev/null
+        sleep 1
+    fi
+done
+
+# Start dense embedding service
+echo -e "\n${GREEN}Starting Dense Embedding Service (port 8000)...${NC}"
+cd ../dense-embedding
+python main.py --port 8000 > dense.log 2>&1 &
+DENSE_PID=$!
+echo "Dense service PID: $DENSE_PID"
+
+# Wait for dense service to start
+echo "Waiting for dense service to initialize..."
+for i in {1..30}; do
+    if check_port 8000; then
+        echo -e "${GREEN}✓ Dense service ready${NC}"
+        break
+    fi
+    sleep 1
+done
+
+# Start sparse embedding service
+echo -e "\n${GREEN}Starting Sparse Embedding Service (port 8001)...${NC}"
+cd ../sparse-embedding
+python server.py --port 8001 > sparse.log 2>&1 &
+SPARSE_PID=$!
+echo "Sparse service PID: $SPARSE_PID"
+
+# Wait for sparse service to start
+echo "Waiting for sparse service to initialize..."
+for i in {1..30}; do
+    if check_port 8001; then
+        echo -e "${GREEN}✓ Sparse service ready${NC}"
+        break
+    fi
+    sleep 1
+done
+
+# Start retrieval pipeline
+echo -e "\n${GREEN}Starting Retrieval Pipeline (port 4242)...${NC}"
+cd ../retrieval-pipeline
+python main.py --port 4242 > pipeline.log 2>&1 &
+PIPELINE_PID=$!
+echo "Pipeline service PID: $PIPELINE_PID"
+
+# Wait for pipeline to start
+echo "Waiting for pipeline to initialize (loading reranker model)..."
+for i in {1..60}; do
+    if check_port 4242; then
+        echo -e "${GREEN}✓ Pipeline ready${NC}"
+        break
+    fi
+    sleep 1
+done
+
+# Check all services
+echo -e "\n========================================="
+echo "Service Status:"
+echo "========================================="
+
+if check_port 8000; then
+    echo -e "${GREEN}✓ Dense Embedding Service: http://localhost:8000${NC}"
+else
+    echo -e "${RED}✗ Dense Embedding Service failed to start${NC}"
+fi
+
+if check_port 8001; then
+    echo -e "${GREEN}✓ Sparse Embedding Service: http://localhost:8001${NC}"
+else
+    echo -e "${RED}✗ Sparse Embedding Service failed to start${NC}"
+fi
+
+if check_port 4242; then
+    echo -e "${GREEN}✓ Retrieval Pipeline: http://localhost:4242${NC}"
+    echo -e "${GREEN}✓ API Documentation: http://localhost:4242/docs${NC}"
+else
+    echo -e "${RED}✗ Retrieval Pipeline failed to start${NC}"
+fi
+
+echo -e "\n========================================="
+echo "All services started!"
+echo "========================================="
+echo ""
+echo "To test the pipeline:"
+echo "  python test_client.py"
+echo ""
+echo "To run the demo:"
+echo "  python demo.py"
+echo ""
+echo "To stop all services:"
+echo "  ./stop_all_services.sh"
+echo ""
+echo "Service logs:"
+echo "  Dense: dense.log"
+echo "  Sparse: sparse.log"
+echo "  Pipeline: pipeline.log"