Bedrock Inference Engine Integration

src/oumi/builders/inference_engines.py

@@ -24,6 +24,7 @@
 from oumi.core.inference import BaseInferenceEngine
 from oumi.inference import (
     AnthropicInferenceEngine,
+    BedrockInferenceEngine,
     DeepSeekInferenceEngine,
     GoogleGeminiInferenceEngine,
     GoogleVertexInferenceEngine,
@@ -44,6 +45,7 @@
     MappingProxyType(
         {
             InferenceEngineType.ANTHROPIC: AnthropicInferenceEngine,
+            InferenceEngineType.BEDROCK: BedrockInferenceEngine,
             InferenceEngineType.DEEPSEEK: DeepSeekInferenceEngine,
             InferenceEngineType.GOOGLE_GEMINI: GoogleGeminiInferenceEngine,
             InferenceEngineType.GOOGLE_VERTEX: GoogleVertexInferenceEngine,

src/oumi/core/configs/inference_engine_type.py

@@ -62,3 +62,6 @@ class InferenceEngineType(str, Enum):
 
     SAMBANOVA = "SAMBANOVA"
     """The inference engine for SambaNova API."""
+
+    BEDROCK = "BEDROCK"
+    """The inference engine for AWS Bedrock API."""

src/oumi/inference/__init__.py

@@ -18,6 +18,7 @@
 """
 
 from oumi.inference.anthropic_inference_engine import AnthropicInferenceEngine
+from oumi.inference.bedrock_inference_engine import BedrockInferenceEngine
 from oumi.inference.deepseek_inference_engine import DeepSeekInferenceEngine
 from oumi.inference.gcp_inference_engine import GoogleVertexInferenceEngine
 from oumi.inference.gemini_inference_engine import GoogleGeminiInferenceEngine
@@ -35,6 +36,7 @@
 
 __all__ = [
     "AnthropicInferenceEngine",
+    "BedrockInferenceEngine",
     "DeepSeekInferenceEngine",
     "GoogleGeminiInferenceEngine",
     "GoogleVertexInferenceEngine",