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

julee/repositories/minio/knowledge_service_config.py

@@ -14,7 +14,6 @@ key.
 """
 
 import logging
-from typing import Optional, List, Dict
 
 from julee.domain.models.knowledge_service_config import (
     KnowledgeServiceConfig,
@@ -22,6 +21,7 @@ from julee.domain.models.knowledge_service_config import (
 from julee.domain.repositories.knowledge_service_config import (
     KnowledgeServiceConfigRepository,
 )
+
 from .client import MinioClient, MinioRepositoryMixin
 
 
@@ -53,7 +53,7 @@ class MinioKnowledgeServiceConfigRepository(
         self.bucket_name = "knowledge-service-configs"
         self.ensure_buckets_exist(self.bucket_name)
 
-    async def get(self, knowledge_service_id: str) -> Optional[KnowledgeServiceConfig]:
+    async def get(self, knowledge_service_id: str) -> KnowledgeServiceConfig | None:
         """Retrieve a knowledge service configuration by ID.
 
         Args:
@@ -98,8 +98,8 @@ class MinioKnowledgeServiceConfigRepository(
         )
 
     async def get_many(
-        self, knowledge_service_ids: List[str]
-    ) -> Dict[str, Optional[KnowledgeServiceConfig]]:
+        self, knowledge_service_ids: list[str]
+    ) -> dict[str, KnowledgeServiceConfig | None]:
         """Retrieve multiple knowledge service configs by ID.
 
         Args:
@@ -124,7 +124,7 @@ class MinioKnowledgeServiceConfigRepository(
         )
 
         # Convert object names back to knowledge service IDs for the result
-        result: Dict[str, Optional[KnowledgeServiceConfig]] = {}
+        result: dict[str, KnowledgeServiceConfig | None] = {}
         for i, service_id in enumerate(knowledge_service_ids):
             object_name = object_names[i]
             result[service_id] = object_results[object_name]
@@ -139,7 +139,7 @@ class MinioKnowledgeServiceConfigRepository(
         """
         return self.generate_id_with_prefix("ks")
 
-    async def list_all(self) -> List[KnowledgeServiceConfig]:
+    async def list_all(self) -> list[KnowledgeServiceConfig]:
         """List all knowledge service configurations.
 
         Returns:

julee/repositories/minio/knowledge_service_query.py

@@ -16,17 +16,14 @@ Each query is stored as a separate object with the query ID as the key.
 import logging
 import uuid
 
-from typing import Optional, List, Dict
-
-
 from julee.domain.models.assembly_specification import (
     KnowledgeServiceQuery,
 )
-from .client import MinioClient, MinioRepositoryMixin
 from julee.domain.repositories.knowledge_service_query import (
     KnowledgeServiceQueryRepository,
 )
 
+from .client import MinioClient, MinioRepositoryMixin
 
 logger = logging.getLogger(__name__)
 
@@ -54,7 +51,7 @@ class MinioKnowledgeServiceQueryRepository(
         self.bucket_name = "knowledge-service-queries"
         self.ensure_buckets_exist(self.bucket_name)
 
-    async def get(self, query_id: str) -> Optional[KnowledgeServiceQuery]:
+    async def get(self, query_id: str) -> KnowledgeServiceQuery | None:
         """Retrieve a knowledge service query by ID.
 
         Args:
@@ -124,8 +121,8 @@ class MinioKnowledgeServiceQueryRepository(
         return query_id
 
     async def get_many(
-        self, query_ids: List[str]
-    ) -> Dict[str, Optional[KnowledgeServiceQuery]]:
+        self, query_ids: list[str]
+    ) -> dict[str, KnowledgeServiceQuery | None]:
         """Retrieve multiple knowledge service queries by ID.
 
         Args:
@@ -159,14 +156,14 @@ class MinioKnowledgeServiceQueryRepository(
         )
 
         # Convert object names back to query IDs for the result
-        result: Dict[str, Optional[KnowledgeServiceQuery]] = {}
+        result: dict[str, KnowledgeServiceQuery | None] = {}
         for i, query_id in enumerate(query_ids):
             object_name = object_names[i]
             result[query_id] = object_results[object_name]
 
         return result
 
-    async def list_all(self) -> List[KnowledgeServiceQuery]:
+    async def list_all(self) -> list[KnowledgeServiceQuery]:
         """List all knowledge service queries.
 
         Returns:

julee/repositories/minio/policy.py

@@ -13,10 +13,10 @@ and transformation queries.
 """
 
 import logging
-from typing import Optional, List, Dict
 
 from julee.domain.models.policy import Policy
 from julee.domain.repositories.policy import PolicyRepository
+
 from .client import MinioClient, MinioRepositoryMixin
 
 
@@ -40,7 +40,7 @@ class MinioPolicyRepository(PolicyRepository, MinioRepositoryMixin):
         self.policies_bucket = "policies"
         self.ensure_buckets_exist(self.policies_bucket)
 
-    async def get(self, policy_id: str) -> Optional[Policy]:
+    async def get(self, policy_id: str) -> Policy | None:
         """Retrieve a policy by ID."""
         return self.get_json_object(
             bucket_name=self.policies_bucket,
@@ -72,7 +72,7 @@ class MinioPolicyRepository(PolicyRepository, MinioRepositoryMixin):
             },
         )
 
-    async def get_many(self, policy_ids: List[str]) -> Dict[str, Optional[Policy]]:
+    async def get_many(self, policy_ids: list[str]) -> dict[str, Policy | None]:
         """Retrieve multiple policies by ID.
 
         Args:
@@ -95,7 +95,7 @@ class MinioPolicyRepository(PolicyRepository, MinioRepositoryMixin):
         )
 
         # Convert object names back to policy IDs for the result
-        result: Dict[str, Optional[Policy]] = {}
+        result: dict[str, Policy | None] = {}
         for policy_id in policy_ids:
             result[policy_id] = object_results[policy_id]
 

julee/repositories/minio/tests/fake_client.py

@@ -7,15 +7,17 @@ state-based testing where you can verify actual storage state rather than
 just mocking method calls.
 """
 
-from typing import Dict, Any, Optional, Callable, BinaryIO, Union, List
+from collections.abc import Callable
+from datetime import datetime, timezone
 from functools import wraps
+from typing import Any, BinaryIO
 from unittest.mock import Mock
-from datetime import datetime, timezone
-from minio.error import S3Error
-from minio.datatypes import Object
+
 from minio.api import ObjectWriteResult
-from urllib3.response import BaseHTTPResponse
+from minio.datatypes import Object
+from minio.error import S3Error
 from urllib3 import HTTPHeaderDict
+from urllib3.response import BaseHTTPResponse
 
 from ..client import MinioClient
 
@@ -83,8 +85,8 @@ class FakeMinioClient(MinioClient):
     """
 
     def __init__(self) -> None:
-        self._buckets: Dict[str, Dict[str, Any]] = {}
-        self._objects: Dict[str, Dict[str, Dict[str, Any]]] = {}
+        self._buckets: dict[str, dict[str, Any]] = {}
+        self._objects: dict[str, dict[str, dict[str, Any]]] = {}
 
     def bucket_exists(self, bucket_name: str) -> bool:
         """Check if a bucket exists."""
@@ -112,7 +114,7 @@ class FakeMinioClient(MinioClient):
         data: BinaryIO,
         length: int,
         content_type: str = "application/octet-stream",
-        metadata: Optional[Dict[str, Union[str, List[str], tuple[str]]]] = None,
+        metadata: dict[str, str | list[str] | tuple[str]] | None = None,
     ) -> ObjectWriteResult:
         """Store an object in the bucket."""
 
@@ -195,7 +197,7 @@ class FakeMinioClient(MinioClient):
         del self._objects[bucket_name][object_name]
 
     # Inspection methods for testing
-    def get_stored_objects(self, bucket_name: str) -> Dict[str, Dict[str, Any]]:
+    def get_stored_objects(self, bucket_name: str) -> dict[str, dict[str, Any]]:
         """Get all stored objects in a bucket (for testing purposes)."""
         return self._objects.get(bucket_name, {}).copy()
 

julee/repositories/minio/tests/test_assembly.py

@@ -6,11 +6,13 @@ repository implementation, using the fake client to avoid external
 dependencies during testing.
 """
 
-import pytest
 from datetime import datetime, timezone
 
+import pytest
+
 from julee.domain.models.assembly import Assembly, AssemblyStatus
 from julee.repositories.minio.assembly import MinioAssemblyRepository
+
 from .fake_client import FakeMinioClient
 
 

julee/repositories/minio/tests/test_assembly_specification.py

@@ -6,9 +6,9 @@ specification repository implementation, using the fake client to avoid
 external dependencies during testing.
 """
 
-import pytest
 from datetime import datetime, timezone
 
+import pytest
 
 from julee.domain.models.assembly_specification import (
     AssemblySpecification,
@@ -17,6 +17,7 @@ from julee.domain.models.assembly_specification import (
 from julee.repositories.minio.assembly_specification import (
     MinioAssemblySpecificationRepository,
 )
+
 from .fake_client import FakeMinioClient
 
 

julee/repositories/minio/tests/test_client_protocol.py

@@ -34,11 +34,11 @@ class TestMinioClientProtocol:
         assert hasattr(real_client, "stat_object")
 
         # Test that methods are callable
-        assert callable(getattr(real_client, "bucket_exists"))
-        assert callable(getattr(real_client, "make_bucket"))
-        assert callable(getattr(real_client, "put_object"))
-        assert callable(getattr(real_client, "get_object"))
-        assert callable(getattr(real_client, "stat_object"))
+        assert callable(real_client.bucket_exists)
+        assert callable(real_client.make_bucket)
+        assert callable(real_client.put_object)
+        assert callable(real_client.get_object)
+        assert callable(real_client.stat_object)
 
     def test_protocol_accepts_real_minio_client(self) -> None:
         """Test that our protocol accepts a real Minio client instance."""

julee/repositories/minio/tests/test_document.py

@@ -6,20 +6,21 @@ without requiring a real MinIO instance. They follow the Clean Architecture
 testing patterns and verify idempotency, error handling, and content.
 """
 
-import io
-import pytest
 import hashlib
-import multihash
+import io
 from typing import Any
 from unittest.mock import Mock
-from minio.error import S3Error
 
+import multihash
+import pytest
+from minio.error import S3Error
 
-from julee.repositories.minio.document import MinioDocumentRepository
-from julee.domain.models.document import Document, DocumentStatus
 from julee.domain.models.custom_fields.content_stream import (
     ContentStream,
 )
+from julee.domain.models.document import Document, DocumentStatus
+from julee.repositories.minio.document import MinioDocumentRepository
+
 from .fake_client import FakeMinioClient
 
 
@@ -428,16 +429,16 @@ class TestMinioDocumentRepositoryMultihash:
         assert len(multihash_result) > 0
 
 
-class TestMinioDocumentRepositoryContentString:
-    """Test content_string functionality."""
+class TestMinioDocumentRepositoryContentBytes:
+    """Test content_bytes functionality."""
 
-    async def test_save_document_with_content_string(
+    async def test_save_document_with_content_bytes(
         self, repository: MinioDocumentRepository
     ) -> None:
-        """Test saving document with content_string (small content)."""
+        """Test saving document with content_bytes (small content)."""
         content = '{"assembled": "document", "data": "test"}'
 
-        # Create document with content_string
+        # Create document with content_bytes
         document = Document(
             document_id="test-doc-content-string",
             original_filename="assembled.json",
@@ -445,27 +446,27 @@ class TestMinioDocumentRepositoryContentString:
             size_bytes=100,  # Will be updated automatically
             content_multihash="placeholder",  # Will be updated automatically
             status=DocumentStatus.CAPTURED,
-            content_string=content,
+            content_bytes=content,
         )
 
-        # Act - save should convert content_string to ContentStream
+        # Act - save should convert content_bytes to ContentStream
         await repository.save(document)
 
         # Assert document was saved successfully
         retrieved = await repository.get(document.document_id)
         assert retrieved is not None
         assert retrieved.content_multihash != "placeholder"  # Hash was calculated
-        assert retrieved.size_bytes == len(content.encode("utf-8"))
+        assert retrieved.size_bytes == len(content)
 
         # Verify content can be read
         assert retrieved.content is not None
         retrieved_content = retrieved.content.read().decode("utf-8")
         assert retrieved_content == content
 
-    async def test_save_document_with_content_string_unicode(
+    async def test_save_document_with_content_bytes_unicode(
         self, repository: MinioDocumentRepository
     ) -> None:
-        """Test saving document with unicode content_string."""
+        """Test saving document with unicode content_bytes."""
         content = '{"title": "测试文档", "emoji": "🚀", "content": "éñ"}'
 
         document = Document(
@@ -475,7 +476,7 @@ class TestMinioDocumentRepositoryContentString:
             size_bytes=100,
             content_multihash="placeholder",
             status=DocumentStatus.CAPTURED,
-            content_string=content,
+            content_bytes=content,
         )
 
         await repository.save(document)
@@ -489,12 +490,12 @@ class TestMinioDocumentRepositoryContentString:
     # Note: Empty content test removed because domain model requires
     # size_bytes > 0
 
-    async def test_save_excludes_content_string_from_metadata(
+    async def test_save_excludes_content_bytes_from_metadata(
         self,
         repository: MinioDocumentRepository,
         fake_minio_client: FakeMinioClient,
     ) -> None:
-        """Test that content_string is not stored in metadata."""
+        """Test that content_bytes is not stored in metadata."""
         content = '{"test": "data that should not be in metadata"}'
 
         document = Document(
@@ -504,7 +505,7 @@ class TestMinioDocumentRepositoryContentString:
             size_bytes=100,
             content_multihash="placeholder",
             status=DocumentStatus.CAPTURED,
-            content_string=content,
+            content_bytes=content,
         )
 
         await repository.save(document)
@@ -520,8 +521,8 @@ class TestMinioDocumentRepositoryContentString:
 
         metadata_dict = json.loads(metadata_json)
 
-        # Verify content_string is not in stored metadata
-        assert "content_string" not in metadata_dict
+        # Verify content_bytes is not in stored metadata
+        assert "content_bytes" not in metadata_dict
         assert "content" not in metadata_dict
 
         # Verify essential fields are still present

julee/repositories/minio/tests/test_document_policy_validation.py

@@ -7,9 +7,10 @@ repository that differs from the inherited mixins. Uses the fake client to
 avoid external dependencies during testing.
 """
 
-import pytest
 from datetime import datetime, timezone
 
+import pytest
+
 from julee.domain.models.policy import (
     DocumentPolicyValidation,
     DocumentPolicyValidationStatus,
@@ -17,6 +18,7 @@ from julee.domain.models.policy import (
 from julee.repositories.minio.document_policy_validation import (
     MinioDocumentPolicyValidationRepository,
 )
+
 from .fake_client import FakeMinioClient
 
 

julee/repositories/minio/tests/test_knowledge_service_config.py

@@ -6,16 +6,18 @@ configuration repository implementation, using the fake client to avoid
 external dependencies during testing.
 """
 
-import pytest
 from datetime import datetime, timezone
 
+import pytest
+
 from julee.domain.models.knowledge_service_config import (
     KnowledgeServiceConfig,
+    ServiceApi,
 )
-from julee.domain.models.knowledge_service_config import ServiceApi
 from julee.repositories.minio.knowledge_service_config import (
     MinioKnowledgeServiceConfigRepository,
 )
+
 from .fake_client import FakeMinioClient
 
 

julee/repositories/minio/tests/test_knowledge_service_query.py

@@ -6,16 +6,17 @@ query repository implementation, using the fake client to avoid external
 dependencies during testing.
 """
 
-import pytest
-
 from datetime import datetime, timezone
 
+import pytest
+
 from julee.domain.models.assembly_specification import (
     KnowledgeServiceQuery,
 )
 from julee.repositories.minio.knowledge_service_query import (
     MinioKnowledgeServiceQueryRepository,
 )
+
 from .fake_client import FakeMinioClient
 
 

julee/repositories/minio/tests/test_policy.py

@@ -6,11 +6,13 @@ implementation, using the fake client to avoid external dependencies during
 testing.
 """
 
-import pytest
 from datetime import datetime, timezone
 
+import pytest
+
 from julee.domain.models.policy import Policy, PolicyStatus
 from julee.repositories.minio.policy import MinioPolicyRepository
+
 from .fake_client import FakeMinioClient
 
 

julee/repositories/temporal/activities.py

@@ -10,12 +10,14 @@ The classes follow the naming pattern documented in systemPatterns.org:
 - Each repository type gets its own activity prefix
 """
 
-from julee.util.temporal.decorators import temporal_activity_registration
 from julee.repositories.minio.assembly import MinioAssemblyRepository
 from julee.repositories.minio.assembly_specification import (
     MinioAssemblySpecificationRepository,
 )
 from julee.repositories.minio.document import MinioDocumentRepository
+from julee.repositories.minio.document_policy_validation import (
+    MinioDocumentPolicyValidationRepository,
+)
 from julee.repositories.minio.knowledge_service_config import (
     MinioKnowledgeServiceConfigRepository,
 )
@@ -25,20 +27,18 @@ from julee.repositories.minio.knowledge_service_query import (
 from julee.repositories.minio.policy import (
     MinioPolicyRepository,
 )
-from julee.repositories.minio.document_policy_validation import (
-    MinioDocumentPolicyValidationRepository,
-)
 
 # Import activity name bases from shared module
 from julee.repositories.temporal.activity_names import (
     ASSEMBLY_ACTIVITY_BASE,
     ASSEMBLY_SPECIFICATION_ACTIVITY_BASE,
     DOCUMENT_ACTIVITY_BASE,
+    DOCUMENT_POLICY_VALIDATION_ACTIVITY_BASE,
     KNOWLEDGE_SERVICE_CONFIG_ACTIVITY_BASE,
     KNOWLEDGE_SERVICE_QUERY_ACTIVITY_BASE,
     POLICY_ACTIVITY_BASE,
-    DOCUMENT_POLICY_VALIDATION_ACTIVITY_BASE,
 )
+from julee.util.temporal.decorators import temporal_activity_registration
 
 
 @temporal_activity_registration(ASSEMBLY_ACTIVITY_BASE)

julee/repositories/temporal/proxies.py

@@ -11,12 +11,14 @@ workflow.execute_activity() with the appropriate activity names, timeouts,
 and retry policies.
 """
 
-from julee.util.temporal.decorators import temporal_workflow_proxy
 from julee.domain.repositories.assembly import AssemblyRepository
 from julee.domain.repositories.assembly_specification import (
     AssemblySpecificationRepository,
 )
 from julee.domain.repositories.document import DocumentRepository
+from julee.domain.repositories.document_policy_validation import (
+    DocumentPolicyValidationRepository,
+)
 from julee.domain.repositories.knowledge_service_config import (
     KnowledgeServiceConfigRepository,
 )
@@ -24,20 +26,18 @@ from julee.domain.repositories.knowledge_service_query import (
     KnowledgeServiceQueryRepository,
 )
 from julee.domain.repositories.policy import PolicyRepository
-from julee.domain.repositories.document_policy_validation import (
-    DocumentPolicyValidationRepository,
-)
 
 # Import activity name bases from shared module
 from julee.repositories.temporal.activity_names import (
     ASSEMBLY_ACTIVITY_BASE,
     ASSEMBLY_SPECIFICATION_ACTIVITY_BASE,
     DOCUMENT_ACTIVITY_BASE,
+    DOCUMENT_POLICY_VALIDATION_ACTIVITY_BASE,
     KNOWLEDGE_SERVICE_CONFIG_ACTIVITY_BASE,
     KNOWLEDGE_SERVICE_QUERY_ACTIVITY_BASE,
     POLICY_ACTIVITY_BASE,
-    DOCUMENT_POLICY_VALIDATION_ACTIVITY_BASE,
 )
+from julee.util.temporal.decorators import temporal_workflow_proxy
 
 
 @temporal_workflow_proxy(

julee/services/knowledge_service/__init__.py

@@ -9,12 +9,11 @@ appropriate implementation based on the service_api configuration.
 import logging
 
 from .knowledge_service import (
+    FileRegistrationResult,
     KnowledgeService,
     QueryResult,
-    FileRegistrationResult,
 )
 
-
 logger = logging.getLogger(__name__)
 
 

julee/services/knowledge_service/anthropic/knowledge_service.py

@@ -10,23 +10,24 @@ Requirements:
     - ANTHROPIC_API_KEY environment variable must be set
 """
 
-import os
 import logging
+import os
 import time
 import uuid
-from typing import Optional, List, Dict, Any
 from datetime import datetime, timezone
+from typing import Any
 
 from anthropic import AsyncAnthropic
 
+from julee.domain.models.document import Document
 from julee.domain.models.knowledge_service_config import (
     KnowledgeServiceConfig,
 )
-from julee.domain.models.document import Document
+
 from ..knowledge_service import (
+    FileRegistrationResult,
     KnowledgeService,
     QueryResult,
-    FileRegistrationResult,
 )
 
 logger = logging.getLogger(__name__)
@@ -171,9 +172,9 @@ class AnthropicKnowledgeService(KnowledgeService):
         self,
         config: KnowledgeServiceConfig,
         query_text: str,
-        service_file_ids: Optional[List[str]] = None,
-        query_metadata: Optional[Dict[str, Any]] = None,
-        assistant_prompt: Optional[str] = None,
+        service_file_ids: list[str] | None = None,
+        query_metadata: dict[str, Any] | None = None,
+        assistant_prompt: str | None = None,
     ) -> QueryResult:
         """Execute a query against Anthropic.
 

julee/services/knowledge_service/anthropic/tests/test_knowledge_service.py

@@ -6,19 +6,20 @@ KnowledgeService protocol, verifying file registration and query
 execution functionality.
 """
 
-import pytest
-from unittest.mock import AsyncMock, MagicMock, patch
-from datetime import datetime, timezone
 import io
+from datetime import datetime, timezone
+from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
 
-from julee.domain.models.knowledge_service_config import (
-    KnowledgeServiceConfig,
-)
-from julee.domain.models.document import Document, DocumentStatus
-from julee.domain.models.knowledge_service_config import ServiceApi
 from julee.domain.models.custom_fields.content_stream import (
     ContentStream,
 )
+from julee.domain.models.document import Document, DocumentStatus
+from julee.domain.models.knowledge_service_config import (
+    KnowledgeServiceConfig,
+    ServiceApi,
+)
 from julee.services.knowledge_service.anthropic import (
     knowledge_service as anthropic_ks,
 )
@@ -184,7 +185,7 @@ class TestAnthropicKnowledgeService:
     ) -> None:
         """Test execute_query handles API errors gracefully."""
         mock_client = MagicMock()
-        mock_client.messages.create = AsyncMock(side_effect=Exception("API Error"))
+        mock_client.messages.create = AsyncMock(side_effect=RuntimeError("API Error"))
 
         with patch(
             "julee.services.knowledge_service.anthropic.knowledge_service.AsyncAnthropic"
@@ -193,7 +194,7 @@ class TestAnthropicKnowledgeService:
 
             service = anthropic_ks.AnthropicKnowledgeService()
 
-            with pytest.raises(Exception):
+            with pytest.raises(RuntimeError):
                 await service.execute_query(knowledge_service_config, "Test query")
 
     @patch.dict("os.environ", {"ANTHROPIC_API_KEY": "test-key"})