aiverify-moonshot 0.4.0__py3-none-any.whl

moonshot/integrations/web_api/routes/endpoint.py

@@ -0,0 +1,282 @@
+from typing import Optional
+
+from dependency_injector.wiring import Provide, inject
+from fastapi import APIRouter, Depends, HTTPException
+
+from ..container import Container
+from ..schemas.endpoint_create_dto import EndpointCreateDTO, EndpointUpdateDTO
+from ..schemas.endpoint_response_model import EndpointDataModel
+from ..services.endpoint_service import EndpointService
+from ..services.utils.exceptions_handler import ServiceException
+
+router = APIRouter(tags=["Endpoint"])
+
+
+@router.post("/api/v1/llm-endpoints")
+@inject
+def add_new_endpoint(
+    endpoint_data: EndpointCreateDTO,
+    endpoint_service: EndpointService = Depends(Provide[Container.endpoint_service]),
+) -> dict[str, str] | tuple[dict[str, str], int]:
+    """
+    Add a new endpoint to the database.
+
+    Args:
+        endpoint_data (EndpointCreateDTO): The data transfer object containing endpoint details.
+        endpoint_service (EndpointService): The service responsible for adding the endpoint.
+
+    Returns:
+        dict[str, str]: A message indicating the successful addition of the endpoint.
+
+    Raises:
+        HTTPException: An error with status code 404 if the file is not found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        endpoint_service.add_endpoint(endpoint_data)
+        return {"message": "Endpoint added successfully"}
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to add endpoint: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to add endpoint: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to add endpoint: {e.msg}"
+            )
+
+
+@router.get("/api/v1/llm-endpoints")
+@inject
+def get_all_endpoints(
+    endpoint_service: EndpointService = Depends(Provide[Container.endpoint_service]),
+) -> list[Optional[EndpointDataModel]]:
+    """
+    Get all the endpoints from the database.
+
+    Args:
+        endpoint_service (EndpointService): The service responsible for retrieving all endpoints.
+
+    Returns:
+        list[Optional[EndpointDataModel]]: A list of endpoint data models,
+        which may contain None if an endpoint is not found.
+
+    Raises:
+        HTTPException: An error with status code 404 if the file is not found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        endpoint_models = endpoint_service.get_all_endpoints()
+        for endpoint_model in endpoint_models:
+            endpoint_model.mask_token()
+        return endpoint_models
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to get endpoint: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to get endpoint: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to get endpoint: {e.msg}"
+            )
+
+
+@router.get("/api/v1/llm-endpoints/name")
+@inject
+def get_all_endpoints_name(
+    endpoint_service: EndpointService = Depends(Provide[Container.endpoint_service]),
+) -> list[Optional[str]]:
+    """
+    Get all the endpoint names from the database.
+
+    Args:
+        endpoint_service (EndpointService): The service responsible for retrieving all endpoint names.
+
+    Returns:
+        list[Optional[str]]: A list of endpoint names, which may contain None if an endpoint name is not found.
+
+    Raises:
+        HTTPException: An error with status code 404 if the file is not found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        return endpoint_service.get_all_endpoints_names()
+    except ServiceException as e:
+        error_status_code = 500
+        if e.error_code == "FileNotFound":
+            error_status_code = 404
+        elif e.error_code == "ValidationError":
+            error_status_code = 400
+        raise HTTPException(
+            status_code=error_status_code,
+            detail=f"Failed to get endpoint names: {e.msg}",
+        )
+
+
+@router.get("/api/v1/llm-endpoints/{endpoint_id}")
+@inject
+def get_endpoint(
+    endpoint_id: str,
+    endpoint_service: EndpointService = Depends(Provide[Container.endpoint_service]),
+) -> EndpointDataModel | None:
+    """
+    Get an endpoint from the database by its ID.
+
+    Args:
+        endpoint_id (str): The unique identifier of the endpoint to retrieve.
+        endpoint_service (EndpointService): The service responsible for retrieving the endpoint.
+
+    Returns:
+        EndpointDataModel | None: The endpoint data model if found, otherwise None.
+
+    Raises:
+        HTTPException: An error with status code 404 if the endpoint is not found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        endpoint_model = endpoint_service.get_endpoint(endpoint_id)
+        if endpoint_model:
+            endpoint_model.mask_token()
+        return endpoint_model
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to get endpoint: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to get endpoint: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to get endpoint: {e.msg}"
+            )
+
+
+@router.put("/api/v1/llm-endpoints/{endpoint_id}")
+@inject
+async def update_endpoint(
+    endpoint_id: str,
+    endpoint_data: EndpointUpdateDTO,
+    endpoint_service: EndpointService = Depends(Provide[Container.endpoint_service]),
+) -> dict[str, str] | tuple[dict[str, str], int]:
+    """
+    Update an existing endpoint in the database by its ID.
+
+    Args:
+        endpoint_id (str): The unique identifier of the endpoint to update.
+        endpoint_data (EndpointCreateDTO): The data transfer object containing the updated endpoint details.
+        endpoint_service (EndpointService): The service responsible for updating the endpoint.
+
+    Returns:
+        dict[str, str]: A message indicating the successful update of the endpoint.
+
+    Raises:
+        HTTPException: An error with status code 404 if the endpoint is not found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        endpoint_service.update_endpoint(endpoint_id, endpoint_data)
+        return {"message": "Endpoint updated successfully"}
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to update endpoint: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to update endpoint: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to update endpoint: {e.msg}"
+            )
+
+
+@router.delete("/api/v1/llm-endpoints/{endpoint_id}")
+@inject
+async def delete_endpoint(
+    endpoint_id: str,
+    endpoint_service: EndpointService = Depends(Provide[Container.endpoint_service]),
+) -> dict[str, str] | tuple[dict[str, str], int]:
+    """
+    Delete an existing endpoint from the database by its ID.
+
+    Args:
+        endpoint_id (str): The unique identifier of the endpoint to delete.
+        endpoint_service (EndpointService): The service responsible for deleting the endpoint.
+
+    Returns:
+        dict[str, str]: A message indicating the successful deletion of the endpoint.
+
+    Raises:
+        HTTPException: An error with status code 404 if the endpoint is not found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        endpoint_service.delete_endpoint(endpoint_id)
+        return {"message": "Endpoint deleted successfully"}
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to delete endpoint: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to delete endpoint: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to delete endpoint: {e.msg}"
+            )
+
+
+@router.get("/api/v1/connectors")
+@inject
+def get_all_connectors(
+    endpoint_service: EndpointService = Depends(Provide[Container.endpoint_service]),
+) -> list[Optional[str]]:
+    """
+    Get all the connector types from the database.
+
+    Args:
+        endpoint_service (EndpointService): The service responsible for retrieving all connector types.
+
+    Returns:
+        list[Optional[str]]: A list of connector types, which may contain None if a connector type is not found.
+
+    Raises:
+        HTTPException: An error with status code 404 if the file is not found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    # TODO - type check and model validation
+    try:
+        return endpoint_service.get_all_connectors()
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to get connector types: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to get connector types: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to get connector types: {e.msg}"
+            )

moonshot/integrations/web_api/routes/metric.py

@@ -0,0 +1,78 @@
+from dependency_injector.wiring import Provide, inject
+from fastapi import APIRouter, Depends, HTTPException
+
+from ..container import Container
+from ..services.metric_service import MetricService
+from ..services.utils.exceptions_handler import ServiceException
+
+router = APIRouter(tags=["Metric"])
+
+
+@router.get("/api/v1/metrics")
+@inject
+def get_all_metrics(
+    metric_service: MetricService = Depends(Provide[Container.metric_service]),
+) -> list[dict]:
+    """
+    Retrieve all metrics from the database.
+
+    Args:
+        metric_service (MetricService): The service responsible for retrieving metrics.
+
+    Returns:
+        list[str]: A list of all metrics.
+
+    Raises:
+        HTTPException: An error with status code 404 if no metrics are found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        return metric_service.get_all_metric()
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(status_code=404, detail=e.msg)
+        elif e.error_code == "ValidationError":
+            raise HTTPException(status_code=400, detail=e.msg)
+        else:
+            raise HTTPException(status_code=500, detail=e.msg)
+
+
+@router.delete("/api/v1/metrics/{metric_id}")
+@inject
+def delete_metric(
+    metric_id: str,
+    metric_service: MetricService = Depends(Provide[Container.metric_service]),
+) -> dict[str, str] | tuple[dict[str, str], int]:
+    """
+    Delete a metric from the database by its ID.
+
+    Args:
+        metric_id (str): The unique identifier of the metric to delete.
+        metric_service (MetricService): The service responsible for deleting the metric.
+
+    Returns:
+        dict[str, str] | tuple[dict[str, str], int]: A message indicating the successful deletion of the metric,
+        or an HTTPException with an appropriate status code.
+
+    Raises:
+        HTTPException: An error with status code 404 if the metric is not found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        metric_service.delete_metric(metric_id)
+        return {"message": "Metric deleted successfully"}
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to delete metric: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to delete metric: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to delete metric: {e.msg}"
+            )

moonshot/integrations/web_api/routes/prompt_template.py

@@ -0,0 +1,128 @@
+from dependency_injector.wiring import Provide, inject
+from fastapi import APIRouter, Depends, HTTPException
+
+from ..container import Container
+from ..schemas.prompt_template_response_model import PromptTemplatesResponseModel
+from ..services.prompt_template_service import PromptTemplateService
+from ..services.utils.exceptions_handler import ServiceException
+
+router = APIRouter(tags=["Prompt Template"])
+
+
+@router.get("/api/v1/prompt-templates")
+@inject
+def get_all_prompt_templates(
+    prompt_template_service: PromptTemplateService = Depends(
+        Provide[Container.prompt_template_service]
+    ),
+) -> PromptTemplatesResponseModel:
+    """
+    Retrieve all prompt templates from the database.
+
+    Args:
+        prompt_template_service (PromptTemplateService): The service responsible for retrieving prompt templates.
+
+    Returns:
+        PromptTemplatesResponseModel: A model representing all prompt templates.
+
+    Raises:
+        HTTPException: An error with status code 404 if no prompt templates are found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        return prompt_template_service.get_prompt_templates()
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to retrieve prompt templates: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to retrieve prompt templates: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to retrieve prompt templates: {e.msg}"
+            )
+
+
+@router.get("/api/v1/prompt-templates/name")
+@inject
+def get_all_prompt_templates_names(
+    prompt_template_service: PromptTemplateService = Depends(
+        Provide[Container.prompt_template_service]
+    ),
+) -> list[str]:
+    """
+    Retrieve the names of all prompt templates from the database.
+
+    Args:
+        prompt_template_service (PromptTemplateService): The service responsible for retrieving prompt template names.
+
+    Returns:
+        list[str]: A list of prompt template names.
+
+    Raises:
+        HTTPException: An error with status code 404 if no prompt template names are found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        return prompt_template_service.get_prompt_templates_name()
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404,
+                detail=f"Failed to retrieve prompt template names: {e.msg}",
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400,
+                detail=f"Failed to retrieve prompt template names: {e.msg}",
+            )
+        else:
+            raise HTTPException(
+                status_code=500,
+                detail=f"Failed to retrieve prompt template names: {e.msg}",
+            )
+
+@router.delete("/api/v1/prompt-templates/{prompt_template_name}")
+@inject
+def delete_prompt_template(
+    prompt_template_name: str,
+    prompt_template_service: PromptTemplateService = Depends(
+        Provide[Container.prompt_template_service]
+    ),
+) -> dict[str, bool]:
+    """
+    Delete a prompt template from the database by its name.
+
+    Args:
+        prompt_template_name (str): The name of the prompt template to delete.
+        prompt_template_service (PromptTemplateService): The service responsible for deleting the prompt template.
+
+    Returns:
+        dict[str, bool]: A dictionary with a key 'success' indicating the result of the deletion operation.
+
+    Raises:
+        HTTPException: An error with status code 404 if the prompt template is not found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        prompt_template_service.delete_prompt_template(prompt_template_name)
+        return {"success": True}
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to delete prompt template: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to delete prompt template: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to delete prompt template: {e.msg}"
+            )