own-rag-cli 0.0.1-snapshot → 0.0.2-snapshot

package/chroma_monitor.sh

@@ -25,8 +25,10 @@ BOLD='\033[1m'; DIM='\033[2m'; NC='\033[0m'
 # ---------------------------------------------------------------------------
 # Configuração
 # ---------------------------------------------------------------------------
+CHROMA_SCHEME="http"
 CHROMA_HOST="localhost"
 CHROMA_PORT="8000"
+OWN_RAG_CONFIG_FILE="${OWN_RAG_CLI_CONFIG_FILE:-${HOME}/.own-rag-cli.json}"
 RAG_DB_DIR="${HOME}/.rag_db"
 VENV_DIR="${HOME}/.rag_venv"
 VENV_PYTHON="${VENV_DIR}/bin/python3"
@@ -219,6 +221,57 @@ log_info()  { echo -e "${GREEN}[+]${NC} $*"; }
 log_warn()  { echo -e "${YELLOW}[!]${NC} $*"; }
 log_error() { echo -e "${RED}[$(t err_prefix)]${NC} $*" >&2; }
 
+load_chroma_endpoint_config() {
+    local parsed=""
+    if parsed="$(python3 - "${OWN_RAG_CONFIG_FILE}" <<'PYEOF'
+import json
+import sys
+from pathlib import Path
+from urllib.parse import urlparse
+
+cfg = Path(sys.argv[1]).expanduser()
+scheme = "http"
+host = "localhost"
+port = 8000
+
+if cfg.exists():
+    try:
+        payload = json.loads(cfg.read_text(encoding="utf-8"))
+    except Exception:
+        payload = {}
+    if isinstance(payload, dict):
+        chroma = payload.get("chroma")
+        if isinstance(chroma, dict):
+            raw_scheme = chroma.get("scheme")
+            if isinstance(raw_scheme, str) and raw_scheme.strip().lower() in {"http", "https"}:
+                scheme = raw_scheme.strip().lower()
+            raw_host = chroma.get("host")
+            if isinstance(raw_host, str) and raw_host.strip():
+                candidate = raw_host.strip()
+                parsed = urlparse(candidate if "://" in candidate else f"//{candidate}")
+                if parsed.scheme in {"http", "https"}:
+                    scheme = parsed.scheme
+                if parsed.hostname:
+                    host = parsed.hostname
+                elif "://" not in candidate:
+                    host = candidate.strip().strip("/")
+                if parsed.port and 1 <= parsed.port <= 65535:
+                    port = parsed.port
+            raw_port = chroma.get("port")
+            try:
+                parsed_port = int(str(raw_port).strip())
+                if 1 <= parsed_port <= 65535:
+                    port = parsed_port
+            except Exception:
+                pass
+
+print(f"{scheme}|{host}|{port}")
+PYEOF
+)"; then
+        IFS='|' read -r CHROMA_SCHEME CHROMA_HOST CHROMA_PORT <<< "${parsed}"
+    fi
+}
+
 # ---------------------------------------------------------------------------
 # Funções: venv e dependências
 # ---------------------------------------------------------------------------
@@ -253,10 +306,12 @@ ensure_venv() {
 # ---------------------------------------------------------------------------
 
 is_chroma_running() {
-    "${VENV_PYTHON}" -c "
-import chromadb, sys
+    CHROMA_SCHEME="${CHROMA_SCHEME}" "${VENV_PYTHON}" -c "
+import chromadb, os, sys
+scheme = os.environ.get('CHROMA_SCHEME', 'http').strip().lower()
+ssl = scheme == 'https'
 try:
-    chromadb.HttpClient(host='${CHROMA_HOST}', port=${CHROMA_PORT}).heartbeat()
+    chromadb.HttpClient(host='${CHROMA_HOST}', port=${CHROMA_PORT}, ssl=ssl).heartbeat()
     sys.exit(0)
 except:
     sys.exit(1)
@@ -270,7 +325,7 @@ print_header() {
     printf "  ║          %-44s║\n" "$(t header_title)"
     echo "  ╚══════════════════════════════════════════════════════╝"
     echo -e "${NC}"
-    echo -e "  ${DIM}$(t db_label): ${CHROMA_HOST}:${CHROMA_PORT}  |  $(t data_label): ${RAG_DB_DIR}${NC}"
+    echo -e "  ${DIM}$(t db_label): ${CHROMA_SCHEME}://${CHROMA_HOST}:${CHROMA_PORT}  |  $(t data_label): ${RAG_DB_DIR}${NC}"
     echo ""
 }
 
@@ -281,16 +336,21 @@ cmd_chunks() {
     ensure_venv
 
     echo ""
-    echo -e "${BOLD}${CYAN}  $(t querying_chroma) ${CHROMA_HOST}:${CHROMA_PORT}...${NC}"
+    echo -e "${BOLD}${CYAN}  $(t querying_chroma) ${CHROMA_SCHEME}://${CHROMA_HOST}:${CHROMA_PORT}...${NC}"
     echo ""
 
-    MONITOR_LANG="${UI_LANG}" "${VENV_PYTHON}" - << 'PYEOF'
+    MONITOR_LANG="${UI_LANG}" CHROMA_SCHEME="${CHROMA_SCHEME}" CHROMA_HOST="${CHROMA_HOST}" CHROMA_PORT="${CHROMA_PORT}" "${VENV_PYTHON}" - << 'PYEOF'
 import os
 import sys
 import chromadb
 
-CHROMA_HOST = "localhost"
-CHROMA_PORT = 8000
+CHROMA_SCHEME = os.getenv("CHROMA_SCHEME", "http").strip().lower()
+CHROMA_HOST = os.getenv("CHROMA_HOST", "localhost").strip() or "localhost"
+try:
+    CHROMA_PORT = int(os.getenv("CHROMA_PORT", "8000"))
+except Exception:
+    CHROMA_PORT = 8000
+CHROMA_SSL = CHROMA_SCHEME == "https"
 LANG = os.getenv("MONITOR_LANG", "pt-br").lower()
 IS_EN = LANG == "en-us"
 
@@ -303,7 +363,7 @@ DIM    = "\033[2m"
 NC     = "\033[0m"
 
 try:
-    client = chromadb.HttpClient(host=CHROMA_HOST, port=CHROMA_PORT)
+    client = chromadb.HttpClient(host=CHROMA_HOST, port=CHROMA_PORT, ssl=CHROMA_SSL)
     client.heartbeat()
 except Exception as e:
     if IS_EN:
@@ -374,11 +434,15 @@ cmd_watch() {
     LAST_COUNT=-1
 
     while true; do
-        RESULT=$("${VENV_PYTHON}" - 2>/dev/null << 'PYEOF'
-import chromadb, sys
+        RESULT=$(CHROMA_SCHEME="${CHROMA_SCHEME}" CHROMA_HOST="${CHROMA_HOST}" CHROMA_PORT="${CHROMA_PORT}" "${VENV_PYTHON}" - 2>/dev/null << 'PYEOF'
+import chromadb, os, sys
 
 try:
-    client = chromadb.HttpClient(host="localhost", port=8000)
+    scheme = os.getenv("CHROMA_SCHEME", "http").strip().lower()
+    ssl = scheme == "https"
+    host = os.getenv("CHROMA_HOST", "localhost").strip() or "localhost"
+    port = int(os.getenv("CHROMA_PORT", "8000"))
+    client = chromadb.HttpClient(host=host, port=port, ssl=ssl)
     client.heartbeat()
     cols = client.list_collections()
     total = sum(c.count() for c in cols)
@@ -669,7 +733,7 @@ cmd_full() {
 
         # Coleções e chunks
         echo -e "  ${BOLD}$(t collections_chunks)${NC}"
-        MONITOR_LANG="${UI_LANG}" "${VENV_PYTHON}" - 2>/dev/null << 'PYEOF'
+        MONITOR_LANG="${UI_LANG}" CHROMA_SCHEME="${CHROMA_SCHEME}" CHROMA_HOST="${CHROMA_HOST}" CHROMA_PORT="${CHROMA_PORT}" "${VENV_PYTHON}" - 2>/dev/null << 'PYEOF'
 import chromadb, os, sys
 
 GREEN  = "\033[0;32m"; CYAN   = "\033[0;36m"
@@ -679,7 +743,11 @@ LANG = os.getenv("MONITOR_LANG", "pt-br").lower()
 IS_EN = LANG == "en-us"
 
 try:
-    client = chromadb.HttpClient(host="localhost", port=8000)
+    scheme = os.getenv("CHROMA_SCHEME", "http").strip().lower()
+    ssl = scheme == "https"
+    host = os.getenv("CHROMA_HOST", "localhost").strip() or "localhost"
+    port = int(os.getenv("CHROMA_PORT", "8000"))
+    client = chromadb.HttpClient(host=host, port=port, ssl=ssl)
     client.heartbeat()
 except Exception as e:
     msg = "ChromaDB unreachable" if IS_EN else "ChromaDB inacessível"
@@ -759,18 +827,23 @@ cmd_reset() {
         exit 0
     fi
 
-    MONITOR_LANG="${UI_LANG}" "${VENV_PYTHON}" - << 'PYEOF'
+    MONITOR_LANG="${UI_LANG}" CHROMA_SCHEME="${CHROMA_SCHEME}" CHROMA_HOST="${CHROMA_HOST}" CHROMA_PORT="${CHROMA_PORT}" "${VENV_PYTHON}" - << 'PYEOF'
 import os
 import sys
 import chromadb
 
-CHROMA_HOST = "localhost"
-CHROMA_PORT = 8000
+CHROMA_SCHEME = os.getenv("CHROMA_SCHEME", "http").strip().lower()
+CHROMA_HOST = os.getenv("CHROMA_HOST", "localhost").strip() or "localhost"
+try:
+    CHROMA_PORT = int(os.getenv("CHROMA_PORT", "8000"))
+except Exception:
+    CHROMA_PORT = 8000
+CHROMA_SSL = CHROMA_SCHEME == "https"
 LANG = os.getenv("MONITOR_LANG", "pt-br").lower()
 IS_EN = LANG == "en-us"
 
 try:
-    client = chromadb.HttpClient(host=CHROMA_HOST, port=CHROMA_PORT)
+    client = chromadb.HttpClient(host=CHROMA_HOST, port=CHROMA_PORT, ssl=CHROMA_SSL)
     client.heartbeat()
 except Exception as e:
     if IS_EN:
@@ -839,6 +912,7 @@ cmd_menu() {
 # Dispatcher
 # ---------------------------------------------------------------------------
 select_ui_language
+load_chroma_endpoint_config
 case "${1:-menu}" in
     chunks)  cmd_chunks ;;
     watch)   cmd_watch  ;;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "own-rag-cli",
-  "version": "0.0.1-snapshot",
+  "version": "0.0.2-snapshot",
   "description": "Local RAG setup with ChromaDB + MCP server (Jina/BGE hybrid support).",
   "license": "MIT",
   "private": false,