julee 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

julee/domain/models/document/tests/test_document.py

@@ -19,11 +19,14 @@ Design decisions documented:
 - Documents act as readable streams with standard methods
 """
 
-import pytest
 import json
 
+import pytest
+from pydantic import ValidationError
+
 from julee.domain.models.document import Document
-from .factories import DocumentFactory, ContentStreamFactory
+
+from .factories import ContentStreamFactory, DocumentFactory
 
 
 class TestDocumentInstantiation:
@@ -164,7 +167,7 @@ class TestDocumentInstantiation:
             assert doc.content_multihash.strip() == multihash.strip()
         else:
             # Should raise validation error
-            with pytest.raises(Exception):  # Could be ValueError or ValidationError
+            with pytest.raises((ValueError, ValidationError)):
                 Document(
                     document_id=document_id,
                     original_filename=original_filename,

julee/domain/models/knowledge_service_config/knowledge_service_config.py

@@ -13,11 +13,11 @@ All domain models use Pydantic BaseModel for validation, serialization,
 and type safety, following the patterns established in the sample project.
 """
 
-from pydantic import BaseModel, Field, field_validator
-from typing import Optional
 from datetime import datetime, timezone
 from enum import Enum
 
+from pydantic import BaseModel, Field, field_validator
+
 
 class ServiceApi(str, Enum):
     """Supported knowledge service APIs."""
@@ -48,10 +48,10 @@ class KnowledgeServiceConfig(BaseModel):
     )
 
     # Timestamps
-    created_at: Optional[datetime] = Field(
+    created_at: datetime | None = Field(
         default_factory=lambda: datetime.now(timezone.utc)
     )
-    updated_at: Optional[datetime] = Field(
+    updated_at: datetime | None = Field(
         default_factory=lambda: datetime.now(timezone.utc)
     )
 

julee/domain/models/policy/__init__.py

@@ -1,11 +1,11 @@
-from .policy import (
-    Policy,
-    PolicyStatus,
-)
 from .document_policy_validation import (
     DocumentPolicyValidation,
     DocumentPolicyValidationStatus,
 )
+from .policy import (
+    Policy,
+    PolicyStatus,
+)
 
 __all__ = [
     "Policy",

julee/domain/models/policy/document_policy_validation.py

@@ -17,11 +17,11 @@ All domain models use Pydantic BaseModel for validation, serialization,
 and type safety, following the patterns established in the sample project.
 """
 
-from pydantic import BaseModel, Field, field_validator
-from typing import Optional, List, Tuple
 from datetime import datetime, timezone
 from enum import Enum
 
+from pydantic import BaseModel, Field, field_validator
+
 
 class DocumentPolicyValidationStatus(str, Enum):
     """Status of a document policy validation process."""
@@ -70,7 +70,7 @@ class DocumentPolicyValidation(BaseModel):
     status: DocumentPolicyValidationStatus = DocumentPolicyValidationStatus.PENDING
 
     # Initial validation results
-    validation_scores: List[Tuple[str, int]] = Field(
+    validation_scores: list[tuple[str, int]] = Field(
         default_factory=list,
         description="List of (knowledge_service_query_id, actual_score) "
         "tuples representing the scores achieved during initial validation. "
@@ -78,13 +78,13 @@ class DocumentPolicyValidation(BaseModel):
     )
 
     # Transformation results (if applicable)
-    transformed_document_id: Optional[str] = Field(
+    transformed_document_id: str | None = Field(
         default=None,
         description="ID of the document after transformations have been "
         "applied. Only present if the policy includes transformation queries "
         "and they were executed",
     )
-    post_transform_validation_scores: Optional[List[Tuple[str, int]]] = Field(
+    post_transform_validation_scores: list[tuple[str, int]] | None = Field(
         default=None,
         description="List of (knowledge_service_query_id, actual_score) "
         "tuples representing scores achieved after transformation. "
@@ -93,19 +93,19 @@ class DocumentPolicyValidation(BaseModel):
     )
 
     # Validation metadata
-    started_at: Optional[datetime] = Field(
+    started_at: datetime | None = Field(
         default_factory=lambda: datetime.now(timezone.utc),
         description="When the validation process was initiated",
     )
-    completed_at: Optional[datetime] = Field(
+    completed_at: datetime | None = Field(
         default=None, description="When the validation process completed"
     )
-    error_message: Optional[str] = Field(
+    error_message: str | None = Field(
         default=None, description="Error message if validation process failed"
     )
 
     # Results summary
-    passed: Optional[bool] = Field(
+    passed: bool | None = Field(
         default=None,
         description="Whether the document passed policy validation. "
         "None while validation is in progress, True/False when complete",
@@ -128,8 +128,8 @@ class DocumentPolicyValidation(BaseModel):
     @field_validator("validation_scores")
     @classmethod
     def validation_scores_must_be_valid(
-        cls, v: List[Tuple[str, int]]
-    ) -> List[Tuple[str, int]]:
+        cls, v: list[tuple[str, int]]
+    ) -> list[tuple[str, int]]:
         if not isinstance(v, list):
             raise ValueError("Validation scores must be a list")
 
@@ -142,8 +142,8 @@ class DocumentPolicyValidation(BaseModel):
     @field_validator("post_transform_validation_scores")
     @classmethod
     def post_transform_scores_must_be_valid(
-        cls, v: Optional[List[Tuple[str, int]]]
-    ) -> Optional[List[Tuple[str, int]]]:
+        cls, v: list[tuple[str, int]] | None
+    ) -> list[tuple[str, int]] | None:
         if v is None:
             return v
 
@@ -158,7 +158,7 @@ class DocumentPolicyValidation(BaseModel):
 
     @field_validator("error_message")
     @classmethod
-    def error_message_must_be_valid(cls, v: Optional[str]) -> Optional[str]:
+    def error_message_must_be_valid(cls, v: str | None) -> str | None:
         if v is None:
             return v
         if not isinstance(v, str):
@@ -167,7 +167,7 @@ class DocumentPolicyValidation(BaseModel):
 
     @field_validator("transformed_document_id")
     @classmethod
-    def transformed_document_id_must_be_valid(cls, v: Optional[str]) -> Optional[str]:
+    def transformed_document_id_must_be_valid(cls, v: str | None) -> str | None:
         if v is None:
             return v
         if not isinstance(v, str) or not v.strip():
@@ -178,8 +178,8 @@ class DocumentPolicyValidation(BaseModel):
 
     @classmethod
     def _validate_score_tuples(
-        cls, scores: List[Tuple[str, int]], field_name: str
-    ) -> List[Tuple[str, int]]:
+        cls, scores: list[tuple[str, int]], field_name: str
+    ) -> list[tuple[str, int]]:
         """Helper method to validate score tuple lists."""
         validated_scores = []
         query_ids_seen = set()

julee/domain/models/policy/policy.py

@@ -13,11 +13,11 @@ All domain models use Pydantic BaseModel for validation, serialization,
 and type safety, following the patterns established in the sample project.
 """
 
-from pydantic import BaseModel, Field, field_validator
-from typing import Optional, List, Tuple
 from datetime import datetime, timezone
 from enum import Enum
 
+from pydantic import BaseModel, Field, field_validator
+
 
 class PolicyStatus(str, Enum):
     """Status of a policy configuration."""
@@ -55,12 +55,12 @@ class Policy(BaseModel):
 
     # Policy configuration
     status: PolicyStatus = PolicyStatus.ACTIVE
-    validation_scores: List[Tuple[str, int]] = Field(
+    validation_scores: list[tuple[str, int]] = Field(
         description="List of (knowledge_service_query_id, required_score) "
         "tuples where required_score is between 0 and 100. All scores "
         "must be met or exceeded for the policy to pass"
     )
-    transformation_queries: Optional[List[str]] = Field(
+    transformation_queries: list[str] | None = Field(
         default=None,
         description="Optional list of knowledge service query IDs for "
         "transformations to apply before re-validation. If not provided "
@@ -69,10 +69,10 @@ class Policy(BaseModel):
 
     # Policy metadata
     version: str = Field(default="0.1.0", description="Policy version")
-    created_at: Optional[datetime] = Field(
+    created_at: datetime | None = Field(
         default_factory=lambda: datetime.now(timezone.utc)
     )
-    updated_at: Optional[datetime] = Field(default=None)
+    updated_at: datetime | None = Field(default=None)
 
     @field_validator("policy_id")
     @classmethod
@@ -98,8 +98,8 @@ class Policy(BaseModel):
     @field_validator("validation_scores")
     @classmethod
     def validation_scores_must_be_valid(
-        cls, v: List[Tuple[str, int]]
-    ) -> List[Tuple[str, int]]:
+        cls, v: list[tuple[str, int]]
+    ) -> list[tuple[str, int]]:
         if not isinstance(v, list):
             raise ValueError("Validation scores must be a list")
 
@@ -147,8 +147,8 @@ class Policy(BaseModel):
     @field_validator("transformation_queries")
     @classmethod
     def transformation_queries_must_be_valid(
-        cls, v: Optional[List[str]]
-    ) -> Optional[List[str]]:
+        cls, v: list[str] | None
+    ) -> list[str] | None:
         if v is None:
             return v
 

julee/domain/models/policy/tests/factories.py

@@ -6,9 +6,10 @@ Policy domain objects with sensible defaults.
 """
 
 from datetime import datetime, timezone
+
 from factory.base import Factory
-from factory.faker import Faker
 from factory.declarations import LazyFunction
+from factory.faker import Faker
 
 from julee.domain.models.policy import (
     DocumentPolicyValidation,

julee/domain/models/policy/tests/test_document_policy_validation.py

@@ -13,14 +13,16 @@ Tests focus on:
 - Edge cases and error conditions
 """
 
-import pytest
 from datetime import datetime, timezone
+
+import pytest
 from pydantic import ValidationError
 
 from julee.domain.models.policy import (
     DocumentPolicyValidation,
     DocumentPolicyValidationStatus,
 )
+
 from .factories import DocumentPolicyValidationFactory
 
 

julee/domain/models/policy/tests/test_policy.py

@@ -5,8 +5,9 @@ These tests verify the behavior of Policy domain objects,
 including validation, serialization, and business logic.
 """
 
-import pytest
 from datetime import datetime, timezone
+
+import pytest
 from pydantic import ValidationError
 
 from julee.domain.models.policy import (

julee/domain/repositories/__init__.py

@@ -6,14 +6,14 @@ Extract, Assemble, Publish workflow, following the Clean Architecture
 patterns established in the Fun-Police framework.
 """
 
-from .base import BaseRepository
-from .document import DocumentRepository
 from .assembly import AssemblyRepository
 from .assembly_specification import AssemblySpecificationRepository
+from .base import BaseRepository
+from .document import DocumentRepository
+from .document_policy_validation import DocumentPolicyValidationRepository
 from .knowledge_service_config import KnowledgeServiceConfigRepository
 from .knowledge_service_query import KnowledgeServiceQueryRepository
 from .policy import PolicyRepository
-from .document_policy_validation import DocumentPolicyValidationRepository
 
 __all__ = [
     "BaseRepository",

julee/domain/repositories/assembly.py

@@ -27,8 +27,10 @@ In Temporal workflow contexts, these protocols are implemented by workflow
 stubs that delegate to activities for durability and proper error handling.
 """
 
-from typing import runtime_checkable, Protocol
+from typing import Protocol, runtime_checkable
+
 from julee.domain.models import Assembly
+
 from .base import BaseRepository
 
 

julee/domain/repositories/assembly_specification.py

@@ -30,9 +30,11 @@ stubs that delegate to activities for durability and proper error handling.
 """
 
 from typing import Protocol, runtime_checkable
+
 from julee.domain.models.assembly_specification import (
     AssemblySpecification,
 )
+
 from .base import BaseRepository
 
 

julee/domain/repositories/base.py

@@ -23,7 +23,8 @@ In Temporal workflow contexts, these protocols are implemented by workflow
 stubs that delegate to activities for durability and proper error handling.
 """
 
-from typing import Protocol, Optional, runtime_checkable, TypeVar, List, Dict
+from typing import Protocol, TypeVar, runtime_checkable
+
 from pydantic import BaseModel
 
 # Type variable bound to Pydantic BaseModel for domain entities
@@ -42,7 +43,7 @@ class BaseRepository(Protocol[T]):
         T: The domain entity type (must extend Pydantic BaseModel)
     """
 
-    async def get(self, entity_id: str) -> Optional[T]:
+    async def get(self, entity_id: str) -> T | None:
         """Retrieve an entity by ID.
 
         Args:
@@ -60,7 +61,7 @@ class BaseRepository(Protocol[T]):
         """
         ...
 
-    async def get_many(self, entity_ids: List[str]) -> Dict[str, Optional[T]]:
+    async def get_many(self, entity_ids: list[str]) -> dict[str, T | None]:
         """Retrieve multiple entities by ID.
 
         Args:
@@ -103,7 +104,7 @@ class BaseRepository(Protocol[T]):
         """
         ...
 
-    async def list_all(self) -> List[T]:
+    async def list_all(self) -> list[T]:
         """List all entities.
 
         Returns:

julee/domain/repositories/document.py

@@ -29,8 +29,10 @@ In Temporal workflow contexts, these protocols are implemented by workflow
 stubs that delegate to activities for durability and proper error handling.
 """
 
-from typing import runtime_checkable, Protocol
+from typing import Protocol, runtime_checkable
+
 from julee.domain.models import Document
+
 from .base import BaseRepository
 
 

julee/domain/repositories/document_policy_validation.py

@@ -29,8 +29,10 @@ In Temporal workflow contexts, these protocols are implemented by workflow
 stubs that delegate to activities for durability and proper error handling.
 """
 
-from typing import runtime_checkable, Protocol
+from typing import Protocol, runtime_checkable
+
 from julee.domain.models.policy import DocumentPolicyValidation
+
 from .base import BaseRepository
 
 

julee/domain/repositories/knowledge_service_config.py

@@ -31,9 +31,11 @@ stubs that delegate to activities for durability and proper error handling.
 """
 
 from typing import Protocol, runtime_checkable
+
 from julee.domain.models.knowledge_service_config import (
     KnowledgeServiceConfig,
 )
+
 from .base import BaseRepository
 
 

julee/domain/repositories/knowledge_service_query.py

@@ -25,6 +25,7 @@ from typing import Protocol, runtime_checkable
 from julee.domain.models.assembly_specification import (
     KnowledgeServiceQuery,
 )
+
 from .base import BaseRepository
 
 

julee/domain/repositories/policy.py

@@ -29,8 +29,10 @@ In Temporal workflow contexts, these protocols are implemented by workflow
 stubs that delegate to activities for durability and proper error handling.
 """
 
-from typing import runtime_checkable, Protocol
+from typing import Protocol, runtime_checkable
+
 from julee.domain.models import Policy
+
 from .base import BaseRepository
 
 

julee/domain/use_cases/decorators.py

@@ -7,8 +7,9 @@ following the patterns established in the sample use cases.
 """
 
 import logging
+from collections.abc import Awaitable, Callable
 from functools import wraps
-from typing import Any, Callable, Dict, Optional, TypeVar, Awaitable
+from typing import Any, TypeVar
 
 logger = logging.getLogger(__name__)
 
@@ -17,7 +18,7 @@ F = TypeVar("F", bound=Callable[..., Awaitable[Any]])
 
 def try_use_case_step(
     step_name: str,
-    extra_context: Optional[Dict[str, Any]] = None,
+    extra_context: dict[str, Any] | None = None,
 ) -> Callable[[F], F]:
     """
     Decorator that wraps use case steps with consistent error handling and

julee/domain/use_cases/extract_assemble_data.py

@@ -10,8 +10,9 @@ instances following the Clean Architecture principles.
 import hashlib
 import json
 import logging
+from collections.abc import Callable
 from datetime import datetime, timezone
-from typing import Any, Callable, Dict
+from typing import Any
 
 import jsonpointer  # type: ignore
 import jsonschema
@@ -256,8 +257,8 @@ class ExtractAssembleDataUseCase:
     async def _register_document_with_services(
         self,
         document: Document,
-        queries: Dict[str, KnowledgeServiceQuery],
-    ) -> Dict[str, str]:
+        queries: dict[str, KnowledgeServiceQuery],
+    ) -> dict[str, str]:
         """
         Register the document with all knowledge services needed for assembly.
 
@@ -301,7 +302,7 @@ class ExtractAssembleDataUseCase:
     @try_use_case_step("queries_retrieval")
     async def _retrieve_all_queries(
         self, assembly_specification: AssemblySpecification
-    ) -> Dict[str, KnowledgeServiceQuery]:
+    ) -> dict[str, KnowledgeServiceQuery]:
         """Retrieve all knowledge service queries needed for this assembly."""
         query_ids = list(assembly_specification.knowledge_service_queries.values())
 
@@ -351,8 +352,8 @@ class ExtractAssembleDataUseCase:
         self,
         document: Document,
         assembly_specification: AssemblySpecification,
-        document_registrations: Dict[str, str],
-        queries: Dict[str, KnowledgeServiceQuery],
+        document_registrations: dict[str, str],
+        queries: dict[str, KnowledgeServiceQuery],
     ) -> str:
         """
         Perform a single assembly iteration using knowledge services.
@@ -379,7 +380,7 @@ class ExtractAssembleDataUseCase:
 
         """
         # Initialize the result data structure
-        assembled_data: Dict[str, Any] = {}
+        assembled_data: dict[str, Any] = {}
 
         # Process each knowledge service query
         # TODO: This is where we may want to fan-out/fan-in to do these
@@ -470,7 +471,7 @@ class ExtractAssembleDataUseCase:
         return document
 
     def _extract_schema_section(
-        self, jsonschema: Dict[str, Any], schema_pointer: str
+        self, jsonschema: dict[str, Any], schema_pointer: str
     ) -> Any:
         """Extract relevant section of JSON schema using JSON Pointer."""
         if not schema_pointer:
@@ -495,7 +496,7 @@ Please structure your response according to this JSON schema:
 Return only valid JSON that conforms to this schema, without any surrounding
 text or markdown formatting."""
 
-    def _parse_query_result(self, result_data: Dict[str, Any]) -> Any:
+    def _parse_query_result(self, result_data: dict[str, Any]) -> Any:
         """Parse the query result to extract the JSON response."""
         response_text = result_data.get("response", "")
         if not response_text:
@@ -514,7 +515,7 @@ text or markdown formatting."""
 
     def _store_result_in_assembled_data(
         self,
-        assembled_data: Dict[str, Any],
+        assembled_data: dict[str, Any],
         schema_pointer: str,
         result_data: Any,
     ) -> None:
@@ -572,7 +573,7 @@ text or markdown formatting."""
     @try_use_case_step("assembled_document_creation")
     async def _create_assembled_document(
         self,
-        assembled_data: Dict[str, Any],
+        assembled_data: dict[str, Any],
         assembly_specification: AssemblySpecification,
     ) -> str:
         """Create and store the assembled document."""
@@ -605,7 +606,7 @@ text or markdown formatting."""
 
     def _validate_assembled_data(
         self,
-        assembled_data: Dict[str, Any],
+        assembled_data: dict[str, Any],
         assembly_specification: AssemblySpecification,
     ) -> None:
         """Validate that the assembled data conforms to the JSON schema."""

julee/domain/use_cases/initialize_system_data.py

@@ -16,7 +16,7 @@ import hashlib
 import logging
 from datetime import datetime, timezone
 from pathlib import Path
-from typing import Any, Dict, List
+from typing import Any
 
 import yaml
 
@@ -192,7 +192,7 @@ class InitializeSystemDataUseCase:
             )
             raise
 
-    def _load_fixture_configurations(self) -> List[Dict[str, Any]]:
+    def _load_fixture_configurations(self) -> list[dict[str, Any]]:
         """
         Load knowledge service configurations from the YAML fixture file.
 
@@ -217,7 +217,7 @@ class InitializeSystemDataUseCase:
             )
 
         try:
-            with open(fixture_path, "r", encoding="utf-8") as f:
+            with open(fixture_path, encoding="utf-8") as f:
                 fixture_data = yaml.safe_load(f)
 
             if not fixture_data or "knowledge_services" not in fixture_data:
@@ -240,7 +240,7 @@ class InitializeSystemDataUseCase:
             raise yaml.YAMLError(f"Invalid YAML in fixture file: {e}")
 
     def _create_config_from_fixture_data(
-        self, config_data: Dict[str, Any]
+        self, config_data: dict[str, Any]
     ) -> KnowledgeServiceConfig:
         """
         Create a KnowledgeServiceConfig from fixture data.
@@ -363,7 +363,7 @@ class InitializeSystemDataUseCase:
             )
             raise
 
-    def _load_fixture_queries(self) -> List[Dict[str, Any]]:
+    def _load_fixture_queries(self) -> list[dict[str, Any]]:
         """
         Load knowledge service queries from the YAML fixture file.
 
@@ -388,7 +388,7 @@ class InitializeSystemDataUseCase:
             )
 
         try:
-            with open(fixture_path, "r", encoding="utf-8") as f:
+            with open(fixture_path, encoding="utf-8") as f:
                 fixture_data = yaml.safe_load(f)
 
             if not fixture_data or "knowledge_service_queries" not in fixture_data:
@@ -413,7 +413,7 @@ class InitializeSystemDataUseCase:
             raise yaml.YAMLError(f"Invalid YAML in queries fixture file: {e}")
 
     def _create_query_from_fixture_data(
-        self, query_data: Dict[str, Any]
+        self, query_data: dict[str, Any]
     ) -> KnowledgeServiceQuery:
         """
         Create a KnowledgeServiceQuery from fixture data.
@@ -533,7 +533,7 @@ class InitializeSystemDataUseCase:
             )
             raise
 
-    def _load_fixture_assembly_specifications(self) -> List[Dict[str, Any]]:
+    def _load_fixture_assembly_specifications(self) -> list[dict[str, Any]]:
         """
         Load assembly specifications from the YAML fixture file.
 
@@ -558,7 +558,7 @@ class InitializeSystemDataUseCase:
             )
 
         try:
-            with open(fixture_path, "r", encoding="utf-8") as f:
+            with open(fixture_path, encoding="utf-8") as f:
                 fixture_data = yaml.safe_load(f)
 
             if not fixture_data or "assembly_specifications" not in fixture_data:
@@ -586,7 +586,7 @@ class InitializeSystemDataUseCase:
             )
 
     def _create_assembly_spec_from_fixture_data(
-        self, spec_data: Dict[str, Any]
+        self, spec_data: dict[str, Any]
     ) -> AssemblySpecification:
         """
         Create an AssemblySpecification from fixture data.
@@ -719,7 +719,7 @@ class InitializeSystemDataUseCase:
             )
             raise
 
-    def _load_fixture_documents(self) -> List[Dict[str, Any]]:
+    def _load_fixture_documents(self) -> list[dict[str, Any]]:
         """
         Load documents from the YAML fixture file.
 
@@ -742,7 +742,7 @@ class InitializeSystemDataUseCase:
             raise FileNotFoundError(f"Documents fixture file not found: {fixture_path}")
 
         try:
-            with open(fixture_path, "r", encoding="utf-8") as f:
+            with open(fixture_path, encoding="utf-8") as f:
                 fixture_data = yaml.safe_load(f)
 
             if not fixture_data or "documents" not in fixture_data:
@@ -764,7 +764,7 @@ class InitializeSystemDataUseCase:
         except yaml.YAMLError as e:
             raise yaml.YAMLError(f"Invalid YAML in documents fixture file: {e}")
 
-    def _create_document_from_fixture_data(self, doc_data: Dict[str, Any]) -> Document:
+    def _create_document_from_fixture_data(self, doc_data: dict[str, Any]) -> Document:
         """
         Create a Document from fixture data.