chore(Enum): Use Enum instead of plain strings

Issue: APPAI-46

Author: timothedelion

packages/gg_api_core/src/gg_api_core/tools/list_repo_incidents.py

@@ -3,14 +3,34 @@
 
 from pydantic import BaseModel, Field
 
+from gg_api_core.client import IncidentSeverity, IncidentStatus, IncidentValidity, TagNames
 from gg_api_core.utils import get_client
 
 logger = logging.getLogger(__name__)
 
-DEFAULT_EXCLUDED_TAGS = ["TEST_FILE", "FALSE_POSITIVE", "CHECK_RUN_SKIP_FALSE_POSITIVE", "CHECK_RUN_SKIP_LOW_RISK", "CHECK_RUN_SKIP_TEST_CRED"]
-DEFAULT_SEVERITIES = ["critical", "high", "medium"]
-DEFAULT_STATUSES = ["TRIGGERED", "ASSIGNED", "RESOLVED"]  # We exclude "IGNORED" ones
-DEFAULT_VALIDITIES = ["valid", "failed_to_check", "no_checker", "unknown"]  # We exclude "invalid" ones
+DEFAULT_EXCLUDED_TAGS = [
+    TagNames.TEST_FILE,
+    TagNames.FALSE_POSITIVE,
+    TagNames.CHECK_RUN_SKIP_FALSE_POSITIVE,
+    TagNames.CHECK_RUN_SKIP_LOW_RISK,
+    TagNames.CHECK_RUN_SKIP_TEST_CRED,
+]
+DEFAULT_SEVERITIES = [
+    IncidentSeverity.CRITICAL,
+    IncidentSeverity.HIGH,
+    IncidentSeverity.MEDIUM,
+]
+DEFAULT_STATUSES = [
+    IncidentStatus.TRIGGERED,
+    IncidentStatus.ASSIGNED,
+    IncidentStatus.RESOLVED,
+]  # We exclude "IGNORED" ones
+DEFAULT_VALIDITIES = [
+    IncidentValidity.VALID,
+    IncidentValidity.FAILED_TO_CHECK,
+    IncidentValidity.NO_CHECKER,
+    IncidentValidity.UNKNOWN,
+]  # We exclude "INVALID" ones
 
 
 class ListRepoIncidentsParams(BaseModel):

packages/gg_api_core/src/gg_api_core/tools/list_repo_occurrences.py

@@ -3,15 +3,36 @@
 
 from pydantic import BaseModel, Field
 
+from gg_api_core.client import IncidentSeverity, IncidentStatus, IncidentValidity, TagNames
 from gg_api_core.utils import get_client
 
 logger = logging.getLogger(__name__)
 
 
-DEFAULT_EXCLUDED_TAGS = ["TEST_FILE", "FALSE_POSITIVE", "CHECK_RUN_SKIP_FALSE_POSITIVE", "CHECK_RUN_SKIP_LOW_RISK", "CHECK_RUN_SKIP_TEST_CRED"]
-DEFAULT_SEVERITIES = ["critical", "high", "medium"]
-DEFAULT_STATUSES = ["TRIGGERED", "ASSIGNED", "RESOLVED"]  # We exclude "IGNORED" ones
-DEFAULT_VALIDITIES = ["valid", "failed_to_check", "no_checker", "unknown"]  # We exclude "invalid" ones
+DEFAULT_EXCLUDED_TAGS = [
+    TagNames.TEST_FILE,
+    TagNames.FALSE_POSITIVE,
+    TagNames.CHECK_RUN_SKIP_FALSE_POSITIVE,
+    TagNames.CHECK_RUN_SKIP_LOW_RISK,
+    TagNames.CHECK_RUN_SKIP_TEST_CRED,
+]
+DEFAULT_SEVERITIES = [
+    IncidentSeverity.CRITICAL,
+    IncidentSeverity.HIGH,
+    IncidentSeverity.MEDIUM,
+    IncidentSeverity.UNKNOWN,
+]
+DEFAULT_STATUSES = [
+    IncidentStatus.TRIGGERED,
+    IncidentStatus.ASSIGNED,
+    IncidentStatus.RESOLVED,
+]  # We exclude "IGNORED" ones
+DEFAULT_VALIDITIES = [
+    IncidentValidity.VALID,
+    IncidentValidity.FAILED_TO_CHECK,
+    IncidentValidity.NO_CHECKER,
+    IncidentValidity.UNKNOWN,
+]  # We exclude "INVALID" ones
 
 
 class ListRepoOccurrencesParams(BaseModel):

scripts/run_tool.py

@@ -6,7 +6,7 @@
 
 async def fetch_repo_occurrences():
     result = await list_repo_occurrences(
-        ListRepoOccurrencesParams(source_id="9036019")
+        ListRepoOccurrencesParams(source_id="9036019", get_all=False, status=None, severity=["critical", "high", "medium", "low", "info", "unknown"])
     )
     print(result)