Repo setup : Adding CI and pre commit

backend/app/alembic/env.py

@@ -23,9 +23,11 @@
 # my_important_option = config.get_main_option("my_important_option")
 # ... etc.
 
+
 def get_url():
     return str(settings.SQLALCHEMY_DATABASE_URI)
 
+
 def run_migrations_offline() -> None:
     """Run migrations in 'offline' mode.
 

backend/app/alembic/versions/001_added_request_log.py

@@ -13,27 +13,32 @@
 
 
 # revision identifiers, used by Alembic.
-revision: str = '001'
+revision: str = "001"
 down_revision: str | None = None
 branch_labels = None
 depends_on = None
 
 
 def upgrade() -> None:
-    op.create_table('request_log',
-    sa.Column('id', sa.Uuid(), nullable=False),
-    sa.Column('request_id', sa.Uuid(), nullable=False),
-    sa.Column('response_id', sa.Uuid(), nullable=True),
-    sa.Column('status', sa.Enum('PROCESSING','SUCCESS', 'ERROR', 'WARNING', name='requeststatus'), nullable=False, default='PROCESSING'),
-    sa.Column('request_text', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
-    sa.Column('response_text', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
-    sa.Column('inserted_at', sa.DateTime(), nullable=False),
-    sa.Column('updated_at', sa.DateTime(), nullable=False),
-    sa.PrimaryKeyConstraint('id')
+    op.create_table(
+        "request_log",
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("request_id", sa.Uuid(), nullable=False),
+        sa.Column("response_id", sa.Uuid(), nullable=True),
+        sa.Column(
+            "status",
+            sa.Enum("PROCESSING", "SUCCESS", "ERROR", "WARNING", name="requeststatus"),
+            nullable=False,
+            default="PROCESSING",
+        ),
+        sa.Column("request_text", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+        sa.Column("response_text", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
+        sa.Column("inserted_at", sa.DateTime(), nullable=False),
+        sa.Column("updated_at", sa.DateTime(), nullable=False),
+        sa.PrimaryKeyConstraint("id"),
     )
 
 
 def downgrade() -> None:
-    op.drop_table('request_log')
+    op.drop_table("request_log")
     # todo : drop requeststatus enum type
-

backend/app/alembic/versions/002_added_validator_log.py

@@ -13,26 +13,33 @@
 
 
 # revision identifiers, used by Alembic.
-revision: str = '002'
-down_revision: str = '001'
+revision: str = "002"
+down_revision: str = "001"
 branch_labels = None
 depends_on = None
 
 
 def upgrade() -> None:
-    op.create_table('validator_log',
-    sa.Column('id', sa.Uuid(), nullable=False),
-    sa.Column('request_id', sa.Uuid(), nullable=False),
-    sa.Column('name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
-    sa.Column('input', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
-    sa.Column('output', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
-    sa.Column('error', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
-    sa.Column('outcome', sa.Enum('PASS', 'FAIL', name='validatoroutcome'), nullable=False),
-    sa.Column('inserted_at', sa.DateTime(), nullable=False),
-    sa.Column('updated_at', sa.DateTime(), nullable=False),
-    sa.ForeignKeyConstraint(['request_id'], ['request_log.id'], ),
-    sa.PrimaryKeyConstraint('id')
+    op.create_table(
+        "validator_log",
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("request_id", sa.Uuid(), nullable=False),
+        sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+        sa.Column("input", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+        sa.Column("output", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
+        sa.Column("error", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
+        sa.Column(
+            "outcome", sa.Enum("PASS", "FAIL", name="validatoroutcome"), nullable=False
+        ),
+        sa.Column("inserted_at", sa.DateTime(), nullable=False),
+        sa.Column("updated_at", sa.DateTime(), nullable=False),
+        sa.ForeignKeyConstraint(
+            ["request_id"],
+            ["request_log.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
     )
 
+
 def downgrade() -> None:
-    op.drop_table('validator_log')
+    op.drop_table("validator_log")

backend/app/alembic/versions/003_added_validator_config.py

@@ -12,39 +12,47 @@
 import sqlalchemy as sa
 
 # revision identifiers, used by Alembic.
-revision: str = '003'
-down_revision: str = '002'
+revision: str = "003"
+down_revision: str = "002"
 branch_labels = None
 depends_on = None
 
 
 def upgrade() -> None:
-    op.create_table('validator_config',
-        sa.Column('id', sa.Uuid(), nullable=False),
-        sa.Column('organization_id', sa.Integer(), nullable=False),
-        sa.Column('project_id', sa.Integer(), nullable=False),
-        sa.Column('type', sa.String(), nullable=False),
-        sa.Column('stage', sa.String(), nullable=False),
-        sa.Column('on_fail_action', sa.String(), nullable=False),
+    op.create_table(
+        "validator_config",
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("organization_id", sa.Integer(), nullable=False),
+        sa.Column("project_id", sa.Integer(), nullable=False),
+        sa.Column("type", sa.String(), nullable=False),
+        sa.Column("stage", sa.String(), nullable=False),
+        sa.Column("on_fail_action", sa.String(), nullable=False),
         sa.Column(
             "config",
             postgresql.JSONB(astext_type=sa.Text()),
             nullable=False,
             server_default=sa.text("'{}'::jsonb"),
         ),
-        sa.Column('is_enabled', sa.Boolean(), nullable=False, server_default=sa.true()),
-        sa.Column('created_at', sa.DateTime(), nullable=False),
-        sa.Column('updated_at', sa.DateTime(), nullable=False),
-
-        sa.PrimaryKeyConstraint('id'),
-        sa.UniqueConstraint('organization_id', 'project_id', 'type', 'stage', name='uq_validator_identity')
+        sa.Column("is_enabled", sa.Boolean(), nullable=False, server_default=sa.true()),
+        sa.Column("created_at", sa.DateTime(), nullable=False),
+        sa.Column("updated_at", sa.DateTime(), nullable=False),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint(
+            "organization_id",
+            "project_id",
+            "type",
+            "stage",
+            name="uq_validator_identity",
+        ),
     )
 
-    op.create_index("idx_validator_organization", "validator_config", ["organization_id"])
+    op.create_index(
+        "idx_validator_organization", "validator_config", ["organization_id"]
+    )
     op.create_index("idx_validator_project", "validator_config", ["project_id"])
     op.create_index("idx_validator_type", "validator_config", ["type"])
     op.create_index("idx_validator_stage", "validator_config", ["stage"])
 
 
 def downgrade() -> None:
-    op.drop_table('validator_config')
+    op.drop_table("validator_config")

backend/app/api/deps.py

@@ -8,13 +8,16 @@
 from app.core.config import settings
 from app.core.db import engine
 
+
 def get_db() -> Generator[Session, None, None]:
     with Session(engine) as session:
         yield session
 
+
 SessionDep = Annotated[Session, Depends(get_db)]
 security = HTTPBearer(auto_error=False)
 
+
 def verify_bearer_token(
     credentials: Annotated[
         HTTPAuthorizationCredentials | None,
@@ -37,4 +40,5 @@ def verify_bearer_token(
 
     return True
 
+
 AuthDep = Annotated[bool, Depends(verify_bearer_token)]

backend/app/api/routes/guardrails.py

@@ -11,16 +11,16 @@
 from app.crud.request_log import RequestLogCrud
 from app.crud.validator_log import ValidatorLogCrud
 from app.schemas.guardrail_config import GuardrailRequest, GuardrailResponse
-from app.models.logging.request_log import  RequestLogUpdate, RequestStatus
+from app.models.logging.request_log import RequestLogUpdate, RequestStatus
 from app.models.logging.validator_log import ValidatorLog, ValidatorOutcome
 from app.utils import APIResponse
 
 router = APIRouter(prefix="/guardrails", tags=["guardrails"])
 
+
 @router.post(
-        "/",
-        response_model=APIResponse[GuardrailResponse],
-        response_model_exclude_none=True)
+    "/", response_model=APIResponse[GuardrailResponse], response_model_exclude_none=True
+)
 async def run_guardrails(
     payload: GuardrailRequest,
     session: SessionDep,
@@ -35,16 +35,17 @@ async def run_guardrails(
     except ValueError:
         return APIResponse.failure_response(error="Invalid request_id")
 
-    request_log = request_log_crud.create(request_id, input_text=payload.input)    
+    request_log = request_log_crud.create(request_id, input_text=payload.input)
     return await _validate_with_guard(
         payload.input,
         payload.validators,
         request_log_crud,
         request_log.id,
         validator_log_crud,
-        suppress_pass_logs
+        suppress_pass_logs,
     )
 
+
 @router.get("/")
 async def list_validators(_: AuthDep):
     """
@@ -57,10 +58,12 @@ async def list_validators(_: AuthDep):
         try:
             schema = model.model_json_schema()
             validator_type = schema["properties"]["type"]["const"]
-            validators.append({
-                "type": validator_type,
-                "config": schema,
-            })
+            validators.append(
+                {
+                    "type": validator_type,
+                    "config": schema,
+                }
+            )
 
         except (KeyError, TypeError) as e:
             return APIResponse.failure_response(
@@ -69,6 +72,7 @@ async def list_validators(_: AuthDep):
 
     return {"validators": validators}
 
+
 async def _validate_with_guard(
     data: str,
     validators: list,
@@ -84,7 +88,7 @@ async def _validate_with_guard(
     This function treats validation failures as first-class outcomes (not exceptions),
     while still safely handling unexpected runtime errors.
     """
-    response_id = uuid.uuid4() 
+    response_id = uuid.uuid4()
     guard: Guard | None = None
 
     def _finalize(
@@ -115,11 +119,12 @@ def _finalize(
         )
 
         if guard is not None:
-            add_validator_logs(guard, request_log_id, validator_log_crud, suppress_pass_logs)
+            add_validator_logs(
+                guard, request_log_id, validator_log_crud, suppress_pass_logs
+            )
 
-        rephrase_needed = (
-            validated_output is not None
-            and validated_output.startswith(REPHRASE_ON_FAIL_PREFIX)
+        rephrase_needed = validated_output is not None and validated_output.startswith(
+            REPHRASE_ON_FAIL_PREFIX
         )
 
         response_model = GuardrailResponse(
@@ -160,7 +165,13 @@ def _finalize(
             error_message=str(exc),
         )
 
-def add_validator_logs(guard: Guard, request_log_id: UUID, validator_log_crud: ValidatorLogCrud, suppress_pass_logs: bool = False):
+
+def add_validator_logs(
+    guard: Guard,
+    request_log_id: UUID,
+    validator_log_crud: ValidatorLogCrud,
+    suppress_pass_logs: bool = False,
+):
     history = getattr(guard, "history", None)
     if not history:
         return

backend/app/api/routes/utils.py

@@ -2,6 +2,7 @@
 
 router = APIRouter(prefix="/utils", tags=["utils"])
 
+
 @router.get("/health-check/")
 async def health_check() -> bool:
-    return True
+    return True