fusionkit-core 0.1.1__tar.gz

fusionkit_core-0.1.1/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.3
+Name: fusionkit-core
+Version: 0.1.1
+Summary: Core local model fusion primitives.
+Requires-Dist: anthropic>=0.40.0
+Requires-Dist: google-genai>=1.0.0
+Requires-Dist: openai>=2.13.0
+Requires-Dist: pydantic>=2.12.5
+Requires-Dist: pyyaml>=6.0.3
+Requires-Python: >=3.11

fusionkit_core-0.1.1/pyproject.toml

@@ -0,0 +1,16 @@
+[project]
+name = "fusionkit-core"
+version = "0.1.1"
+description = "Core local model fusion primitives."
+requires-python = ">=3.11"
+dependencies = [
+  "anthropic>=0.40.0",
+  "google-genai>=1.0.0",
+  "openai>=2.13.0",
+  "pydantic>=2.12.5",
+  "pyyaml>=6.0.3",
+]
+
+[build-system]
+requires = ["uv_build>=0.11.21,<0.12.0"]
+build-backend = "uv_build"

fusionkit_core-0.1.1/src/fusionkit_core/__init__.py

@@ -0,0 +1,181 @@
+from fusionkit_core.artifacts import LocalArtifactStore, hash_bytes, hash_text
+from fusionkit_core.clients import (
+    AnthropicModelClient,
+    FakeModelClient,
+    GoogleModelClient,
+    LocalModelClient,
+    OpenAICompatibleClient,
+    build_client,
+    build_clients,
+)
+from fusionkit_core.config import (
+    CostMetadata,
+    EndpointCapabilities,
+    FusionConfig,
+    FusionMode,
+    ModelEndpoint,
+    ProviderKind,
+    RunBudget,
+    SamplingConfig,
+)
+from fusionkit_core.contracts import (
+    ArtifactRefV1,
+    BenchmarkTaskRecordV1,
+    ContractMetadata,
+    ContractRecord,
+    EnsembleReceiptV1,
+    FusionRecordV1,
+    FusionRunRequestV1,
+    FusionRunState,
+    HarnessCandidateRecordV1,
+    HarnessRunResultV1,
+    HarnessTrajectoryV1,
+    JudgeSynthesisRecordV1,
+    ModelCallRecordV1,
+    ModelEndpointV1,
+    ToolCallPlanV1,
+    ToolExecutionRecordV1,
+    contract_metadata,
+    contract_model_for_schema,
+    producer,
+    producer_git_sha,
+    producer_version,
+    schema_bundle_hash,
+    status_for_run_state,
+)
+from fusionkit_core.fusion import FusionEngine
+from fusionkit_core.judge import CandidateEvidence, JudgeSynthesisResult, JudgeSynthesizer
+from fusionkit_core.providers import (
+    endpoint_to_contract,
+    estimate_cost,
+    normalize_usage,
+    provider_metadata,
+    resolve_api_key,
+)
+from fusionkit_core.router import HeuristicRouter
+from fusionkit_core.run import (
+    CandidateInspection,
+    CreateRunResult,
+    FusionRunEvent,
+    FusionRunManager,
+    IdempotencyRecord,
+    NativeRunError,
+    RunEventPage,
+    RunInspection,
+    RunStateSummary,
+    ToolExecutionMode,
+    ToolExecutionPolicy,
+    ToolPausePlaceholder,
+    ToolResultSubmission,
+    canonical_json,
+    hash_json,
+    make_id,
+)
+from fusionkit_core.run_store import FileSystemRunStore
+from fusionkit_core.trace import (
+    TRACE_CANDIDATE_HEADER,
+    TRACE_ID_HEADER,
+    TRACE_PARENT_SPAN_HEADER,
+    TRACE_SPAN_HEADER,
+    TraceEmitter,
+    ambient_trace_id,
+    emit,
+    get_emitter,
+    new_span_id,
+    new_trace_id,
+)
+from fusionkit_core.types import (
+    Candidate,
+    ChatMessage,
+    ModelResponse,
+    StreamChunk,
+    ToolCall,
+    Usage,
+)
+
+__all__ = [
+    "AnthropicModelClient",
+    "ArtifactRefV1",
+    "BenchmarkTaskRecordV1",
+    "Candidate",
+    "CandidateEvidence",
+    "CandidateInspection",
+    "ChatMessage",
+    "ContractMetadata",
+    "ContractRecord",
+    "CostMetadata",
+    "CreateRunResult",
+    "EndpointCapabilities",
+    "EnsembleReceiptV1",
+    "FakeModelClient",
+    "FileSystemRunStore",
+    "FusionConfig",
+    "FusionEngine",
+    "FusionMode",
+    "FusionRunEvent",
+    "FusionRunManager",
+    "FusionRecordV1",
+    "FusionRunRequestV1",
+    "FusionRunState",
+    "GoogleModelClient",
+    "HarnessCandidateRecordV1",
+    "HarnessTrajectoryV1",
+    "HarnessRunResultV1",
+    "HeuristicRouter",
+    "IdempotencyRecord",
+    "JudgeSynthesisRecordV1",
+    "JudgeSynthesizer",
+    "JudgeSynthesisResult",
+    "LocalArtifactStore",
+    "LocalModelClient",
+    "ModelCallRecordV1",
+    "ModelEndpoint",
+    "ModelEndpointV1",
+    "ModelResponse",
+    "NativeRunError",
+    "OpenAICompatibleClient",
+    "ProviderKind",
+    "RunEventPage",
+    "RunInspection",
+    "RunBudget",
+    "RunStateSummary",
+    "SamplingConfig",
+    "StreamChunk",
+    "TRACE_CANDIDATE_HEADER",
+    "TRACE_ID_HEADER",
+    "TRACE_PARENT_SPAN_HEADER",
+    "TRACE_SPAN_HEADER",
+    "TraceEmitter",
+    "ToolCall",
+    "ToolExecutionMode",
+    "ToolExecutionPolicy",
+    "ToolPausePlaceholder",
+    "ToolResultSubmission",
+    "ToolCallPlanV1",
+    "ToolExecutionRecordV1",
+    "Usage",
+    "ambient_trace_id",
+    "build_client",
+    "build_clients",
+    "canonical_json",
+    "contract_metadata",
+    "contract_model_for_schema",
+    "emit",
+    "endpoint_to_contract",
+    "estimate_cost",
+    "get_emitter",
+    "hash_bytes",
+    "hash_json",
+    "hash_text",
+    "make_id",
+    "new_span_id",
+    "new_trace_id",
+    "normalize_usage",
+    "producer",
+    "producer_git_sha",
+    "producer_version",
+    "provider_metadata",
+    "resolve_api_key",
+    "schema_bundle_hash",
+    "status_for_run_state",
+]

fusionkit_core-0.1.1/src/fusionkit_core/artifacts.py

@@ -0,0 +1,48 @@
+from __future__ import annotations
+
+import hashlib
+from pathlib import Path
+
+from fusionkit_core.contracts import ArtifactKind, ContractArtifactRef
+
+
+def hash_bytes(content: bytes) -> str:
+    return "sha256:" + hashlib.sha256(content).hexdigest()
+
+
+def hash_text(content: str) -> str:
+    return hash_bytes(content.encode("utf-8"))
+
+
+class LocalArtifactStore:
+    def __init__(self, root: str | Path) -> None:
+        self.root = Path(root)
+        self.root.mkdir(parents=True, exist_ok=True)
+
+    def write_text(
+        self,
+        run_id: str,
+        artifact_id: str,
+        kind: ArtifactKind,
+        content: str,
+        *,
+        suffix: str = ".txt",
+    ) -> ContractArtifactRef:
+        artifact_dir = self.root / run_id / "artifacts"
+        artifact_dir.mkdir(parents=True, exist_ok=True)
+        path = artifact_dir / f"{artifact_id}{suffix}"
+        path.write_text(content, encoding="utf-8")
+        return ContractArtifactRef(
+            artifact_id=artifact_id,
+            kind=kind,
+            uri=str(path),
+            hash=hash_text(content),
+            redaction_status="synthetic",
+        )
+
+
+__all__ = [
+    "LocalArtifactStore",
+    "hash_bytes",
+    "hash_text",
+]