|
| 1 | +set_model_id() { |
| 2 | + if [ -z "$1" ] && [ -z "$2" ]; then |
| 3 | + yq -o json '.| select(.data | has("MODEL_ID"))| {"ConfigMap": .metadata.name, "MODEL_ID": .data.MODEL_ID}' *.yaml |
| 4 | + echo "usage:" |
| 5 | + echo " set_model_id \${ConfigMap} \${MODEL_ID}" |
| 6 | + return |
| 7 | + fi |
| 8 | + conf=$1 |
| 9 | + file=${1%%-*} |
| 10 | + sed -i '/name: '"${conf}"'/,/---/s|\(MODEL_ID:\).*|\1 "'"${2}"'"|' ${file}.yaml |
| 11 | +} |
| 12 | + |
| 13 | +set_model_mirror() { |
| 14 | + if [ -z "$1" ] ; then |
| 15 | + yq -o json '.| select(.data | has("MODEL_ID"))| {"ConfigMap": .metadata.name, "MODEL_MIRROR": .data.HF_ENDPOINT}' *.yaml |
| 16 | + echo "usage:" |
| 17 | + echo " set_model_mirror \${MODEL_MIRROR}" |
| 18 | + return |
| 19 | + fi |
| 20 | + cm=$(yq -r -o json '.| select(.data | has("MODEL_ID"))| .metadata.name' *.yaml) |
| 21 | + mirror=$1 |
| 22 | + for i in $cm; do |
| 23 | + conf=$i |
| 24 | + file=${i%%-*} |
| 25 | + echo "ConfigMap: $conf set mirror as $mirror" |
| 26 | + has_mirror=$(yq -r -o json '.| select(.metadata.name == "'"${conf}"'")| .data.HF_ENDPOINT' ${file}.yaml) |
| 27 | + if [ "$has_mirror" == "null" ]; then |
| 28 | + sed -i '/name: '"${conf}"'/,/---/s|\(data:\)|\1\n HF_ENDPOINT: "'"${mirror}"'"|' ${file}.yaml |
| 29 | + else |
| 30 | + sed -i '/name: '"${conf}"'/,/---/s|\(HF_ENDPOINT:\).*|\1 "'"${1}"'"|' ${file}.yaml |
| 31 | + fi |
| 32 | + done |
| 33 | +} |
| 34 | + |
| 35 | +set_hf_token() { |
| 36 | + if [ -z "$1" ] ; then |
| 37 | + echo "usage:" |
| 38 | + echo " set_hf_token \${HF_TOKEN}" |
| 39 | + return |
| 40 | + fi |
| 41 | + sed -i "s/\(HF_TOKEN:\).*/\1 \"${1}\"/g" *.yaml |
| 42 | + sed -i "s/\(HUGGINGFACEHUB_API_TOKEN:\).*/\1 \"${1}\"/g" *.yaml |
| 43 | + sed -i "s/\(HUGGING_FACE_HUB_TOKEN:\).*/\1 \"${1}\"/g" *.yaml |
| 44 | +} |
| 45 | + |
| 46 | +set_https_proxy() { |
| 47 | + if [ -z "$1" ] ; then |
| 48 | + echo "usage:" |
| 49 | + echo " set_https_proxy \${https_proxy}" |
| 50 | + return |
| 51 | + fi |
| 52 | + https_proxy=$1 |
| 53 | + sed -i -e "s|\(https_proxy:\)\s*\"\"|\1 \"$https_proxy\"|g" *.yaml |
| 54 | + sed -i '/https_proxy/{n;s|\(value:\)\s.*""|\1 "'"$https_proxy"'"|g}' *.yaml |
| 55 | +} |
| 56 | + |
| 57 | +set_http_proxy() { |
| 58 | + if [ -z "$1" ] ; then |
| 59 | + echo "usage:" |
| 60 | + echo " set_http_proxy \${http_proxy}" |
| 61 | + return |
| 62 | + fi |
| 63 | + http_proxy=$1 |
| 64 | + sed -i -e "s|\(http_proxy:\)\s*\"\"|\1 \"$http_proxy\"|g" *.yaml |
| 65 | + sed -i '/http_proxy/{n;s|\(value:\)\s.*""|\1 "'"$http_proxy"'"|g}' *.yaml |
| 66 | +} |
| 67 | + |
| 68 | +set_no_proxy() { |
| 69 | + if [ -z "$1" ] ; then |
| 70 | + echo "usage:" |
| 71 | + echo " set_no_proxy \${no_proxy}" |
| 72 | + return |
| 73 | + fi |
| 74 | + no_proxy=$1 |
| 75 | + sed -i -e "s|\(no_proxy:\)\s*\"\"|\1 \"$no_proxy\"|g" *.yaml |
| 76 | + sed -i '/no_proxy/{n;s|\(value:\)\s.*""|\1 "'"$no_proxy"'"|g}' *.yaml |
| 77 | +} |
| 78 | + |
| 79 | +set_backend_service_endpoint() { |
| 80 | + for i in $(grep -oP "(?<=APP_BACKEND_SERVICE_ENDPOINT_).*" *.yaml); do |
| 81 | + echo $i |
| 82 | + name=${i##*:} |
| 83 | + file=${name,,}.yaml |
| 84 | + svc=$(yq -o json '. | select(.metadata.name == "'"${name,,}"'" and .kind=="Service")' $file) |
| 85 | + port=$(jq .spec.ports[0].port <<< $svc) |
| 86 | + |
| 87 | + url=http://${name,,}.${ns:-default}.svc.cluster.local:${port} |
| 88 | + echo $url |
| 89 | + sed -i -e '/APP_BACKEND_SERVICE_ENDPOINT_'"$name"'/{n;s|\(value:\)\s.*|\1 "'"$url"'"|}' productivity_suite_reactui.yaml |
| 90 | + done |
| 91 | +} |
| 92 | + |
| 93 | + |
| 94 | +set_dataprep_service_endpoint() { |
| 95 | + name=chatqna-data-prep |
| 96 | + file=chatqna.yaml |
| 97 | + svc=$(yq -o json '. | select(.metadata.name == "'"$name"'" and .kind=="Service")' $file) |
| 98 | + port=$(jq .spec.ports[0].port <<< $svc) |
| 99 | + url=http://${name}.${ns:-default}.svc.cluster.local:${port} |
| 100 | + echo $url |
| 101 | + for i in $(grep -oP "(?<=APP_)DATAPREP.*(?=_ENDPOINT)" *.yaml); do |
| 102 | + echo $i |
| 103 | + curd=${i##*:}; |
| 104 | + sed -i -e '/'"$curd"'/{n;s|\(value:\)\s.*|\1 "'"$url"'"|}' productivity_suite_reactui.yaml; |
| 105 | + done |
| 106 | +} |
| 107 | + |
| 108 | + |
| 109 | +set_chat_history_endpoint() { |
| 110 | + for i in $(grep -oP "(?<=APP_)CHAT_HISTORY.*(?=_ENDPOINT)" *.yaml); do |
| 111 | + echo $i; |
| 112 | + curd=${i##*:}; |
| 113 | + name=${curd%_*}; |
| 114 | + file=${name,,}.yaml; |
| 115 | + name=${name/_/-}; |
| 116 | + svc=$(yq -o json '. | select(.metadata.name == "'"${name,,}"'" and .kind=="Service")' $file) |
| 117 | + port=$(jq .spec.ports[0].port <<< $svc) |
| 118 | + url=http://${name,,}.${ns:-default}.svc.cluster.local:${port}; |
| 119 | + echo $url; |
| 120 | + sed -i -e '/'"$curd"'/{n;s|\(value:\)\s.*|\1 "'"$url"'"|}' productivity_suite_reactui.yaml; |
| 121 | + done |
| 122 | +} |
| 123 | + |
| 124 | + |
| 125 | +set_prompt_service_endpoint() { |
| 126 | + for i in $(grep -oP "(?<=APP_)PROMPT_SERVICE.*(?=_ENDPOINT)" *.yaml); do |
| 127 | + echo $i; |
| 128 | + curd=${i##*:}; |
| 129 | + curdr=${curd/SERVICE/REGISTRY}; |
| 130 | + name=${curdr%_*}; |
| 131 | + file=${name,,}.yaml; |
| 132 | + name=${name/_/-}; |
| 133 | + svc=$(yq -o json '. | select(.metadata.name == "'"${name,,}"'" and .kind=="Service")' $file) |
| 134 | + port=$(jq .spec.ports[0].port <<< $svc) |
| 135 | + url=http://${name,,}.${ns:-default}.svc.cluster.local:${port}; |
| 136 | + echo $url; |
| 137 | + sed -i -e '/'"$curd"'/{n;s|\(value:\)\s.*|\1 "'"$url"'"|}' productivity_suite_reactui.yaml ; |
| 138 | + done |
| 139 | +} |
| 140 | + |
| 141 | +set_keycloak_service_endpoint() { |
| 142 | + name=keycloak |
| 143 | + file=keycloak_install.yaml |
| 144 | + svc=$(yq -o json '. | select(.metadata.name == "'"$name"'" and .kind=="Service")' $file) |
| 145 | + port=$(jq .spec.ports[0].port <<< $svc) |
| 146 | + url=http://${name}.${ns:-default}.svc.cluster.local:${port} |
| 147 | + echo $url |
| 148 | + sed -i -e '/APP_KEYCLOAK_SERVICE_ENDPOINT/{n;s|\(value:\)\s.*|\1 "'"$url"'"|}' productivity_suite_reactui.yaml |
| 149 | +} |
| 150 | + |
| 151 | +set_services_endpoint() { |
| 152 | + set_backend_service_endpoint |
| 153 | + set_keycloak_service_endpoint |
| 154 | + set_chat_history_endpoint |
| 155 | + set_prompt_service_endpoint |
| 156 | + set_dataprep_service_endpoint |
| 157 | +} |
0 commit comments