chroma-core / chroma-core/chroma

[Bug]: Adapt to MistralAI version migration

Open
#7,633 0 comments 0 reactions 0 assignees View on GitHub
bug
Dominant language
Rust
Stars
29.3k
Forks
2.5k
Avg merge
1d 4h
Merged PRs (30d)
38

Description

### What happened?

Hi!

I have opened a pull request 3 months ago, but there had been no activity on it, so I open this issue.

MistralAI has deployed the version 2.0.0 of its python package on March 10th, introducing a change in its structure. One of the main changes is the switch `from mistralai import Mistral` to `from mistralai.client import Mistral`.

However, in [`chromadb/utils/embedding_functions/mistral_embedding_function.py`](https://github.com/chroma-core/chroma/blob/main/chromadb/utils/embedding_functions/mistral_embedding_function.py#L22), you make a check on MistralAI Python package by doing `from mistralai import Mistral`. This import/check does not work with Mistral python package v2.

My [PR ](https://github.com/chroma-core/chroma/pull/7105) contains a suggestion on how to handle it. However, it might not be the cleanest implementation.

### Versions

Chroma v1.5.9, Python 3.11.13

### Relevant log output

```shell

```

Contributor guide

No contributing guide indexed for this repository

Research direction

Start in chromadb/utils/embedding_functions/mistral_embedding_function.py and compare the proposed approach in PR #7105 with the MistralAI v2 import change described here. Confirm the dependency check recognizes the supported package structure and that the embedding function retains existing behavior.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
Difficulty
2/5
Estimated time
1-3 hours
Activity status
Stale
Clarity
Clearly specified
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.