LOT 1 - Optimizare script build -Instalare mono comanda
This commit is contained in:
parent
5380c3fc63
commit
42ff22bf85
127 changed files with 16163 additions and 532 deletions
49
ai_platform/local_gpu_stack/seed/seed_brain.sh
Normal file
49
ai_platform/local_gpu_stack/seed/seed_brain.sh
Normal file
|
|
@ -0,0 +1,49 @@
|
|||
#!/usr/bin/env bash
|
||||
# Brain stack helpers: render env, align postgres password, create atomic token,
|
||||
# seed the canonical taxonomy. Idempotent. Requires lib.sh sourced + master .env
|
||||
# loaded ($POSTGRES_PASSWORD, $ATOMIC_TOKEN, $HOST_IP, $ENV_FILE, $BRAIN_DIR).
|
||||
|
||||
# Write didi_brain/.env (read by brain-api/scheduler) + infra/.env (compose interpolation)
|
||||
brain_write_env() {
|
||||
local pw=$1 tok=$2
|
||||
cp "${BRAIN_DIR}/.env.example" "${BRAIN_DIR}/.env"
|
||||
sed -i \
|
||||
-e "s#^LLM_ROUTER_URL=.*#LLM_ROUTER_URL=http://didiAI-llm-api:14011#" \
|
||||
-e "s#^LLM_VLLM_URL=.*#LLM_VLLM_URL=http://didiAI-vllm-qwen3.5:14001#" \
|
||||
-e "s#^EMBEDDING_URL=.*#EMBEDDING_URL=http://didiAI-embeddings-api:14100#" \
|
||||
-e "s#^RERANKER_URL=.*#RERANKER_URL=http://didiAI-rerank-api:14200#" \
|
||||
-e "s#^MODEL_FAST_ENABLED=.*#MODEL_FAST_ENABLED=true#" \
|
||||
-e "s#^POSTGRES_PASSWORD=.*#POSTGRES_PASSWORD=${pw}#" \
|
||||
"${BRAIN_DIR}/.env"
|
||||
if grep -q '^ATOMIC_TOKEN=' "${BRAIN_DIR}/.env"; then
|
||||
sed -i "s#^ATOMIC_TOKEN=.*#ATOMIC_TOKEN=${tok}#" "${BRAIN_DIR}/.env"
|
||||
else
|
||||
echo "ATOMIC_TOKEN=${tok}" >> "${BRAIN_DIR}/.env"
|
||||
fi
|
||||
printf 'POSTGRES_USER=atomic\nPOSTGRES_PASSWORD=%s\nPOSTGRES_DB=atomic\n' "$pw" > "${BRAIN_DIR}/infra/.env"
|
||||
}
|
||||
|
||||
# Force the postgres role password to match (volume may have been initialized with a different one)
|
||||
brain_align_pg() {
|
||||
local pw=$1
|
||||
docker exec didibrain-postgres psql -U atomic -d atomic \
|
||||
-c "ALTER USER atomic WITH PASSWORD '${pw}';" >/dev/null 2>&1 || true
|
||||
}
|
||||
|
||||
# Create an atomic API token if the master .env doesn't have one yet. Echos the token.
|
||||
brain_ensure_token() {
|
||||
local pw=$1 cur=$2
|
||||
if [ -n "$cur" ]; then printf '%s' "$cur"; return 0; fi
|
||||
local out tok
|
||||
out=$(docker exec didibrain-atomic atomic-server token create --name didibrain \
|
||||
--storage postgres --database-url "postgres://atomic:${pw}@postgres:5432/atomic" 2>&1)
|
||||
tok=$(printf '%s' "$out" | grep -oE 'at_[A-Za-z0-9_.-]{20,}' | head -1)
|
||||
printf '%s' "$tok"
|
||||
}
|
||||
|
||||
# Seed canonical taxonomy into atomic (idempotent; tags created if missing).
|
||||
brain_seed_taxonomy() {
|
||||
docker cp "${BRAIN_DIR}/scripts" didibrain-api:/app/scripts >/dev/null 2>&1 || true
|
||||
# local _tag_ids.json write is read-only in the image — harmless, atomic gets the tags
|
||||
docker exec -e PYTHONPATH=/app didibrain-api python /app/scripts/04_seed_taxonomy.py >/dev/null 2>&1 || true
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue