Fix ollama (#2351)

## Description

- **Summary of changes**: Describe the key changes in this PR and their
purpose.
- **Related issues**: Mention if this PR fixes or is connected to any
issues.
- **Motivation and context**: Explain the reason for the changes and the
problem they solve.
- **Environment or dependencies**: Specify any changes in dependencies
or environment configurations required for this update.
- **Impact on metrics**: (If applicable) Describe changes in any metrics
or performance benchmarks.

Fixes #2349

---

## Type of change

Please check the options that are relevant:

- [x] Bug fix (non-breaking change which fixes an issue)
- [ ] New feature (non-breaking change which adds functionality)
- [x] Breaking change (fix or feature that would cause existing
functionality to not work as expected)
- [ ] Model update (Addition or modification of models)
- [ ] Other (please describe):

---

## Checklist

- [ ] Adherence to standards: Code complies with Agno’s style guidelines
and best practices.
- [ ] Formatting and validation: You have run `./scripts/format.sh` and
`./scripts/validate.sh` to ensure code is formatted and linted.
- [ ] Self-review completed: A thorough review has been performed by the
contributor(s).
- [ ] Documentation: Docstrings and comments have been added or updated
for any complex logic.
- [ ] Examples and guides: Relevant cookbook examples have been included
or updated (if applicable).
- [x] Tested in a clean environment: Changes have been tested in a clean
environment to confirm expected behavior.
- [ ] Tests (optional): Tests have been added or updated to cover any
new or changed functionality.

---

## Additional Notes

Include any deployment notes, performance implications, security
considerations, or other relevant information (e.g., screenshots or logs
if applicable).

Author: devalexandre

libs/agno/agno/embedder/ollama.py

@@ -5,22 +5,34 @@
 from agno.utils.log import logger
 
 try:
-    import pkg_resources
+    from ollama import Client as OllamaClient
+    import importlib.metadata as metadata
     from packaging import version
 
-    ollama_version = pkg_resources.get_distribution("ollama").version
-    if version.parse(ollama_version).major == 0 and version.parse(ollama_version).minor < 3:
+    # Get installed Ollama version
+    ollama_version = metadata.version("ollama")
+    
+    # Check version compatibility (requires v0.3.x or higher)
+    parsed_version = version.parse(ollama_version)
+    if parsed_version.major == 0 and parsed_version.minor < 3:
         import warnings
+        warnings.warn("Only Ollama v0.3.x and above are supported", UserWarning)
+        raise RuntimeError("Incompatible Ollama version detected")
 
-        warnings.warn(
-            "We only support Ollama v0.3.x and above.",
-            UserWarning,
-        )
-        raise RuntimeError("Incompatible Ollama version detected. Execution halted.")
+except ImportError as e:
+    # Handle different import error scenarios
+    if "ollama" in str(e):
+        raise ImportError(
+            "Ollama not installed. Install with `pip install ollama`"
+        ) from e
+    else:
+        raise ImportError(
+            "Missing dependencies. Install with `pip install packaging importlib-metadata`"
+        ) from e
 
-    from ollama import Client as OllamaClient
-except (ModuleNotFoundError, ImportError):
-    raise ImportError("`ollama` not installed. Please install using `pip install ollama`")
+except Exception as e:
+    # Catch-all for unexpected errors
+    print(f"An unexpected error occurred: {e}")
 
 
 @dataclass
@@ -53,14 +65,23 @@ def _response(self, text: str) -> Dict[str, Any]:
         if self.options is not None:
             kwargs["options"] = self.options
 
-        return self.client.embed(input=text, model=self.id, **kwargs)  # type: ignore
+        response = self.client.embed(input=text, model=self.id, **kwargs)
+        if response and "embeddings" in response:
+            embeddings = response["embeddings"]
+            if isinstance(embeddings, list) and len(embeddings) > 0 and isinstance(embeddings[0], list):
+                return {"embeddings": embeddings[0]}  # Use the first element
+            elif isinstance(embeddings, list) and all(isinstance(x, (int, float)) for x in embeddings):
+                return {"embeddings": embeddings}  # Return as-is if already flat
+        return {"embeddings": []}  # Return an empty list if no valid embedding is found
 
     def get_embedding(self, text: str) -> List[float]:
         try:
             response = self._response(text=text)
-            if response is None:
+            embedding = response.get("embeddings", [])
+            if len(embedding) != self.dimensions:
+                logger.warning(f"Expected embedding dimension {self.dimensions}, but got {len(embedding)}")
                 return []
-            return response.get("embeddings", [])
+            return embedding
         except Exception as e:
             logger.warning(e)
             return []