agentinputlab 0.1.0a1__py3-none-any.whl

agentinputlab/py.typed

@@ -0,0 +1 @@
+

agentinputlab/recipes/__init__.py

@@ -0,0 +1,333 @@
+"""Offline optimization and training recipes."""
+
+from .ace import (
+    ACEContextArtifact,
+    ACEControls,
+    ACEExample,
+    ACEOfflineRecipe,
+    SplitAudit,
+    validate_split_isolation,
+)
+from .ace_bfcl import ACEBFCLFeedback, ACEOnlineBFCLFeedback
+from .ace_common_support import (
+    ACE_COMMON_SUPPORT_PLACEMENT_RULE,
+    ACE_COMMON_SUPPORT_RELEASE,
+    ACEPlaybookCommonSupport,
+)
+from .ace_online import ACEOnlineControls, ACEOnlineRecipe, ACEOnlineResult
+from .ace_retrieval import ACEOnlineRetrievalFeedback, ACERetrievalFeedback
+from .ace_statebench import ACEStateBenchFeedback, build_ace_statebench_feedback_input
+from .ace_tau import ACETauFeedback, build_ace_tau_feedback_input
+from .ace_toolsandbox import ACEToolSandboxFeedback, build_ace_toolsandbox_feedback_input
+from .acon_guideline import ACONGuidelineArtifact, ACONGuidelineControls, ACONGuidelinePipeline
+from .adaptive_rag import (
+    AdaptiveRAGFitControls,
+    AdaptiveRAGOfflineTraining,
+    AdaptiveRAGTrainedRouterArtifact,
+    rebuild_adaptive_silver,
+)
+from .llmlingua2_inference import LLMLingua2TrainedCompressor
+from .llmlingua2_training import (
+    LLMLingua2DistillationTraining,
+    LLMLingua2FitControls,
+    LLMLingua2TrainedArtifact,
+    label_llmlingua_compressed_records,
+)
+from .mce import (
+    MCE_UPSTREAM_COMMIT,
+    MCEArtifactFile,
+    MCEContextArtifact,
+    MCEControls,
+    MCEExample,
+    MCEIteration,
+    MCEOfflineRecipe,
+    MCESandboxAttestation,
+    MCESplitAudit,
+    validate_mce_split_isolation,
+)
+from .mce_final import (
+    MCEFinalArtifactEvaluator,
+    MCEFinalControls,
+    MCEFinalEvaluation,
+    MCEFinalExample,
+)
+from .mce_text_fit import MCETextFitControls, MCETextOfflineRecipe, validate_mce_text_checkout
+from .mce_text_only import (
+    MCE_TEXT_ARTIFACT_PLACEMENT_RULE,
+    MCE_TEXT_ARTIFACT_PROTOCOL,
+    MCE_TEXT_ARTIFACT_RELEASE,
+    MCETextArtifact,
+    MCETextArtifactCommonSupport,
+)
+from .qrag import QRAGFitControls, QRAGTrainedPolicyArtifact, QRAGValuePolicyTraining
+from .qrag_inference import QRAGTrainedPolicyIntervention, QRAGTrainedPolicyRuntime
+from .r2ag_features import R2AGNativeFeatureEncoder, R2AGRetrievalDocument
+from .r2ag_inference import R2AGReaderDocument, R2AGTrainedReader
+from .r2ag_training import (
+    R2AGFitControls,
+    R2AGOfflineTraining,
+    R2AGTrainedArtifact,
+    prepare_r2ag_native_features,
+)
+from .recomp_abstractive import (
+    RECOMPAbstractiveFitControls,
+    RECOMPAbstractiveTraining,
+    RECOMPTrainedSeq2SeqArtifact,
+)
+from .recomp_extractive import (
+    RECOMPExtractiveFitControls,
+    RECOMPExtractiveTraining,
+    RECOMPTrainedEncoderArtifact,
+)
+from .rqrag import (
+    RQ_RAG_ACTION_TOKENS,
+    RQ_RAG_METHOD_RELEASE_ID,
+    RQ_RAG_UPSTREAM_COMMIT,
+    RQRAGCandidateScore,
+    RQRAGControls,
+    RQRAGLicenseClearance,
+    RQRAGMethodControls,
+    RQRAGMethodTreeSearch,
+    RQRAGPatchAttestation,
+    RQRAGReleased7BExternal,
+    RQRAGTreeNode,
+)
+from .rqrag_common_support import (
+    RQ_RAG_COMMON_SUPPORT_MAPPING,
+    RQ_RAG_COMMON_SUPPORT_RELEASE_ID,
+    RQRAGCommonSupportQueries,
+)
+from .rqrag_completion import RQRAGCompletionBackend
+from .skillopt import (
+    SKILLOPT_PROTOCOL,
+    SKILLOPT_RELEASE_ID,
+    SKILLOPT_SOURCE_HASHES,
+    SKILLOPT_UPSTREAM_COMMIT,
+    SkillOptArtifact,
+    SkillOptControls,
+    SkillOptEnvironmentBinding,
+    SkillOptOfflineRecipe,
+    load_skillopt_artifact,
+    validate_skillopt_checkout,
+)
+from .skillopt_common_support import (
+    SKILLOPT_COMMON_SUPPORT_PLACEMENT_RULE,
+    SKILLOPT_COMMON_SUPPORT_RELEASE,
+    SkillOptSkillCommonSupport,
+)
+from .skillopt_retrieval import create_skillopt_retrieval_adapter
+from .smartsearch import (
+    SMARTSEARCH_COMMON_SUPPORT_RELEASE,
+    SmartSearchArtifactIdentity,
+    SmartSearchCommonSupportQueries,
+    SmartSearchControls,
+    SmartSearchFrozenInference,
+    SmartSearchStep,
+)
+from .smartsearch_curriculum import (
+    SmartSearchCurriculumArtifact,
+    SmartSearchCurriculumPreparation,
+    SmartSearchTrainingCase,
+    validate_smartsearch_training_splits,
+    verify_smartsearch_preparation_source,
+)
+from .smartsearch_grpo_data import (
+    SmartSearchGRPOControls,
+    SmartSearchGRPODataArtifact,
+    SmartSearchGRPOExample,
+    prepare_smartsearch_grpo_data,
+    verify_smartsearch_grpo_source,
+)
+from .smartsearch_grpo_meter import (
+    MeteredNativeGRPOEngine,
+    SmartSearchGRPOCostLedger,
+    SmartSearchGRPOCostLimits,
+    meter_smartsearch_grpo_helper,
+    observe_native_grpo_optimizer,
+)
+from .smartsearch_grpo_prefix import SmartSearchNativeGRPOPrefixBuilder
+from .smartsearch_grpo_tools import (
+    SMARTSEARCH_GRPO_ENVIRONMENT,
+    SmartSearchGRPORetrievalService,
+    SmartSearchNativeGRPOReward,
+    make_smartsearch_grpo_retrieval_server,
+)
+from .smartsearch_grpo_training import SmartSearchGRPOArtifact, SmartSearchNativeGRPOTraining
+from .smartsearch_inference import (
+    SmartSearchInferenceControls,
+    SmartSearchTrainedLocalModel,
+    SmartSearchTrainedPolicyInference,
+)
+from .smartsearch_native_data import (
+    SMARTSEARCH_LLAMAFACTORY_COMMIT,
+    prepare_smartsearch_native_data_source,
+    verify_smartsearch_native_data_source,
+)
+from .smartsearch_policy import (
+    SmartSearchNativePolicyTraining,
+    SmartSearchPolicyFitControls,
+    SmartSearchPolicyStageArtifact,
+)
+from .smartsearch_sampling import (
+    SmartSearchPolicySampler,
+    SmartSearchSamplingControls,
+    verify_smartsearch_sampling_source,
+)
+from .smartsearch_teacher import (
+    SmartSearchProcessTeacher,
+    SmartSearchTeacherCompletion,
+    SmartSearchTeacherControls,
+)
+from .smartsearch_training import SmartSearchNativeTrainingRecipe, SmartSearchTrainingChainArtifact
+
+__all__ = [
+    "ACE_COMMON_SUPPORT_PLACEMENT_RULE",
+    "ACE_COMMON_SUPPORT_RELEASE",
+    "MCE_TEXT_ARTIFACT_PLACEMENT_RULE",
+    "MCE_TEXT_ARTIFACT_PROTOCOL",
+    "MCE_TEXT_ARTIFACT_RELEASE",
+    "MCE_UPSTREAM_COMMIT",
+    "RQ_RAG_ACTION_TOKENS",
+    "RQ_RAG_COMMON_SUPPORT_MAPPING",
+    "RQ_RAG_COMMON_SUPPORT_RELEASE_ID",
+    "RQ_RAG_METHOD_RELEASE_ID",
+    "RQ_RAG_UPSTREAM_COMMIT",
+    "SKILLOPT_COMMON_SUPPORT_PLACEMENT_RULE",
+    "SKILLOPT_COMMON_SUPPORT_RELEASE",
+    "SKILLOPT_PROTOCOL",
+    "SKILLOPT_RELEASE_ID",
+    "SKILLOPT_SOURCE_HASHES",
+    "SKILLOPT_UPSTREAM_COMMIT",
+    "SMARTSEARCH_COMMON_SUPPORT_RELEASE",
+    "SMARTSEARCH_GRPO_ENVIRONMENT",
+    "SMARTSEARCH_LLAMAFACTORY_COMMIT",
+    "ACEBFCLFeedback",
+    "ACEContextArtifact",
+    "ACEControls",
+    "ACEExample",
+    "ACEOfflineRecipe",
+    "ACEOnlineBFCLFeedback",
+    "ACEOnlineControls",
+    "ACEOnlineRecipe",
+    "ACEOnlineResult",
+    "ACEOnlineRetrievalFeedback",
+    "ACEPlaybookCommonSupport",
+    "ACERetrievalFeedback",
+    "ACEStateBenchFeedback",
+    "ACETauFeedback",
+    "ACEToolSandboxFeedback",
+    "ACONGuidelineArtifact",
+    "ACONGuidelineControls",
+    "ACONGuidelinePipeline",
+    "AdaptiveRAGFitControls",
+    "AdaptiveRAGOfflineTraining",
+    "AdaptiveRAGTrainedRouterArtifact",
+    "LLMLingua2DistillationTraining",
+    "LLMLingua2FitControls",
+    "LLMLingua2TrainedArtifact",
+    "LLMLingua2TrainedCompressor",
+    "MCEArtifactFile",
+    "MCEContextArtifact",
+    "MCEControls",
+    "MCEExample",
+    "MCEFinalArtifactEvaluator",
+    "MCEFinalControls",
+    "MCEFinalEvaluation",
+    "MCEFinalExample",
+    "MCEIteration",
+    "MCEOfflineRecipe",
+    "MCESandboxAttestation",
+    "MCESplitAudit",
+    "MCETextArtifact",
+    "MCETextArtifactCommonSupport",
+    "MCETextFitControls",
+    "MCETextOfflineRecipe",
+    "MeteredNativeGRPOEngine",
+    "QRAGFitControls",
+    "QRAGTrainedPolicyArtifact",
+    "QRAGTrainedPolicyIntervention",
+    "QRAGTrainedPolicyRuntime",
+    "QRAGValuePolicyTraining",
+    "R2AGFitControls",
+    "R2AGNativeFeatureEncoder",
+    "R2AGOfflineTraining",
+    "R2AGReaderDocument",
+    "R2AGRetrievalDocument",
+    "R2AGTrainedArtifact",
+    "R2AGTrainedReader",
+    "RECOMPAbstractiveFitControls",
+    "RECOMPAbstractiveTraining",
+    "RECOMPExtractiveFitControls",
+    "RECOMPExtractiveTraining",
+    "RECOMPTrainedEncoderArtifact",
+    "RECOMPTrainedSeq2SeqArtifact",
+    "RQRAGCandidateScore",
+    "RQRAGCommonSupportQueries",
+    "RQRAGCompletionBackend",
+    "RQRAGControls",
+    "RQRAGLicenseClearance",
+    "RQRAGMethodControls",
+    "RQRAGMethodTreeSearch",
+    "RQRAGPatchAttestation",
+    "RQRAGReleased7BExternal",
+    "RQRAGTreeNode",
+    "SkillOptArtifact",
+    "SkillOptControls",
+    "SkillOptEnvironmentBinding",
+    "SkillOptOfflineRecipe",
+    "SkillOptSkillCommonSupport",
+    "SmartSearchArtifactIdentity",
+    "SmartSearchCommonSupportQueries",
+    "SmartSearchControls",
+    "SmartSearchCurriculumArtifact",
+    "SmartSearchCurriculumPreparation",
+    "SmartSearchFrozenInference",
+    "SmartSearchGRPOArtifact",
+    "SmartSearchGRPOControls",
+    "SmartSearchGRPOCostLedger",
+    "SmartSearchGRPOCostLimits",
+    "SmartSearchGRPODataArtifact",
+    "SmartSearchGRPOExample",
+    "SmartSearchGRPORetrievalService",
+    "SmartSearchInferenceControls",
+    "SmartSearchNativeGRPOPrefixBuilder",
+    "SmartSearchNativeGRPOReward",
+    "SmartSearchNativeGRPOTraining",
+    "SmartSearchNativePolicyTraining",
+    "SmartSearchNativeTrainingRecipe",
+    "SmartSearchPolicyFitControls",
+    "SmartSearchPolicySampler",
+    "SmartSearchPolicyStageArtifact",
+    "SmartSearchProcessTeacher",
+    "SmartSearchSamplingControls",
+    "SmartSearchStep",
+    "SmartSearchTeacherCompletion",
+    "SmartSearchTeacherControls",
+    "SmartSearchTrainedLocalModel",
+    "SmartSearchTrainedPolicyInference",
+    "SmartSearchTrainingCase",
+    "SmartSearchTrainingChainArtifact",
+    "SplitAudit",
+    "build_ace_statebench_feedback_input",
+    "build_ace_tau_feedback_input",
+    "build_ace_toolsandbox_feedback_input",
+    "create_skillopt_retrieval_adapter",
+    "label_llmlingua_compressed_records",
+    "load_skillopt_artifact",
+    "make_smartsearch_grpo_retrieval_server",
+    "meter_smartsearch_grpo_helper",
+    "observe_native_grpo_optimizer",
+    "prepare_r2ag_native_features",
+    "prepare_smartsearch_grpo_data",
+    "prepare_smartsearch_native_data_source",
+    "rebuild_adaptive_silver",
+    "validate_mce_split_isolation",
+    "validate_mce_text_checkout",
+    "validate_skillopt_checkout",
+    "validate_smartsearch_training_splits",
+    "validate_split_isolation",
+    "verify_smartsearch_grpo_source",
+    "verify_smartsearch_native_data_source",
+    "verify_smartsearch_preparation_source",
+    "verify_smartsearch_sampling_source",
+]