chore: Metrics , tracks commands , errors for now

cli/commands.py

@@ -14,10 +14,12 @@
 from rich.progress import (
     Progress, SpinnerColumn, TextColumn, TimeElapsedColumn
 )
-from typing import Dict, Any, Optional, Callable
+from typing import Dict, Any, Callable
 from rich import box
 from core.runner import execute_prompt_tests, execute_rerun_test
 from core.config_manager.cli_adapter import CLIConfigAdapter
+from core.analytics.tracker import analytics_tracker
+from core.analytics.tracker import UsageEvent, ErrorEvent, InteractionType
 
 
 def dict_to_cli_table(
@@ -120,6 +122,7 @@ def test(config_path, prompt, strategy, provider, output, report, parallel, verb
     cli_adapter = CLIConfigAdapter()
 
     try:
+        analytics_tracker.track(UsageEvent(name="test", interaction_type=InteractionType.CLI))
         # Load configuration from CLI arguments
         cli_adapter.load_from_cli(
             config_path=config_path,
@@ -160,13 +163,16 @@ def test(config_path, prompt, strategy, provider, output, report, parallel, verb
 
     except FileNotFoundError as e:
         click.echo(str(e), err=True)
+        analytics_tracker.track(ErrorEvent(name="test", interaction_type=InteractionType.CLI, error_msg=str(e)))
         sys.exit(1)
     except Exception as e:
         click.echo(f"Error processing configuration: {e}", err=True)
+        analytics_tracker.track(ErrorEvent(name="test", interaction_type=InteractionType.CLI, error_msg=str(e)))
         sys.exit(1)
 
     # Run the tests with a progress indicator
     console.print("\nRunning tests...")
+
     with Progress(
         SpinnerColumn(),
         TextColumn("[bold blue]{task.description}"),
@@ -176,6 +182,7 @@ def test(config_path, prompt, strategy, provider, output, report, parallel, verb
         report_data = execute_prompt_tests(config_dict=runner_config)
         progress.update(task, completed=True)
 
+
     console.print("[bold green]Tests completed successfully![/]")
     console.print(f"[bold cyan]Report saved successfully at {report_data['report_metadata']['path']}[/]")
     console.print("\n")
@@ -238,6 +245,7 @@ def test(config_path, prompt, strategy, provider, output, report, parallel, verb
 @click.option('--summary', '-s', is_flag=True, help='Show only summary statistics')
 def report(report_file, format, summary):
     """Analyze and view previous test results. If no file is specified, uses the latest report."""
+    analytics_tracker.track(UsageEvent(name="report", interaction_type=InteractionType.CLI))
     try:
         # If no report file is specified, find the latest one
         if not report_file:
@@ -298,11 +306,13 @@ def report(report_file, format, summary):
                 f.write("</body></html>")
             click.echo(f"HTML report saved to {html_path}")
 
-    except FileNotFoundError:
-        click.echo(f"Report file not found: {report_file}", err=True)
+    except FileNotFoundError as e:
+        analytics_tracker.track(ErrorEvent(name="report", interaction_type=InteractionType.CLI, error_msg=str(e)))    
+        click.echo(f"Error: Report file not found: {e}", err=True)
         sys.exit(1)
     except json.JSONDecodeError:
-        click.echo(f"Invalid JSON in report file: {report_file}", err=True)
+        analytics_tracker.track(ErrorEvent(name="report", interaction_type=InteractionType.CLI, error_msg="Invalid JSON format in report file."))
+        click.echo("Error: Invalid JSON format in report file.", err=True)
         sys.exit(1)
 
 
@@ -398,7 +408,6 @@ def generate(type, template, output):
 @click.option('--validate', '-v', help='Validate a configuration file')
 def config(list, show, validate):
     """Manage configuration files."""
-
     if list:
         click.echo("Available configurations:")
         # Get the list of config directories from core.config
@@ -447,7 +456,7 @@ def config(list, show, validate):
             click.echo("Processed configuration for runner:")
             click.echo(yaml.dump(runner_config, default_flow_style=False))
         except FileNotFoundError as e:
-            click.echo(str(e), err=True)
+            click.echo(f"Error: Configuration file not found: {e}", err=True)
             sys.exit(1)
         except Exception as e:
             click.echo(f"Configuration validation failed: {e}", err=True)
@@ -471,6 +480,7 @@ def rerun(prompt, report_file):
     """
     # Create a rich console for showing output
     console = Console()
+    analytics_tracker.track(UsageEvent(name="rerun", interaction_type=InteractionType.CLI))
     # Validate inputs
     if not prompt:
         console.print("[red]Error: System prompt is required.[/red]")

core/analytics/client.py

@@ -0,0 +1,17 @@
+import os
+import uuid
+
+def get_client_id() -> str:
+    """Retrieve or create a unique, anonymous client ID for this user."""
+    path = os.path.join(os.path.expanduser("~"), ".compliant-llm", ".client-id")
+    os.makedirs(os.path.dirname(path), exist_ok=True)
+
+    if os.path.exists(path):
+        with open(path, "r") as f:
+            return f.read().strip()
+
+    # Generate and save a new UUID
+    new_id = str(uuid.uuid4())
+    with open(path, "w") as f:
+        f.write(new_id)
+    return new_id

core/analytics/tracker.py

@@ -0,0 +1,193 @@
+import os
+import uuid
+from enum import Enum
+from typing import Optional
+from pydantic import BaseModel, Field
+import logging
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+# OpenTelemetry Tracing
+from opentelemetry import trace
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor, ConsoleSpanExporter
+from azure.monitor.opentelemetry.exporter import AzureMonitorTraceExporter
+
+# OpenTelemetry Metrics
+from opentelemetry.sdk.resources import SERVICE_NAME, Resource
+from opentelemetry.sdk.metrics import MeterProvider
+from opentelemetry.metrics import set_meter_provider, get_meter_provider
+from opentelemetry.sdk.metrics.export import (
+    PeriodicExportingMetricReader,
+    ConsoleMetricExporter
+)
+from azure.monitor.opentelemetry.exporter import AzureMonitorMetricExporter
+
+# ----------------------------
+# Client ID Handling
+# ----------------------------
+from .client import get_client_id
+
+CLIENT_ID = get_client_id()
+
+# ----------------------------
+# Enums and Event Models
+# ----------------------------
+class EventType(str, Enum):
+    USAGE = "usage"
+    ERROR = "error"
+
+class InteractionType(str, Enum):
+    CLI = "cli"
+    DASHBOARD = "dashboard"
+    API = "api"
+    BATCH = "batch"
+
+class BaseEvent(BaseModel):
+    name: str
+    interaction_type: InteractionType
+    client_id: str = Field(default_factory=get_client_id)
+    type: EventType
+
+class UsageEvent(BaseEvent):
+    type: EventType = EventType.USAGE
+
+class ErrorEvent(BaseEvent):
+    error_msg: str
+    type: EventType = EventType.ERROR
+
+# ----------------------------
+# OpenTelemetry Initialization
+# ----------------------------
+
+# Define common resource attributes
+resource = Resource.create({
+    SERVICE_NAME: "compliant-llm",
+    "service.version": "1.0.0",
+    "environment": os.getenv("ENVIRONMENT", "prod")
+})
+
+# ----- Initialize Metrics -----
+try:
+    instrumentation_key = os.getenv("AZURE_INSTRUMENTATION_KEY", "ed532436-db1f-46bb-aeef-17cb4f3dcf8b")
+    ingestion_endpoint = os.getenv("AZURE_INGESTION_ENDPOINT", "https://westus-0.in.applicationinsights.azure.com/")
+
+    metric_exporter = AzureMonitorMetricExporter(
+        connection_string=f"InstrumentationKey={instrumentation_key};IngestionEndpoint={ingestion_endpoint}"
+    )
+
+    metric_readers = [
+        PeriodicExportingMetricReader(ConsoleMetricExporter()),  # Optional for local debugging
+        PeriodicExportingMetricReader(metric_exporter)
+    ]
+
+    meter_provider = MeterProvider(resource=resource, metric_readers=metric_readers)
+    set_meter_provider(meter_provider)
+
+    print("✅ Azure Monitor metrics initialized.")
+except Exception as e:
+    print(f"❌ Failed to initialize Azure Monitor metrics: {e}")
+
+# ----- Initialize Tracing -----
+try:
+    trace_exporter = AzureMonitorTraceExporter(
+        connection_string=f"InstrumentationKey={instrumentation_key};IngestionEndpoint={ingestion_endpoint}"
+    )
+
+    trace.set_tracer_provider(TracerProvider(resource=resource))
+    tracer_provider = trace.get_tracer_provider()
+
+    tracer_provider.add_span_processor(BatchSpanProcessor(trace_exporter))
+    tracer_provider.add_span_processor(BatchSpanProcessor(ConsoleSpanExporter()))  # Optional for local debugging
+
+    print("✅ Azure Monitor traces initialized.")
+except Exception as e:
+    print(f"❌ Failed to initialize Azure Monitor tracing: {e}")
+
+# ----------------------------
+# Tracker Class (Traces + Metrics)
+# ----------------------------
+class AnalyticsTracker:
+    def __init__(self):
+        # Initialize tracer
+        self.tracer = trace.get_tracer("compliant-llm")
+        self.tracer_provider = trace.get_tracer_provider()
+
+        if not self.tracer_provider:
+            logger.warning("⚠️ No tracer provider available")
+            self.tracer = None
+
+        # Initialize metrics
+        try:
+            self.meter = get_meter_provider().get_meter("compliant-llm")
+            self.usage_counter = self.meter.create_counter(
+                name="compliant_llm.command_invocations",
+                description="Number of times a CLI/dashboard/api command was invoked"
+            )
+            self.error_counter = self.meter.create_counter(
+                name="compliant_llm.errors",
+                description="Number of errors encountered"
+            )
+        except Exception as e:
+            logger.error(f"⚠️ Metrics counters unavailable: {e}")
+            self.usage_counter = None
+            self.error_counter = None
+
+    def track(self, event: BaseEvent):
+        logger.info(f"TRACKING EVENT: {event.name} ({event.type.value})")
+
+        if not self.tracer:
+            logger.warning("⚠️ No tracer available")
+            return
+
+        # ----- Trace Span -----
+        span_name = f"{event.type.value}:{event.name}"
+        try:
+            with self.tracer.start_as_current_span(span_name) as span:
+                if span:  # Check if span was created successfully
+                    span.set_attribute("client_id", event.client_id)
+                    span.set_attribute("interaction_type", event.interaction_type.value)
+                    span.set_attribute("event_type", event.type.value)
+                    span.set_attribute("command", event.name)
+                    if isinstance(event, ErrorEvent):
+                        span.set_attribute("error_msg", event.error_msg[:100])
+        except Exception as e:
+            logger.error(f"❌ Error creating span: {e}")
+
+        # ----- Metrics -----
+        attributes = {
+            "client_id": event.client_id,
+            "interaction_type": event.interaction_type.value,
+            "name": event.name
+        }
+
+        try:
+            if event.type == EventType.USAGE and self.usage_counter:
+                self.usage_counter.add(1, attributes)
+            elif event.type == EventType.ERROR and self.error_counter:
+                attributes["error_msg"] = getattr(event, "error_msg", "unknown")[:100]
+                self.error_counter.add(1, attributes)
+        except Exception as e:
+            logger.error(f"❌ Error recording metrics: {e}")
+
+# ----------------------------
+# Decorator for Usage Tracking
+# ----------------------------
+def track_usage(name: str, interaction_type: InteractionType = InteractionType.CLI):
+    def decorator(func):
+        def wrapper(*args, **kwargs):
+            event = UsageEvent(name=name, interaction_type=interaction_type)
+            analytics_tracker.track(event)
+            return func(*args, **kwargs)
+        return wrapper
+    return decorator
+
+# ----------------------------
+# Tracker Instance
+# ----------------------------
+analytics_tracker = AnalyticsTracker()

pyproject.toml

@@ -23,7 +23,15 @@ dependencies = [
     "aiofiles",
     "plotly",
     "psutil",
-    "markdown"
+    "markdown",
+    "opentelemetry-api",
+    "opentelemetry-sdk",
+    "opentelemetry-exporter-otlp",
+    "opentelemetry-instrumentation",
+    "azure-monitor-opentelemetry-exporter",
+    "azure-core",
+    "azure-identity",
+    "azure-monitor-opentelemetry"
 ]
 
 [project.scripts]

requirements-lock.txt

@@ -83,3 +83,11 @@ tzdata==2025.2
 urllib3==2.4.0
 yarl==1.20.0
 zipp==3.21.0
+opentelemetry-api
+opentelemetry-sdk
+opentelemetry-exporter-otlp
+opentelemetry-instrumentation
+azure-monitor-opentelemetry-exporter
+azure-core
+azure-identity
+azure-monitor-opentelemetry

requirements.txt

@@ -8,4 +8,12 @@ pytest
 aiofiles
 plotly
 psutil
-markdown
+markdown
+opentelemetry-api
+opentelemetry-sdk
+opentelemetry-exporter-otlp
+opentelemetry-instrumentation
+azure-monitor-opentelemetry-exporter
+azure-core
+azure-identity
+azure-monitor-opentelemetry

ui/dashboard.py

@@ -12,8 +12,10 @@
 from dotenv import load_dotenv, set_key, get_key
 import socket
 from core.config_manager.ui_adapter import UIConfigAdapter
+
 from rich.console import Console
 from ui.constants.provider import PROVIDER_SETUP
+from core.analytics.tracker import UsageEvent, InteractionType, ErrorEvent, analytics_tracker
 
 console = Console()
 load_dotenv()
@@ -119,8 +121,19 @@ def run_test(prompt, selected_strategies, config):
         adapter = UIConfigAdapter()
         # adapter.update_config(config)
         results = adapter.run_test(prompt, selected_strategies, config)
+        analytics_tracker.track(UsageEvent(
+            name="test",
+            interaction_type=InteractionType.DASHBOARD,
+            command="run_test"
+        ))
         return json.dumps(results), ""
     except Exception as e:
+        analytics_tracker.track(ErrorEvent(
+            name="test",
+            interaction_type=InteractionType.DASHBOARD,
+            command="run_test",
+            error_msg=str(e)
+        ))
         return "", str(e)
 
 def render_beautiful_json_output(json_output):