Skip to content

NeuroML/neuroml-ai

Repository files navigation

NeuroML-AI

AI assistant for helping with NeuroML queries and model generation.

GitHub CI GitHub GitHub pull requests GitHub issues GitHub Org's stars Twitter Follow Gitter DOI

Please note that this project is under active development and does not currently provide a stable release/API/ABI.

Set up

The package is pip installable.

Models

The following models are currently being used for testing:

Ollama for local deployments

Currently, Ollama is used for local deployments. Please see the code to see what models are currently used. You can modify these to use different models to suit your hardware. However, do note that picking smaller models will most certainly affect the correctness/performance of the RAG. To install Ollama and pull the models, please follow the official documentation: https://ollama.com/download

Gemini

You can use the Gemini chat model and embeddings. For this, please export the GOOGLE_API_KEY environment variable.

License

This code is licensed under the MIT license. Please refer to the licenses of the various LLM models for information on their licensing and usage terms.

About

WIP: RAG for NeuroML

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published