Fixed error with category, added pip cache

Signed-off-by: JonahSussman <sussmanjonah@gmail.com>

.github/workflows/test-code-on-pr.yml

@@ -29,6 +29,10 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
           architecture: ${{ matrix.architecture }}
+          cache: pip
+          # After merging #225, use this as the cache instead.The cahce merely
+          # uses a hash of a specififed file to determine if it needs to update
+          # cache-dependency-path: requirements.in
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip setuptools wheel coverage

kai/service/incident_store/sql_types.py

@@ -1,7 +1,7 @@
 import datetime
-from typing import Any, Optional
+from enum import Enum
+from typing import Any, Optional, Type
 
-import sqlalchemy
 from sqlalchemy import (
     VARCHAR,
     Column,
@@ -40,6 +40,21 @@ def process_result_value(self, value: str, dialect: Dialect):
         return Solution.model_validate_json(value)
 
 
+def SQLEnum(enum_type: Type[Enum]):
+    """
+    The default behavior of the Enum type in SQLAlchemy is to store the enum's name,
+    but we want to store the enum's value. This class is a workaround for that.
+    """
+
+    return Enum(
+        value=enum_type.__name__,
+        names=[(e.value, e.value) for e in enum_type],
+    )
+
+
+SQLCategory: Type = SQLEnum(report_types.Category)
+
+
 class SQLBase(DeclarativeBase):
     type_annotation_map = {
         dict[str, Any]: JSON()
@@ -100,9 +115,7 @@ class SQLViolation(SQLBase):
         ForeignKey("rulesets.ruleset_name"), primary_key=True
     )
 
-    category = Column(
-        "category", sqlalchemy.Enum(report_types.Category).values_callable
-    )
+    category: Mapped[SQLCategory]  # type: ignore
     labels: Mapped[list[str]]
 
     ruleset: Mapped[SQLRuleset] = relationship(back_populates="violations")