fix: make base url as part of params for jina text embedder

Author: srini047

integrations/jina/src/haystack_integrations/components/embedders/jina/text_embedder.py

@@ -37,6 +37,7 @@ def __init__(
         self,
         api_key: Secret = Secret.from_env_var("JINA_API_KEY"),  # noqa: B008
         model: str = "jina-embeddings-v3",
+        base_url: str = JINA_API_URL,
         prefix: str = "",
         suffix: str = "",
         task: str | None = None,
@@ -68,6 +69,7 @@ def __init__(
 
         self.api_key = api_key
         self.model_name = model
+        self.base_url = base_url
         self.prefix = prefix
         self.suffix = suffix
         self._session = requests.Session()
@@ -98,6 +100,7 @@ def to_dict(self) -> dict[str, Any]:
         kwargs: dict[str, Any] = {
             "api_key": self.api_key.to_dict(),
             "model": self.model_name,
+            "base_url": self.base_url,
             "prefix": self.prefix,
             "suffix": self.suffix,
         }
@@ -152,7 +155,7 @@ def run(self, text: str) -> dict[str, Any]:
             parameters["late_chunking"] = self.late_chunking
 
         resp = self._session.post(
-            JINA_API_URL,
+            self.base_url,
             json={"input": [text_to_embed], "model": self.model_name, **parameters},
         ).json()
 

integrations/jina/tests/test_text_embedder.py

@@ -19,19 +19,22 @@ def test_init_default(self, monkeypatch):
 
         assert embedder.api_key == Secret.from_env_var("JINA_API_KEY")
         assert embedder.model_name == "jina-embeddings-v3"
+        assert embedder.base_url == "https://api.jina.ai/v1/embeddings"
         assert embedder.prefix == ""
         assert embedder.suffix == ""
 
     def test_init_with_parameters(self):
         embedder = JinaTextEmbedder(
             api_key=Secret.from_token("fake-api-key"),
             model="model",
+            base_url="http://api.jina.ai/v1/embeddings",
             prefix="prefix",
             suffix="suffix",
             late_chunking=True,
         )
         assert embedder.api_key == Secret.from_token("fake-api-key")
         assert embedder.model_name == "model"
+        assert embedder.base_url == "http://api.jina.ai/v1/embeddings"
         assert embedder.prefix == "prefix"
         assert embedder.suffix == "suffix"
         assert embedder.late_chunking is True
@@ -50,6 +53,7 @@ def test_to_dict(self, monkeypatch):
             "init_parameters": {
                 "api_key": {"env_vars": ["JINA_API_KEY"], "strict": True, "type": "env_var"},
                 "model": "jina-embeddings-v3",
+                "base_url": "https://api.jina.ai/v1/embeddings",
                 "prefix": "",
                 "suffix": "",
             },
@@ -59,6 +63,7 @@ def test_to_dict_with_custom_init_parameters(self, monkeypatch):
         monkeypatch.setenv("JINA_API_KEY", "fake-api-key")
         component = JinaTextEmbedder(
             model="model",
+            base_url="http://api.jina.ai/v1/embeddings",
             prefix="prefix",
             suffix="suffix",
             task="retrieval.query",
@@ -70,6 +75,7 @@ def test_to_dict_with_custom_init_parameters(self, monkeypatch):
             "init_parameters": {
                 "api_key": {"env_vars": ["JINA_API_KEY"], "strict": True, "type": "env_var"},
                 "model": "model",
+                "base_url": "http://api.jina.ai/v1/embeddings",
                 "prefix": "prefix",
                 "suffix": "suffix",
                 "task": "retrieval.query",