add meta to return message

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

@@ -38,7 +38,7 @@ def _create_json_payload(self, text: str, generation_kwargs: Optional[Dict[str,
         """
         return {"model": self.model, "prompt": text, "options": {**self.generation_kwargs, **(generation_kwargs or {})}}
 
-    @component.output_types(embedding=List[float])
+    @component.output_types(embedding=List[float], meta=Dict[str, Any])
     def run(self, text: str, generation_kwargs: Optional[Dict[str, Any]] = None):
         """
         Run an Ollama Model on a given chat history.
@@ -55,4 +55,7 @@ def run(self, text: str, generation_kwargs: Optional[Dict[str, Any]] = None):
 
         response.raise_for_status()
 
-        return response.json()
+        result = response.json()
+        result["meta"] = {"model": self.model, "duration": response.elapsed}
+
+        return result

integrations/ollama/tests/test_text_embedder.py

@@ -25,6 +25,7 @@ def test_init(self):
         assert embedder.url == "http://my-custom-endpoint:11434/api/embeddings"
         assert embedder.model == "llama2"
 
+    @pytest.mark.integration
     def test_model_not_found(self):
         embedder = OllamaTextEmbedder(model="cheese")
 
@@ -33,9 +34,10 @@ def test_model_not_found(self):
 
     @pytest.mark.integration
     def test_run(self):
-        embedder = OllamaTextEmbedder()
+        embedder = OllamaTextEmbedder(model="orca-mini")
 
         reply = embedder.run("hello")
 
         assert isinstance(reply, dict)
         assert all(isinstance(element, float) for element in reply["embedding"])
+        assert reply["meta"]["model"] == "orca-mini"