Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -2,18 +2,18 @@

from __future__ import annotations

from enum import Enum
from enum import StrEnum

from pythinker_review.store.models import SEVERITY_ORDER, Finding, RunMeta, Severity


class OutputFormat(str, Enum):
class OutputFormat(StrEnum):
pretty = "pretty"
json = "json"
sarif = "sarif"


class FailOn(str, Enum):
class FailOn(StrEnum):
critical = "critical"
high = "high"
medium = "medium"
Expand Down
10 changes: 5 additions & 5 deletions packages/pythinker-review/src/pythinker_review/cli/review.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
import sys
import tomllib
from collections.abc import Callable, Sequence
from enum import Enum
from enum import StrEnum
from pathlib import Path
from typing import Any, NoReturn

Expand Down Expand Up @@ -82,22 +82,22 @@
app = typer.Typer(add_completion=False, no_args_is_help=True)


class ReviewMode(str, Enum):
class ReviewMode(StrEnum):
default = "default"
deslopify = "deslopify"


class ArtifactFormat(str, Enum):
class ArtifactFormat(StrEnum):
pretty = "pretty"
json = "json"


class DiffSide(str, Enum):
class DiffSide(StrEnum):
right = "RIGHT"
left = "LEFT"


class SimilarIssuesBackend(str, Enum):
class SimilarIssuesBackend(StrEnum):
chroma = "chroma"
lexical = "lexical"
auto = "auto"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,13 +4,13 @@

import subprocess
from dataclasses import dataclass, field
from enum import Enum
from enum import StrEnum
from pathlib import Path

_GIT_TIMEOUT_S = 20.0


class DiffMode(str, Enum):
class DiffMode(StrEnum):
base = "base"
staged = "staged"
working_tree = "working_tree"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,13 @@
from __future__ import annotations

from datetime import datetime
from enum import Enum
from enum import StrEnum
from typing import Literal, Self

from pydantic import BaseModel, ConfigDict, Field, model_validator


class Severity(str, Enum):
class Severity(StrEnum):
critical = "critical"
high = "high"
medium = "medium"
Expand All @@ -26,7 +26,7 @@ class Severity(str, Enum):
}


class Category(str, Enum):
class Category(StrEnum):
correctness = "correctness"
security = "security"
debugging = "debugging"
Expand Down
5 changes: 2 additions & 3 deletions tests/core/test_pythinkersoul_ralph_loop.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import asyncio
from collections.abc import AsyncIterator, Sequence
from pathlib import Path
from typing import Self, TypeVar
from typing import Self

import pytest
from inline_snapshot import Snapshot, snapshot
Expand Down Expand Up @@ -31,15 +31,14 @@ def approval() -> Approval:
return Approval(yolo=False)


T = TypeVar("T")
RALPH_IMAGE_URL = "https://example.com/test.png"
RALPH_IMAGE_USER_INPUT = [
TextPart(text="Check this image"),
ImageURLPart(image_url=ImageURLPart.ImageURL(url=RALPH_IMAGE_URL)),
]


def expect_snapshot(value: T, expected: Snapshot[T]) -> None:
def expect_snapshot[T](value: T, expected: Snapshot[T]) -> None:
if expected != value:
pytest.fail(f"Snapshot mismatch: {value!r} != {expected!r}")

Expand Down
Loading