testing changes

.dockerignore

@@ -5,16 +5,31 @@
 !client.py
 !compat.py
 !inference.py
-!models.py
+!openenv_models.py
 !openenv.yaml
 !pyproject.toml
 !DEMO_SCRIPT.md
+!launch.py
 !triage.py
 !triage_catalog.py
 !triage_models.py
+!api/
+!api/**
+!app/
+!app/**
+!analyzers/
+!analyzers/**
+!models/
+!models/**
+!schemas/
+!schemas/**
 !server/
 !server/**
+!services/
+!services/**
 !tasks/
 !tasks/**
+!utils/
+!utils/**
 !graders/
 !graders/**

__init__.py

@@ -1,6 +1,7 @@
 """Public package exports for python_code_review_env."""
 
 from .client import PythonCodeReviewEnv, PythonEnv
+from .models import PyTorchCodeAnalyzerModel
 from .openenv_models import (
     PythonAction,
     PythonCodeReviewAction,
@@ -9,6 +10,8 @@ from .openenv_models import (
     PythonObservation,
     PythonState,
 )
+from .schemas import AnalyzeCodeRequest, AnalyzeCodeResponse
+from .services import AnalysisService
 from .triage import CodeTriageEngine, HashingEmbeddingBackend, TransformersEmbeddingBackend, get_default_engine
 from .triage_models import TriageResult
 
@@ -21,8 +24,12 @@ __all__ = [
     "PythonCodeReviewState",
     "PythonCodeReviewEnv",
     "PythonEnv",
+    "AnalyzeCodeRequest",
+    "AnalyzeCodeResponse",
+    "AnalysisService",
     "CodeTriageEngine",
     "HashingEmbeddingBackend",
+    "PyTorchCodeAnalyzerModel",
     "TransformersEmbeddingBackend",
     "TriageResult",
     "get_default_engine",

tests/test_multi_domain_platform.py

@@ -0,0 +1,52 @@
+from __future__ import annotations
+
+from fastapi.testclient import TestClient
+
+from api.main import app
+from schemas.request import AnalyzeCodeRequest
+from services.analysis_service import AnalysisService
+
+
+def test_analysis_service_detects_web_code() -> None:
+    service = AnalysisService()
+    request = AnalyzeCodeRequest(
+        code="from fastapi import FastAPI\napp = FastAPI()\n\n@app.get('/health')\ndef health():\n    return {'status': 'ok'}\n",
+        domain_hint="auto",
+    )
+
+    result = service.analyze(request)
+
+    assert result.detected_domain == "web"
+    assert 0.0 <= result.score_breakdown.reward <= 1.0
+    assert len(result.improvement_plan) == 3
+
+
+def test_analysis_service_detects_dsa_code() -> None:
+    service = AnalysisService()
+    request = AnalyzeCodeRequest(
+        code="def has_pair(nums, target):\n    for i in range(len(nums)):\n        for j in range(i + 1, len(nums)):\n            if nums[i] + nums[j] == target:\n                return True\n    return False\n",
+        domain_hint="auto",
+    )
+
+    result = service.analyze(request)
+
+    assert result.detected_domain == "dsa"
+    assert result.static_analysis.time_complexity in {"O(n^2)", "O(n^3)"}
+
+
+def test_api_analyze_endpoint_returns_valid_payload() -> None:
+    client = TestClient(app)
+    response = client.post(
+        "/analyze",
+        json={
+            "code": "import torch\n\ndef predict(model, x):\n    return model(x)\n",
+            "context_window": "Inference helper for a classifier",
+            "traceback_text": "",
+            "domain_hint": "auto",
+        },
+    )
+
+    assert response.status_code == 200
+    payload = response.json()
+    assert "detected_domain" in payload
+    assert "score_breakdown" in payload