add required header anthropic-version for AnthropicProvider

Signed-off-by: Xiaohan Wang <xhwa73@gmail.com>
mlflow · Feb 4, 2024 · e7b2852 · e7b2852
1 parent a06e82c
commit e7b2852
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mlflow/gateway/providers/anthropic.py b/mlflow/gateway/providers/anthropic.py
@@ -104,7 +104,10 @@ def __init__(self, config: RouteConfig) -> None:
         if config.model.config is None or not isinstance(config.model.config, AnthropicConfig):
             raise TypeError(f"Invalid config type {config.model.config}")
         self.anthropic_config: AnthropicConfig = config.model.config
-        self.headers = {"x-api-key": self.anthropic_config.anthropic_api_key}
+        self.headers = {
+            "x-api-key": self.anthropic_config.anthropic_api_key,
+            "anthropic-version": "2023-06-01",
+        }
         self.base_url = "https://api.anthropic.com/v1/"
 
     async def completions(self, payload: completions.RequestPayload) -> completions.ResponsePayload: