Upload setup_endpoints.sh

setup_endpoints.sh

@@ -1,145 +1,88 @@
 #!/bin/bash
-set -euo pipefail
 
-BASE_URL="${BASE_URL:-http://localhost:8080}"
-ADMIN_EMAIL="${WEBUI_ADMIN_EMAIL:-}"
-ADMIN_PASSWORD="${WEBUI_ADMIN_PASSWORD:-}"
-ADMIN_NAME="${WEBUI_ADMIN_NAME:-admin}"
-SYSTEM_PROMPT="${DEFAULT_SYSTEM_PROMPT:-}"
-export BASE_URL
-export WEBUI_ADMIN_EMAIL="$ADMIN_EMAIL"
-export WEBUI_ADMIN_PASSWORD="$ADMIN_PASSWORD"
-export WEBUI_ADMIN_NAME="$ADMIN_NAME"
-export DEFAULT_SYSTEM_PROMPT="$SYSTEM_PROMPT"
+BASE_URL="http://localhost:8080"
 
-echo "Cekam da WebUI bude spreman..."
+echo "Čekam da WebUI bude spreman..."
 for i in {1..30}; do
-  STATUS=$(curl -s -o /dev/null -w "%{http_code}" "$BASE_URL/health" || true)
+  STATUS=$(curl -s -o /dev/null -w "%{http_code}" "$BASE_URL/health")
   if [ "$STATUS" = "200" ]; then
-    echo "WebUI je spreman."
+    echo "✅ WebUI je spreman!"
     break
   fi
+  echo "Pokušaj $i — čekam još 5s..."
   sleep 5
 done
 
-if [ -z "$ADMIN_EMAIL" ] || [ -z "$ADMIN_PASSWORD" ]; then
-  echo "WEBUI_ADMIN_EMAIL i WEBUI_ADMIN_PASSWORD nisu postavljeni; preskacem automatsku konfiguraciju."
-  exit 0
-fi
-
-python3 - <<'PY'
-import json
-import os
-import sys
-import urllib.error
-import urllib.request
-
-base_url = os.environ.get("BASE_URL", "http://localhost:8080")
-admin_email = os.environ["WEBUI_ADMIN_EMAIL"]
-admin_password = os.environ["WEBUI_ADMIN_PASSWORD"]
-admin_name = os.environ.get("WEBUI_ADMIN_NAME", "admin")
-system_prompt = os.environ.get("DEFAULT_SYSTEM_PROMPT", "")
-
-def request_json(path, payload, token=None):
-    data = json.dumps(payload).encode("utf-8")
-    headers = {"Content-Type": "application/json"}
-    if token:
-        headers["Authorization"] = f"Bearer {token}"
-    req = urllib.request.Request(f"{base_url}{path}", data=data, headers=headers, method="POST")
-    with urllib.request.urlopen(req, timeout=60) as resp:
-        body = resp.read().decode("utf-8")
-        return json.loads(body) if body else {}
-
-def login_or_signup():
-    try:
-        resp = request_json(
-            "/api/v1/auths/signin",
-            {"email": admin_email, "password": admin_password},
-        )
-        token = resp.get("token")
-        if token:
-            return token
-    except Exception:
-        pass
-
-    resp = request_json(
-        "/api/v1/auths/signup",
-        {"name": admin_name, "email": admin_email, "password": admin_password},
-    )
-    token = resp.get("token")
-    if not token:
-        raise RuntimeError("Nije moguce dobiti token za admin nalog.")
-    return token
+# Login
+echo "Logujem se kao admin..."
+LOGIN_RESPONSE=$(curl -s -X POST "$BASE_URL/api/v1/auths/signin" \
+  -H "Content-Type: application/json" \
+  -d '{"email": "nermindurma1981@gmail.com", "password": "nerko1508"}')
 
-def build_provider_config():
-    openai_base = os.environ.get("OPENAI_API_BASE_URL", "https://api.openai.com/v1")
-    providers = [
-        (openai_base, os.environ.get("OPENAI_API_KEY", ""), False),
-        ("https://api.anthropic.com/v1", os.environ.get("ANTHROPIC_API_KEY", ""), False),
-        ("https://generativelanguage.googleapis.com/v1beta/openai", os.environ.get("GOOGLE_API_KEY", ""), False),
-        ("https://models.github.ai/inference", os.environ.get("GITHUB_TOKEN", ""), False),
-        ("https://api.groq.com/openai/v1", os.environ.get("GROQ_API_KEY", ""), False),
-        ("https://api.mistral.ai/v1", os.environ.get("MISTRAL_API_KEY", ""), False),
-        ("https://api.deepseek.com/v1", os.environ.get("DEEPSEEK_API_KEY", ""), False),
-        ("https://api.together.xyz/v1", os.environ.get("TOGETHER_API_KEY", ""), False),
-        ("https://integrate.api.nvidia.com/v1", os.environ.get("NVIDIA_API_KEY", ""), False),
-        ("https://api.sambanova.ai/v1", os.environ.get("SAMBANOVA_API_KEY", ""), False),
-        ("https://openrouter.ai/api/v1", os.environ.get("OPENROUTER_API_KEY", ""), False),
-        ("https://gen.pollinations.ai/v1", os.environ.get("POLLINATIONS_API_KEY", ""), True),
-        ("https://api.cerebras.ai/v1", os.environ.get("CEREBRAS_API_KEY", ""), False),
-        ("https://api.fireworks.ai/inference/v1", os.environ.get("FIREWORKS_API_KEY", ""), False),
-        ("https://api.x.ai/v1", os.environ.get("XAI_API_KEY", ""), False),
-        ("https://api.cohere.ai/compatibility/v1", os.environ.get("COHERE_API_KEY", ""), False),
-        ("https://api.tavily.com", os.environ.get("TAVILY_API_KEY", ""), False),
-        ("https://router.huggingface.co/v1", os.environ.get("HF_INFERENCE_TOKEN", ""), False),
-    ]
+TOKEN=$(echo $LOGIN_RESPONSE | python3 -c "import sys, json; print(json.load(sys.stdin).get('token', ''))")
 
-    urls = []
-    keys = []
-    for url, key, allow_empty in providers:
-        if url and (key or allow_empty):
-            urls.append(url)
-            keys.append(key)
-
-    cf_account = os.environ.get("CLOUDFLARE_ACCOUNT_ID", "")
-    cf_token = os.environ.get("CLOUDFLARE_API_TOKEN", "")
-    if cf_account and cf_token:
-        urls.append(f"https://api.cloudflare.com/client/v4/accounts/{cf_account}/ai/v1")
-        keys.append(cf_token)
-
-    return urls, keys
-
-token = login_or_signup()
-
-if system_prompt.strip():
-    try:
-        request_json(
-            "/api/v1/configs/default/system/prompt",
-            {"DEFAULT_SYSTEM_PROMPT": system_prompt},
-            token=token,
-        )
-        print("System prompt postavljen.")
-    except Exception as exc:
-        print(f"System prompt nije postavljen: {exc}")
+if [ -z "$TOKEN" ]; then
+  echo "❌ Login neuspješan, pokušavam registraciju..."
+  REGISTER_RESPONSE=$(curl -s -X POST "$BASE_URL/api/v1/auths/signup" \
+    -H "Content-Type: application/json" \
+    -d '{"name": "nermin", "email": "nermindurma1981@gmail.com", "password": "nerko1508"}')
+  TOKEN=$(echo $REGISTER_RESPONSE | python3 -c "import sys, json; print(json.load(sys.stdin).get('token', ''))")
+fi
 
-urls, keys = build_provider_config()
-if urls and keys:
-    try:
-        request_json(
-            "/openai/config/update",
-            {
-                "ENABLE_OPENAI_API": True,
-                "OPENAI_API_BASE_URLS": urls,
-                "OPENAI_API_KEYS": keys,
-                "OPENAI_API_CONFIGS": {},
-            },
-            token=token,
-        )
-        print("Endpointi konfigurirani.")
-    except Exception as exc:
-        print(f"Konfiguracija endpointa nije uspjela: {exc}")
-else:
-    print("Nema postavljenih provider secret-a; preskacem openai endpoint konfiguraciju.")
+if [ -z "$TOKEN" ]; then
+  echo "❌ Nije moguće dobiti token."
+  exit 0
+fi
 
-print("Automatska konfiguracija zavrsena.")
-PY
+echo "✅ Ulogovan, konfiguriram endpointe kroz /openai/config/update..."
+
+curl -s -X POST "$BASE_URL/openai/config/update" \
+  -H "Content-Type: application/json" \
+  -H "Authorization: Bearer $TOKEN" \
+  -d "{
+    \"ENABLE_OPENAI_API\": true,
+    \"OPENAI_API_BASE_URLS\": [
+      \"https://api.openai.com/v1\",
+      \"https://api.anthropic.com/v1\",
+      \"https://generativelanguage.googleapis.com/v1beta/openai\",
+      \"https://models.inference.ai.azure.com\",
+      \"https://api.groq.com/openai/v1\",
+      \"https://api.mistral.ai/v1\",
+      \"https://api.deepseek.com/v1\",
+      \"https://api.together.xyz/v1\",
+      \"https://integrate.api.nvidia.com/v1\",
+      \"https://api.sambanova.ai/v1\",
+      \"https://openrouter.ai/api/v1\",
+      \"https://api.cloudflare.com/client/v4/accounts/$CLOUDFLARE_ACCOUNT_ID/ai/v1\",
+      \"https://api.cerebras.ai/v1\",
+      \"https://api.fireworks.ai/inference/v1\",
+      \"https://api.x.ai/v1\",
+      \"https://api-inference.huggingface.co/v1\",
+      \"https://api.cohere.ai/compatibility/v1\",
+      \"https://gen.pollinations.ai\"
+    ],
+    \"OPENAI_API_KEYS\": [
+      \"$OPENAI_API_KEY\",
+      \"$ANTHROPIC_API_KEY\",
+      \"$GOOGLE_API_KEY\",
+      \"$GITHUB_TOKEN\",
+      \"$GROQ_API_KEY\",
+      \"$MISTRAL_API_KEY\",
+      \"$DEEPSEEK_API_KEY\",
+      \"$TOGETHER_API_KEY\",
+      \"$NVIDIA_API_KEY\",
+      \"$SAMBANOVA_API_KEY\",
+      \"$OPENROUTER_API_KEY\",
+      \"$CLOUDFLARE_API_TOKEN\",
+      \"$CEREBRAS_API_KEY\",
+      \"$FIREWORKS_API_KEY\",
+      \"$XAI_API_KEY\",
+      \"$HUGGINGFACE_API_KEY\",
+      \"$COHERE_API_KEY\",
+      \"$POLLINATIONS_API_KEY\"
+    ],
+    \"OPENAI_API_CONFIGS\": {}
+  }"
+
+echo ""
+echo "✅ Svi endpointi su konfigurirani!"