Implement-LLM-WP-0005-cost-model-estimators

llm_connect/__init__.py

@@ -15,6 +15,7 @@ Quick start::
 from llm_connect.adapter import ErrorLLMAdapter, LLMAdapter, MockLLMAdapter
 from llm_connect.claude_code import ClaudeCodeAdapter
 from llm_connect.config import LLMConfig, load_config
+from llm_connect.costs import CostEstimate, CostModel, estimate_cost
 from llm_connect.embedding_adapter import EmbeddingAdapter
 from llm_connect.embedding_cache import EmbeddingCache
 from llm_connect.embedding_factory import create_embedding_adapter
@@ -42,7 +43,20 @@ from llm_connect.grading import (
 from llm_connect.models import BudgetTracker, LLMResponse, RunConfig
 from llm_connect.openai import OpenAIAdapter
 from llm_connect.openrouter import OpenRouterAdapter
+from llm_connect.problem_classes import (
+    ChunkSummarizationProblemClass,
+    EntityExtractionProblemClass,
+    JudgeEvalProblemClass,
+    Observation,
+    ProblemClass,
+    ProblemClassRegistry,
+    RelationExtractionProblemClass,
+    ReportSynthesisProblemClass,
+    TokenEstimate,
+    default_problem_class_registry,
+)
 from llm_connect.quality import QualityLedger, QualityObservation, is_stale
+from llm_connect.rates import ModelRate, ModelRateRegistry
 from llm_connect.routing import AdaptiveRoutingPolicy, RoutingPolicy, RoutingRule
 from llm_connect.server import LLMServer
 from llm_connect.shadowing import ShadowingAdapter
@@ -95,4 +109,19 @@ __all__ = [
     "AdaptiveRoutingPolicy",
     "ShadowingAdapter",
     "LLMServer",
+    "ModelRate",
+    "ModelRateRegistry",
+    "CostEstimate",
+    "CostModel",
+    "estimate_cost",
+    "TokenEstimate",
+    "Observation",
+    "ProblemClass",
+    "ProblemClassRegistry",
+    "default_problem_class_registry",
+    "ChunkSummarizationProblemClass",
+    "EntityExtractionProblemClass",
+    "RelationExtractionProblemClass",
+    "JudgeEvalProblemClass",
+    "ReportSynthesisProblemClass",
 ]