relationship persistence, context entities, idempotent asset creation, audit/version handling for relationship changes

src/kontextual_engine/ports/repositories.py

@@ -9,6 +9,9 @@ from kontextual_engine.core import (
     AssetRepresentation,
     AssetVersion,
     AuditEvent,
+    ContextEntity,
+    CoreRelationship,
+    IdempotencyRecord,
     KnowledgeAsset,
     LifecycleState,
     MetadataRecord,
@@ -41,10 +44,24 @@ class AssetRegistryRepository(Protocol):
     def save_metadata_record(self, asset_id: str, record: MetadataRecord) -> MetadataRecord: ...
     def list_metadata_records(self, asset_id: str) -> list[MetadataRecord]: ...
 
+    def save_context_entity(self, entity: ContextEntity) -> ContextEntity: ...
+    def get_context_entity(self, entity_id: str) -> ContextEntity: ...
+    def list_context_entities(self) -> list[ContextEntity]: ...
+
+    def save_relationship(self, relationship: CoreRelationship) -> CoreRelationship: ...
+    def get_relationship(self, relationship_id: str) -> CoreRelationship: ...
+    def list_relationships(
+        self,
+        *,
+        source_id: str | None = None,
+        target_id: str | None = None,
+    ) -> list[CoreRelationship]: ...
+
     def save_version(self, version: AssetVersion) -> AssetVersion: ...
     def list_versions(self, asset_id: str) -> list[AssetVersion]: ...
 
     def save_audit_event(self, event: AuditEvent) -> AuditEvent: ...
+    def get_audit_event(self, event_id: str) -> AuditEvent: ...
     def list_audit_events(
         self,
         *,
@@ -52,3 +69,5 @@ class AssetRegistryRepository(Protocol):
         correlation_id: str | None = None,
     ) -> list[AuditEvent]: ...
 
+    def save_idempotency_record(self, record: IdempotencyRecord) -> IdempotencyRecord: ...
+    def get_idempotency_record(self, key: str) -> IdempotencyRecord | None: ...