[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add support for Bumblebee functions? #88

Open
brainlid opened this issue Mar 30, 2024 · 3 comments
Open

Add support for Bumblebee functions? #88

brainlid opened this issue Mar 30, 2024 · 3 comments
Milestone

Comments

@brainlid
Copy link
Owner

Bumblebee doesn't support a constrained output of only valid JSON.

In the early days of LangChain, they implemented an alternate approach for functions that was more of a hack, but worked well enough.

Investigate if this approach could work for bringing functions to Bumblebee models. It would still help if the model being run understood JSON, functions, etc.

@brainlid brainlid added this to the v0.3.0 milestone Mar 30, 2024
@hwuethrich
Copy link
hwuethrich commented May 2, 2024

Hi @brainlid

I'd be very interested in this feature. Especially now that Mistral released their Mixtral-8x22B-Instruct-v0.1 model, which supports function calling! 🎉

Here is an example for the expected prompt template:
https://github.com/mistralai/mistral-common/blob/main/examples/tokenizer.ipynb

My company (@zebbra) might have some capacity to make this contribution in case you are interested? I assume it is just a matter of adding a new clause to https://github.com/brainlid/langchain/blob/main/lib/utils/chat_templates.ex, right? Or are there any other pieces missing from the puzzle?

Anyway, thanks for this library (and podcast! 🙂)

Edit: I guess ChatBumblebee would have to be able to parse the function calls as well

@hwuethrich
Copy link

Probably not as easy as I thought, because Mixtral is currently no supported by Bumblebee: elixir-nx/bumblebee#365

@marcnnn
Copy link
marcnnn commented Dec 2, 2024

Hello,
I have added JSON tool use for Llama3.1 in #198
I will look into the other tool use methods and models as well if I am on the right track with my implementation, so please try it out!
And give feedback!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants