From 8aff258a93ab00245ad0e5f83437903bed5bf492 Mon Sep 17 00:00:00 2001
From: Sameer Kankute <sameer@berri.ai>
Date: Tue, 6 Jan 2026 14:20:30 +0530
Subject: [PATCH] fix: test_databricks_embeddings[True]

---
 tests/llm_translation/test_databricks.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tests/llm_translation/test_databricks.py b/tests/llm_translation/test_databricks.py
index 40fc712f2b..3013d00288 100644
--- a/tests/llm_translation/test_databricks.py
+++ b/tests/llm_translation/test_databricks.py
@@ -15,6 +15,7 @@ import litellm
 from litellm.exceptions import BadRequestError
 from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler
 from litellm.utils import CustomStreamWrapper
+from litellm._version import version
 from base_llm_unit_tests import BaseLLMChatTest, BaseAnthropicChatTest
 
 try:
@@ -725,6 +726,7 @@ def test_embeddings_with_sync_http_handler(monkeypatch):
             headers={
                 "Authorization": f"Bearer {api_key}",
                 "Content-Type": "application/json",
+                "User-Agent": f"litellm/{version}",
             },
             data=json.dumps(
                 {
@@ -767,6 +769,7 @@ def test_embeddings_with_async_http_handler(monkeypatch):
             headers={
                 "Authorization": f"Bearer {api_key}",
                 "Content-Type": "application/json",
+                "User-Agent": f"litellm/{version}",
             },
             data=json.dumps(
                 {
@@ -823,6 +826,7 @@ def test_embeddings_uses_databricks_sdk_if_api_key_and_base_not_specified(monkey
             headers={
                 "Authorization": f"Bearer {api_key}",
                 "Content-Type": "application/json",
+                "User-Agent": f"litellm/{version}",
             },
             data=json.dumps(
                 {
@@ -895,6 +899,7 @@ async def test_databricks_embeddings(sync_mode, monkeypatch):
                 headers={
                     "Authorization": f"Bearer {api_key}",
                     "Content-Type": "application/json",
+                    "User-Agent": f"litellm/{version}",
                 },
                 data=json.dumps(
                     {
@@ -923,6 +928,7 @@ async def test_databricks_embeddings(sync_mode, monkeypatch):
                 headers={
                     "Authorization": f"Bearer {api_key}",
                     "Content-Type": "application/json",
+                    "User-Agent": f"litellm/{version}",
                 },
                 data=json.dumps(
                     {