61 lines
2.0 KiB
Bash
Executable File
61 lines
2.0 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Script to configure Ollama settings for Docker Compose or external server
|
|
|
|
echo "Ollama Configuration Helper"
|
|
echo "============================"
|
|
echo ""
|
|
echo "Choose your Ollama setup:"
|
|
echo "1) Docker Compose (Ollama runs in container)"
|
|
echo "2) External Server (Ollama runs on host machine)"
|
|
echo ""
|
|
read -p "Enter choice [1-2]: " choice
|
|
|
|
ENV_FILE="backend/.env"
|
|
|
|
if [ ! -f "$ENV_FILE" ]; then
|
|
echo "Error: $ENV_FILE not found!"
|
|
exit 1
|
|
fi
|
|
|
|
case $choice in
|
|
1)
|
|
echo "Configuring for Docker Compose..."
|
|
# Update OLLAMA_BASE_URL to use internal Docker network
|
|
if grep -q "OLLAMA_BASE_URL=" "$ENV_FILE"; then
|
|
sed -i.bak 's|OLLAMA_BASE_URL=.*|OLLAMA_BASE_URL=http://ollama:11434|' "$ENV_FILE"
|
|
else
|
|
echo "OLLAMA_BASE_URL=http://ollama:11434" >> "$ENV_FILE"
|
|
fi
|
|
echo "✓ Updated OLLAMA_BASE_URL to http://ollama:11434"
|
|
echo ""
|
|
echo "Next steps:"
|
|
echo "1. Start services: docker-compose up -d"
|
|
echo "2. Wait for model download: docker-compose logs -f ollama-setup"
|
|
echo "3. Test: docker-compose exec crawler python crawler_service.py 1"
|
|
;;
|
|
2)
|
|
echo "Configuring for external Ollama server..."
|
|
# Update OLLAMA_BASE_URL to use host machine
|
|
if grep -q "OLLAMA_BASE_URL=" "$ENV_FILE"; then
|
|
sed -i.bak 's|OLLAMA_BASE_URL=.*|OLLAMA_BASE_URL=http://host.docker.internal:11434|' "$ENV_FILE"
|
|
else
|
|
echo "OLLAMA_BASE_URL=http://host.docker.internal:11434" >> "$ENV_FILE"
|
|
fi
|
|
echo "✓ Updated OLLAMA_BASE_URL to http://host.docker.internal:11434"
|
|
echo ""
|
|
echo "Next steps:"
|
|
echo "1. Install Ollama: https://ollama.ai/download"
|
|
echo "2. Pull model: ollama pull phi3:latest"
|
|
echo "3. Start Ollama: ollama serve"
|
|
echo "4. Start services: docker-compose up -d"
|
|
;;
|
|
*)
|
|
echo "Invalid choice!"
|
|
exit 1
|
|
;;
|
|
esac
|
|
|
|
echo ""
|
|
echo "Configuration complete!"
|