rag-sentinel 0.1.3__tar.gz → 0.1.4__tar.gz

{rag_sentinel-0.1.3/src/rag_sentinel.egg-info → rag_sentinel-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rag-sentinel
-Version: 0.1.3
+Version: 0.1.4
 Summary: RAG Evaluation Framework using Ragas metrics and MLflow tracking
 Author: RAGSentinel Team
 License: MIT

{rag_sentinel-0.1.3 → rag_sentinel-0.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "rag-sentinel"
-version = "0.1.3"
+version = "0.1.4"
 description = "RAG Evaluation Framework using Ragas metrics and MLflow tracking"
 readme = "README.md"
 license = {text = "MIT"}

{rag_sentinel-0.1.3 → rag_sentinel-0.1.4}/src/rag_sentinel/cli.py

@@ -20,7 +20,9 @@ import socket
 import subprocess
 import time
 import argparse
+import configparser
 from pathlib import Path
+from urllib.parse import urlparse
 
 
 # =============================================================================
@@ -34,6 +36,39 @@ TEMPLATES_DIR = Path(__file__).parent / "templates"
 # Helper Functions
 # =============================================================================
 
+def get_mlflow_host_port():
+    """
+    Read MLflow tracking_uri from config.ini and parse host and port.
+
+    Returns:
+        tuple: (host, port) - defaults to ("127.0.0.1", 5001) if not configured
+    """
+    default_host = "127.0.0.1"
+    default_port = 5000
+
+    config_path = Path("config.ini")
+    if not config_path.exists():
+        return default_host, default_port
+
+    try:
+        ini = configparser.ConfigParser()
+        ini.read(config_path)
+
+        tracking_uri = ini.get("mlflow", "tracking_uri", fallback=None)
+        if not tracking_uri:
+            return default_host, default_port
+
+        # Parse the URI (e.g., "http://192.168.1.100:5000")
+        parsed = urlparse(tracking_uri)
+
+        host = parsed.hostname or default_host
+        port = parsed.port or default_port
+
+        return host, port
+    except Exception:
+        return default_host, default_port
+
+
 def is_port_in_use(host, port):
     """
     Check if a port is already in use.
@@ -49,7 +84,7 @@ def is_port_in_use(host, port):
         return s.connect_ex((host, port)) == 0
 
 
-def start_mlflow_server(host="127.0.0.1", port=5001):
+def start_mlflow_server(host, port):
     """
     Start MLflow tracking server as a background process.
 
@@ -57,8 +92,8 @@ def start_mlflow_server(host="127.0.0.1", port=5001):
     will skip starting a new instance.
 
     Args:
-        host: The hostname to bind the server to (default: "127.0.0.1")
-        port: The port number for the server (default: 5001)
+        host: The hostname to bind the server to
+        port: The port number for the server
 
     Returns:
         subprocess.Popen or None: The server process, or None if already running
@@ -144,8 +179,9 @@ def cmd_run(args):
 
     This command:
         1. Validates that all required config files exist
-        2. Starts MLflow server (unless --no-server is specified)
-        3. Runs the evaluation using the evaluator module
+        2. Reads MLflow host/port from config.ini
+        3. Starts MLflow server (unless --no-server is specified)
+        4. Runs the evaluation using the evaluator module
 
     Args:
         args: Parsed command-line arguments (includes --no-server flag)
@@ -161,9 +197,10 @@ def cmd_run(args):
         print("\nRun 'rag-sentinel init' first to create config files.")
         sys.exit(1)
 
-    # Start MLflow server if not disabled
+    # Start MLflow server if not disabled (uses host/port from config.ini)
     if not args.no_server:
-        start_mlflow_server()
+        host, port = get_mlflow_host_port()
+        start_mlflow_server(host, port)
 
     # Import and run the evaluation
     from rag_sentinel.evaluator import run_evaluation

{rag_sentinel-0.1.3 → rag_sentinel-0.1.4/src/rag_sentinel.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rag-sentinel
-Version: 0.1.3
+Version: 0.1.4
 Summary: RAG Evaluation Framework using Ragas metrics and MLflow tracking
 Author: RAGSentinel Team
 License: MIT