xai-review 0.32.0__py3-none-any.whl → 0.33.0__py3-none-any.whl

ai_review/services/artifacts/tools.py

@@ -1,8 +1,8 @@
 import hashlib
-from datetime import datetime
+from datetime import datetime, timezone
 
 
 def make_artifact_id(text: str) -> str:
     sha = hashlib.sha1(text.encode()).hexdigest()[:8]
-    ts = datetime.utcnow().strftime("%Y.%m.%d_%H-%M-%S")
+    ts = datetime.now(timezone.utc).strftime("%Y.%m.%d_%H-%M-%S")
     return f"{ts}_{sha}"

ai_review/tests/fixtures/services/artifacts.py

@@ -2,6 +2,7 @@ from pathlib import Path
 
 import pytest
 
+from ai_review.services.artifacts.service import ArtifactsService
 from ai_review.services.artifacts.types import ArtifactsServiceProtocol
 
 
@@ -49,3 +50,8 @@ class FakeArtifactsService(ArtifactsServiceProtocol):
 @pytest.fixture
 def fake_artifacts_service() -> FakeArtifactsService:
     return FakeArtifactsService()
+
+
+@pytest.fixture
+def artifacts_service() -> ArtifactsService:
+    return ArtifactsService()

ai_review/tests/suites/services/artifacts/test_service.py

@@ -0,0 +1,92 @@
+import json
+from pathlib import Path
+
+import aiofiles
+import pytest
+
+from ai_review.config import settings
+from ai_review.libs.config.artifacts import ArtifactsConfig
+from ai_review.services.artifacts.service import ArtifactsService
+
+
+@pytest.mark.asyncio
+async def test_save_llm_interaction_creates_file(
+        tmp_path: Path,
+        monkeypatch: pytest.MonkeyPatch,
+        artifacts_service: ArtifactsService
+):
+    """Checks that a JSON file is created with the correct name and content when LLM saving is enabled."""
+    monkeypatch.setattr(settings, "artifacts", ArtifactsConfig(llm_dir=tmp_path, llm_enabled=True))
+
+    artifact_id = await artifacts_service.save_llm_interaction(
+        prompt="Hello world",
+        prompt_system="system prompt",
+        response="model answer"
+    )
+
+    assert artifact_id is not None
+    artifact_path = tmp_path / f"{artifact_id}.json"
+    assert artifact_path.exists(), "Artifact file was not created"
+
+    async with aiofiles.open(artifact_path, "r", encoding="utf-8") as file:
+        content = await file.read()
+    data = json.loads(content)
+    assert data["id"] == artifact_id
+    assert data["prompt"] == "Hello world"
+    assert data["response"] == "model answer"
+    assert data["prompt_system"] == "system prompt"
+
+
+@pytest.mark.asyncio
+async def test_save_llm_interaction_disabled(
+        monkeypatch: pytest.MonkeyPatch,
+        artifacts_service: ArtifactsService
+):
+    """Checks that the method returns None and does not create a file if LLM saving is disabled."""
+    monkeypatch.setattr(settings, "artifacts", ArtifactsConfig(llm_enabled=False))
+
+    artifact_id = await artifacts_service.save_llm_interaction(
+        prompt="ignored",
+        prompt_system="ignored",
+        response="ignored"
+    )
+
+    assert artifact_id is None
+
+
+@pytest.mark.asyncio
+async def test_save_artifact_writes_file(tmp_path: Path, artifacts_service: ArtifactsService):
+    """Checks that save_artifact writes content to the given file path."""
+    file_path = tmp_path / "test.json"
+    content = '{"key": "value"}'
+
+    result = await artifacts_service.save_artifact(file=file_path, content=content, kind="test")
+
+    assert result == file_path
+    assert file_path.exists()
+    async with aiofiles.open(file_path, "r", encoding="utf-8") as f:
+        saved_content = await f.read()
+    assert saved_content == content
+
+
+@pytest.mark.asyncio
+async def test_save_artifact_handles_exception(
+        monkeypatch: pytest.MonkeyPatch,
+        artifacts_service: ArtifactsService
+):
+    """Checks that save_artifact gracefully returns None on write error."""
+
+    class BrokenAsyncFile:
+        async def __aenter__(self):
+            raise OSError("disk full")
+
+        async def __aexit__(self, exc_type, exc_val, exc_tb):
+            return False
+
+    monkeypatch.setattr(
+        "ai_review.services.artifacts.service.aiofiles.open",
+        lambda *args, **kwargs: BrokenAsyncFile()
+    )
+
+    result = await artifacts_service.save_artifact(Path("/fake/path.json"), "data")
+    assert result is None

ai_review/tests/suites/services/artifacts/test_tools.py

@@ -0,0 +1,28 @@
+import hashlib
+import re
+
+import pytest
+
+from ai_review.services.artifacts.tools import make_artifact_id
+
+
+@pytest.mark.parametrize("text", ["hello", "some longer text", ""])
+def test_make_artifact_id_format_and_sha(text: str):
+    """Checks that the function returns a valid artifact ID format and correct SHA prefix."""
+    artifact_id = make_artifact_id(text)
+
+    pattern = r"^\d{4}\.\d{2}\.\d{2}_\d{2}-\d{2}-\d{2}_[0-9a-f]{8}$"
+    assert re.match(pattern, artifact_id), f"Invalid format: {artifact_id}"
+
+    expected_sha = hashlib.sha1(text.encode()).hexdigest()[:8]
+    assert artifact_id.endswith(expected_sha)
+
+
+def test_make_artifact_id_is_deterministic_for_same_input():
+    """Checks that the SHA part is deterministic (identical for the same input)."""
+    sha1 = hashlib.sha1("repeatable".encode()).hexdigest()[:8]
+    artifact_id1 = make_artifact_id("repeatable")
+    artifact_id2 = make_artifact_id("repeatable")
+
+    assert artifact_id1.split("_")[2] == sha1
+    assert artifact_id2.split("_")[2] == sha1

{xai_review-0.32.0.dist-info → xai_review-0.33.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xai-review
-Version: 0.32.0
+Version: 0.33.0
 Summary: AI-powered code review tool
 Author-email: Nikita Filonov <nikita.filonov@example.com>
 Maintainer-email: Nikita Filonov <nikita.filonov@example.com>
@@ -220,7 +220,7 @@ jobs:
         with:
           fetch-depth: 0
 
-      - uses: Nikita-Filonov/ai-review@v0.32.0
+      - uses: Nikita-Filonov/ai-review@v0.33.0
         with:
           review-command: ${{ inputs.review-command }}
         env:

{xai_review-0.32.0.dist-info → xai_review-0.33.0.dist-info}/RECORD

@@ -137,7 +137,7 @@ ai_review/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 ai_review/services/artifacts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/services/artifacts/schema.py,sha256=o4dqG5LFCdAQY3wjRF5rImANe-X20g4zX_bCIKiHLSk,291
 ai_review/services/artifacts/service.py,sha256=SDHwYm9I4pSPISyNWqHEOR-wTTEa5ThsIi458C9hBt8,1789
-ai_review/services/artifacts/tools.py,sha256=_ZVNwyhZDiGlPAtIgLzb7TF1zBGJy8Hq32H3xoGvOkQ,223
+ai_review/services/artifacts/tools.py,sha256=KrCQxbBEBIgqUhxZwCmbj1f07B5403gkfSgpS9qLpE4,242
 ai_review/services/artifacts/types.py,sha256=VPEDuQQciyQL8qcmgFuZxZUuuh2-xLwqwxmNZr62F3E,448
 ai_review/services/cost/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/services/cost/schema.py,sha256=K3uCIMMxGL8AaIPh4a-d0mT5uIJuk3f805DkP8o8DtY,1323
@@ -237,7 +237,7 @@ ai_review/tests/fixtures/libs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5N
 ai_review/tests/fixtures/libs/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/fixtures/libs/llm/output_json_parser.py,sha256=Eu0tCK5cD1d9X1829ruahih7jQoehGozIEugS71AAA8,275
 ai_review/tests/fixtures/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_review/tests/fixtures/services/artifacts.py,sha256=V5FvUnC9OAo0n-paxxJP5OxAgLz1Zz3OZ8zZvqu_01w,1462
+ai_review/tests/fixtures/services/artifacts.py,sha256=ETxcGT27YUSegSjxKuuXx1B8ZCXmAm0hhUzW5Vqt7Os,1621
 ai_review/tests/fixtures/services/cost.py,sha256=A6Ja0CtQ-k6pR2-B5LRE8EzkqPL34xHGXYtaILjhYvw,1612
 ai_review/tests/fixtures/services/diff.py,sha256=rOLFR-giYJlE2qUYTOT9BxyJhQ-fbXDdYCw3zed4-9M,1471
 ai_review/tests/fixtures/services/git.py,sha256=zDNNLZDoVC7r4LuF1N1MUgzhcAl2nhDdFC9olpR_PjQ,1441
@@ -308,6 +308,9 @@ ai_review/tests/suites/libs/llm/test_output_json_parser.py,sha256=2KwnXZc3dlUWM1
 ai_review/tests/suites/libs/template/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/libs/template/test_render.py,sha256=n-ss5bd_hwc-RzYmqWmFM6KSlP1zLSnlsW1Yki12Bpw,1890
 ai_review/tests/suites/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/tests/suites/services/artifacts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_review/tests/suites/services/artifacts/test_service.py,sha256=OFYHORgvKVLXrAWImTUs9nXAuAUGUVQT-QBZPU3WjVc,3054
+ai_review/tests/suites/services/artifacts/test_tools.py,sha256=kvxGoCcL6JovW10GoVycKE5M3e-fkbVd6OpHLbPvhIA,1022
 ai_review/tests/suites/services/cost/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/services/cost/test_schema.py,sha256=AI3Wg1sR6nzLpkEqJGDu6nDYwiwzbbghsxhRNwRsUFA,3044
 ai_review/tests/suites/services/cost/test_service.py,sha256=9_Mi5hu2cq3w2tIEPfhrn9x8SblCT5m1W-QUOc9BZds,3258
@@ -373,9 +376,9 @@ ai_review/tests/suites/services/vcs/github/test_client.py,sha256=mNt1bA6aVU3REsJ
 ai_review/tests/suites/services/vcs/gitlab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_review/tests/suites/services/vcs/gitlab/test_adapter.py,sha256=BYBP2g1AKF_jCSJYJj16pW7M_6PprwD9reYEpdw3StU,4340
 ai_review/tests/suites/services/vcs/gitlab/test_client.py,sha256=dnI-YxYADmVF2GS9rp6-JPkcqsn4sN8Fjbe4MkeYMaE,8476
-xai_review-0.32.0.dist-info/licenses/LICENSE,sha256=p-v8m7Kmz4KKc7PcvsGiGEmCw9AiSXY4_ylOPy_u--Y,11343
-xai_review-0.32.0.dist-info/METADATA,sha256=zRXQ50EMX8ekmcWLN4F5E5jnNxVS-62t9zbiY9FqKus,12689
-xai_review-0.32.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xai_review-0.32.0.dist-info/entry_points.txt,sha256=JyC5URanMi5io5P_PXQf7H_I1OGIpk5cZQhaPQ0g4Zs,53
-xai_review-0.32.0.dist-info/top_level.txt,sha256=sTsZbfzLoqvRZKdKa-BcxWvjlHdrpbeJ6DrGY0EuR0E,10
-xai_review-0.32.0.dist-info/RECORD,,
+xai_review-0.33.0.dist-info/licenses/LICENSE,sha256=p-v8m7Kmz4KKc7PcvsGiGEmCw9AiSXY4_ylOPy_u--Y,11343
+xai_review-0.33.0.dist-info/METADATA,sha256=olP2J_qC--KQKEyDvq1JNIUtqgd029Lli6fiI6zKAc0,12689
+xai_review-0.33.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xai_review-0.33.0.dist-info/entry_points.txt,sha256=JyC5URanMi5io5P_PXQf7H_I1OGIpk5cZQhaPQ0g4Zs,53
+xai_review-0.33.0.dist-info/top_level.txt,sha256=sTsZbfzLoqvRZKdKa-BcxWvjlHdrpbeJ6DrGY0EuR0E,10
+xai_review-0.33.0.dist-info/RECORD,,