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 feature to select model name for Ollama #98

Open
theobjectivedad opened this issue Apr 30, 2024 · 3 comments
Open

Add feature to select model name for Ollama #98

theobjectivedad opened this issue Apr 30, 2024 · 3 comments
Labels
community community feature feature Frequency - Monthly Frequency - Monthly Reach - Some Reach - Some Severity - S3 Severity - S3

Comments

@theobjectivedad
Copy link

Currently memgraph supports ollama for AI chat however it hardcodes llama2:latest and no other options are available. This enhancement would allow the memgraph-lab user to select the model to use with the Ollama endpoint.

@katarinasupe katarinasupe added feature feature community community Severity - S3 Severity - S3 Frequency - Monthly Frequency - Monthly Reach - VeryFew Reach - VeryFew Reach - Some Reach - Some and removed Reach - VeryFew Reach - VeryFew labels Apr 30, 2024
@katarinasupe
Copy link
Contributor

Hi @theobjectivedad, thank you for filing this feature request on GitHub 🙏 Keep track of the progress here

@theobjectivedad
Copy link
Author

Posting here for folks looking for a workaround. The model can be renamed to llama2/latest on the Ollama side to get things working. The following example is verified to work in the ollama docker image:

Get the desired model:

ollama pull llama3:70b-instruct-q2_K

From within the container, rename whatever to llama2/latest:

mv /root/.ollama/models/manifests/registry.ollama.ai/library/MODEL_NAME /root/.ollama/models/manifests/registry.ollama.ai/library/llama2

mv /root/.ollama/models/manifests/registry.ollama.ai/library/llama2/TAG_NAME /root/.ollama/models/manifests/registry.ollama.ai/library/llama2/latest

@katarinasupe
Copy link
Contributor

Hi @theobjectivedad -- happy news! This feature is coming in Memgraph Lab v2.16 in late August 😄

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
community community feature feature Frequency - Monthly Frequency - Monthly Reach - Some Reach - Some Severity - S3 Severity - S3
Projects
None yet
Development

No branches or pull requests

2 participants