add batch cli process status endpoint

Author: grctest

app/lib/endpoints/llama_cli_endpoints.py

@@ -10,7 +10,8 @@
     LlamaCliChatRequest,
     BatchLlamaCliInitRequest,
     BatchLlamaCliRemoveRequest,
-    BatchLlamaCliChatRequest
+    BatchLlamaCliChatRequest,
+    BatchLlamaCliStatusRequest
 )
 
 # Import the process management functions for persistent sessions
@@ -200,3 +201,21 @@ async def process_request(req: LlamaCliChatRequest):
 
     results = await asyncio.gather(*(process_request(req) for req in batch_request.requests))
     return results
+
+async def handle_get_batch_llama_cli_status(batch_request: BatchLlamaCliStatusRequest) -> List[Dict[str, Any]]:
+    """
+    Processes a batch request to get the status of multiple persistent llama-cli sessions.
+    """
+    async def process_request(alias: str):
+        try:
+            # Reuse the single status handler logic
+            result = await get_llama_cli_session_status(alias)
+            return {"cli_alias": alias, "status": "success", "data": result}
+        except HTTPException as e:
+            return {"cli_alias": alias, "status": "error", "detail": e.detail, "status_code": e.status_code}
+        except Exception as e:
+            logger.error(f"Unexpected error processing batch status for alias {alias}: {str(e)}", exc_info=True)
+            return {"cli_alias": alias, "status": "error", "detail": "An unexpected server error occurred.", "status_code": 500}
+
+    results = await asyncio.gather(*(process_request(alias) for alias in batch_request.aliases))
+    return results

app/lib/models.py

@@ -154,5 +154,14 @@ def validate_aliases(cls, v):
             raise ValueError('An alias in the list contains invalid characters. Use only alphanumeric, hyphens, and underscores.')
         return v
 
+class BatchLlamaCliStatusRequest(BaseModel):
+    aliases: list[str] = Field(..., description="A list of llama-cli configuration aliases to check the status of.")
+
+    @validator('aliases', each_item=True)
+    def validate_aliases(cls, v):
+        if not is_safe_cli_alias(v):
+            raise ValueError('An alias in the list contains invalid characters. Use only alphanumeric, hyphens, and underscores.')
+        return v
+
 class BatchLlamaCliChatRequest(BaseModel):
     requests: list[LlamaCliChatRequest] = Field(..., description="A list of chat requests to process.")

app/main.py

@@ -12,7 +12,8 @@
     LlamaCliChatRequest,
     BatchLlamaCliInitRequest,
     BatchLlamaCliRemoveRequest,
-    BatchLlamaCliChatRequest
+    BatchLlamaCliChatRequest,
+    BatchLlamaCliStatusRequest
 )
 
 from lib.endpoints.chat_endpoints import ChatRequest, MultiChatRequest
@@ -46,7 +47,8 @@
     chat_with_llama_cli_session,
     handle_initialize_batch_llama_cli_configs,
     handle_remove_batch_llama_cli_configs,
-    handle_batch_chat_with_llama_cli
+    handle_batch_chat_with_llama_cli,
+    handle_get_batch_llama_cli_status
 )
 
 # --- Logging Configuration ---
@@ -628,6 +630,33 @@ async def get_llama_cli_status_endpoint(cli_alias: str = Path(..., description="
         logger.error(f"Unexpected error in /llama-cli-status for alias {cli_alias}: {str(e)}", exc_info=True)
         raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=f"An unexpected error occurred: {str(e)}")
 
+@app.post(
+    "/batch-llama-cli-status",
+    summary="Get Status of Multiple Llama CLI Sessions",
+    tags=["Llama CLI Management"],
+    operation_id="get_batch_llama_cli_session_status"
+)
+async def batch_llama_cli_status(request: BatchLlamaCliStatusRequest):
+    """
+    Retrieves the status of multiple `llama-cli` sessions in a single batch request.
+
+    **Request Body**:
+    - `BatchLlamaCliStatusRequest`: A JSON object containing a list of `aliases` (strings)
+      of the `llama-cli` sessions to check.
+
+    **Successful Response (200 OK)**:
+    - A JSON list where each item corresponds to an alias in the request, detailing its
+      status (running/stopped), PID (if running), and configuration.
+
+    **Error Responses**:
+    - `500 Internal Server Error`: If an unexpected error occurs during the batch processing logic.
+    """
+    try:
+        return await handle_get_batch_llama_cli_status(request)
+    except Exception as e:
+        logger.error(f"Error in /batch-llama-cli-status endpoint: {str(e)}", exc_info=True)
+        raise HTTPException(status_code=500, detail=f"An unexpected error occurred while fetching batch CLI status: {str(e)}")
+
 @app.post(
     "/chat-llama-cli", 
     summary="Chat with a Persistent Llama CLI Session",