bee-agent-framework icon indicating copy to clipboard operation
bee-agent-framework copied to clipboard

Feat: EmbeddingModel for Backend

Open vabarbosa opened this issue 1 year ago • 2 comments

Is your feature request related to a problem? Please describe.

this is a follow-up to the backed work which added the ChatModel this issue is to continue this backend work and add the EmbeddingModel similar to what is currently implemented in the TS framework (and also follows the Python ChatModel implementation). e.g.,

EmbeddingModel.from_name('watsonx:text-embedding-3-small')

WatsonXEmbeddingModel('text-embedding-3-small')

Providers

Make embedding for all providers listed in https://i-am-bee.github.io/beeai-framework/#/python/backend?id=supported-providers. Implementation for other providers should be done based on the LiteLLMEmbeddingModel. See reference implementation for WatsonxEmbeddingModel and OllamaEmbeddingModel. Other providers should be implemented in a similar fashion (it is always best to check out how it was done for that chat version of the given provider).

To dos

  • [x] Ollama @Tomas2D to review and merge #814
  • [x] Watsonx @Tomas2D to review and merge #814
  • [ ] Add remaining providers: @araujof to hand off to Yair

Blocking

  • #771

vabarbosa avatar Feb 26 '25 17:02 vabarbosa

Is there a reason this was unassigned from me? I have an open PR on this #814 waiting for review and merge? (cc @Tomas2D )

ajbozarth avatar May 22 '25 18:05 ajbozarth

We got that merged. Yair and the team will add the remaining providers. I also updated the scope.

Tomas2D avatar May 26 '25 09:05 Tomas2D

All core providers were implemented. We can close.

Tomas2D avatar Aug 06 '25 13:08 Tomas2D