Merge branch 'main' into robot2

Author: AnkushMalaker

.github/workflows/advanced-docker-compose-build.yml

@@ -106,6 +106,24 @@ jobs:
             echo ".env not found; continuing"
           fi
 
+      - name: Free Disk Space
+        run: |
+          echo "Freeing disk space..."
+          df -h
+          sudo rm -rf /usr/share/dotnet
+          sudo rm -rf /usr/local/lib/android
+          sudo rm -rf /opt/ghc
+          sudo rm -rf /usr/share/swift
+          docker system prune -a -f
+          df -h
+
+      - name: Initialize OpenMemory cache
+        run: |
+          echo "Initializing OpenMemory cache..."
+          cd ../../extras/openmemory-mcp
+          chmod +x init-cache.sh
+          ./init-cache.sh
+
       - name: Determine version
         id: version
         run: |
@@ -116,6 +134,7 @@ jobs:
           else
             VERSION="sha-${GITHUB_SHA::7}"
           fi
+          
           echo "VERSION=$VERSION" >> "$GITHUB_OUTPUT"
 
       - name: Build, tag, and push services sequentially with version
@@ -177,6 +196,9 @@ jobs:
             docker image rm -f "$target_image" || true
             docker image rm -f "$latest_image" || true
             echo "::endgroup::"
+            
+            # Aggressive cleanup to save space
+            docker system prune -af || true
           done
           
           # Build and push parakeet-asr with CUDA variants (cu121, cu126, cu128)
@@ -205,6 +227,9 @@ jobs:
               docker image rm -f "$target_image" || true
               docker image rm -f "$latest_image" || true
             fi
+            
+            # Aggressive cleanup to save space
+            docker system prune -af || true
           done
           cd - > /dev/null
           echo "::endgroup::"
@@ -237,6 +262,9 @@ jobs:
               docker image rm -f "$target_image" || true
               docker image rm -f "$latest_image" || true
             fi
+            
+            # Aggressive cleanup to save space
+            docker system prune -af || true
           done
           cd - > /dev/null
           echo "::endgroup::"

extras/asr-services/Dockerfile_Moonshine

@@ -12,8 +12,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 # Dependency manifest first for cache‑friendly installs
 COPY pyproject.toml uv.lock ./
-RUN --mount=type=cache,target=/root/.cache/uv \
-    uv sync --no-install-project --compile-bytecode --group moonshine
+RUN uv sync --no-install-project --compile-bytecode --group moonshine && \
+    uv cache clean
 
 # Add source and install project itself
 COPY . .

extras/asr-services/Dockerfile_Parakeet

@@ -16,8 +16,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 # Dependency manifest first for cache‑friendly installs
 COPY pyproject.toml uv.lock ./
-RUN --mount=type=cache,target=/root/.cache/uv \
-    uv sync --no-install-project --group parakeet --extra ${CUDA_VERSION}
+RUN uv sync --no-install-project --group parakeet --extra ${CUDA_VERSION} && \
+    uv cache clean
 
 # Should prepare the .venv for use :)
 

extras/speaker-recognition/Dockerfile

@@ -28,7 +28,8 @@ RUN mkdir -p src/simple_speaker_recognition
 COPY src/simple_speaker_recognition/__init__.py src/simple_speaker_recognition/
 
 # Install dependencies and package
-RUN uv sync --no-dev --extra ${PYTORCH_CUDA_VERSION}
+RUN uv sync --no-dev --extra ${PYTORCH_CUDA_VERSION} && \
+    uv cache clean
 
 # Create directories
 RUN mkdir -p /app/audio_chunks /app/debug /app/data /models

tests/endpoints/client_queue_tests.robot

@@ -2,17 +2,17 @@
 Documentation    Client and Queue Management API Tests
 Library          RequestsLibrary
 Library          Collections
-Resource         ../setup/setup_keywords.robot
-Resource         ../setup/teardown_keywords.robot
-Resource         ../resources/user_keywords.robot
+Resource         ../resources/setup_resources.robot
+Resource         ../resources/session_resources.robot
+Resource         ../resources/user_resources.robot
 Suite Setup      Suite Setup
 Suite Teardown   Delete All Sessions
-Test Setup       Test Cleanup
+
 *** Test Cases ***
 
 Get Active Clients Test
     [Documentation]    Test getting active client information
-    [Tags]    infra
+    [Tags]             client    active    positive
 
     Create API Session    admin_session
     ${response}=       GET On Session    admin_session    /api/clients/active
@@ -30,7 +30,7 @@ Get Active Clients Test
 
 Get Queue Jobs Test
     [Documentation]    Test getting queue jobs with pagination
-    [Tags]    queue
+    [Tags]             queue    jobs    positive
 
     Create API Session    admin_session
     &{params}=         Create Dictionary    limit=20    offset=0
@@ -52,7 +52,7 @@ Get Queue Jobs Test
 
 Get Queue Jobs With Different Limits Test
     [Documentation]    Test queue jobs pagination with different limits
-    [Tags]    queue
+    [Tags]             queue    jobs    pagination    positive
     Get Anonymous Session    anon_session
 
     Create API Session    admin_session
@@ -77,45 +77,53 @@ Get Queue Jobs With Different Limits Test
 
 Get Queue Statistics Test
     [Documentation]    Test getting queue statistics
-    [Tags]    queue
+    [Tags]             queue    statistics    positive
+    Get Anonymous Session    anon_session
+
+    Create API Session    admin_session
 
-    ${response}=   GET On Session    api    /api/queue/stats
+    ${response}=   GET On Session    admin_session    /api/queue/stats
     Should Be Equal As Integers    ${response.status_code}    200
 
     ${stats}=      Set Variable    ${response.json()}
-    Dictionary Should Contain Key    ${stats}    queued_jobs
-    Dictionary Should Contain Key    ${stats}    processing_jobs
-    Dictionary Should Contain Key    ${stats}    completed_jobs
-    Dictionary Should Contain Key    ${stats}    failed_jobs
-    Dictionary Should Contain Key    ${stats}    total_jobs
-    Dictionary Should Contain Key    ${stats}    cancelled_jobs
-    Dictionary Should Contain Key    ${stats}    deferred_jobs
+    Dictionary Should Contain Key    ${stats}    queued
+    Dictionary Should Contain Key    ${stats}    processing
+    Dictionary Should Contain Key    ${stats}    completed
+    Dictionary Should Contain Key    ${stats}    failed
 
+    # All counts should be non-negative
+    Should Be True    ${stats}[queued] >= 0
+    Should Be True    ${stats}[processing] >= 0
+    Should Be True    ${stats}[completed] >= 0
+    Should Be True    ${stats}[failed] >= 0
 
 Get Queue Health Test
     [Documentation]    Test getting queue health status
-    [Tags]    queue	health
+    [Tags]             queue    health    positive
+    Get Anonymous Session    anon_session
 
-    ${response}=   GET On Session    api    /api/queue/worker-details
+    Create API Session    admin_session
+    ${response}=   GET On Session    admin_session    /api/queue/health
     Should Be Equal As Integers    ${response.status_code}    200
 
     ${health}=     Set Variable    ${response.json()}
-    Dictionary Should Contain Key    ${health}    workers
-    Dictionary Should Contain Key    ${health}    redis_connection
+    Dictionary Should Contain Key    ${health}    status
+    Dictionary Should Contain Key    ${health}    worker_running
+    Dictionary Should Contain Key    ${health}    message
 
     # Status should be one of expected values
-    Should Be True    '${health}[redis_connection]' in ['healthy', 'stopped', 'unhealthy']
+    Should Be True    '${health}[status]' in ['healthy', 'stopped', 'unhealthy']
 
 Queue Jobs User Isolation Test
     [Documentation]    Test that regular users only see their own queue jobs
-    [Tags]    queue	permissions
+    [Tags]             queue    security    isolation
     Get Anonymous Session    anon_session
 
     Create API Session    admin_session
 
     # Create a test user
-    ${test_user}=      Create Test User    admin_session
-    Create API Session    user_session    email=${test_user}[email]    password=${TEST_USER_PASSWORD}
+    ${test_user}=      Create Test User    admin_session    test-user-${RANDOM_ID}@example.com    test-password-123
+    Create API Session    user_session    email=test-user-${RANDOM_ID}@example.com    password=test-password-123
 
     # Get user's jobs (should be filtered to their user_id)
     ${response}=       GET On Session    user_session    /api/queue/jobs
@@ -132,11 +140,11 @@ Queue Jobs User Isolation Test
     END
 
     # Cleanup
-    Delete User    admin_session    ${test_user}[id]
+    Delete Test User    ${test_user}[user_id]
 
 Invalid Queue Parameters Test
     [Documentation]    Test queue endpoints with invalid parameters
-    [Tags]    queue
+    [Tags]             queue    negative    validation
     Get Anonymous Session    anon_session
 
     Create API Session    admin_session
@@ -158,29 +166,41 @@ Invalid Queue Parameters Test
 
 Unauthorized Client Access Test
     [Documentation]    Test that client endpoints require authentication
-    [Tags]    infra	permissions
+    [Tags]             client    security    negative
     Get Anonymous Session    session
 
     # Try to access active clients without token
-    ${response}=    GET On Session    session    /api/clients/active    expected_status=401
+    ${response}=    GET On Session    ${session}    /api/clients/active    expected_status=401
     Should Be Equal As Integers    ${response.status_code}    401
 
 Unauthorized Queue Access Test
     [Documentation]    Test that queue endpoints require authentication
-    [Tags]    queue	permissions
+    [Tags]             queue    security    negative
     Get Anonymous Session    session
 
     # Try to access queue jobs without token
-    ${response}=    GET On Session    session    /api/queue/jobs    expected_status=401
+    ${response}=    GET On Session    ${session}    /api/queue/jobs    expected_status=401
     Should Be Equal As Integers    ${response.status_code}    401
 
     # Try to access queue stats without token
-    ${response}=    GET On Session    session    /api/queue/stats    expected_status=401
+    ${response}=    GET On Session    ${session}    /api/queue/stats    expected_status=401
     Should Be Equal As Integers    ${response.status_code}    401
 
+Queue Health Public Access Test
+    [Documentation]    Test that queue health endpoint is publicly accessible
+    [Tags]             queue    health    public
+    Get Anonymous Session    session
+
+    # Queue health should be accessible without authentication
+    ${response}=   GET On Session    ${session}    /api/queue/health
+    Should Be Equal As Integers    ${response.status_code}    200
+
+    ${health}=     Set Variable    ${response.json()}
+    Dictionary Should Contain Key    ${health}    status
+
 Client Manager Integration Test
     [Documentation]    Test client manager functionality
-    [Tags]    infra	e2e
+    [Tags]             client    manager    integration
     Get Anonymous Session    anon_session
 
     Create API Session    admin_session