Using general RAG implementation #44
Description
@pgleeson @SabarishG18 : I've split the RAG out as a generic RAG here now: https://github.com/NeuroML/neuroml-ai/tree/feat-code/rag_pkg (the feat-code branch). This seems fairly complete, so I won't make any major changes to this now. The code assistant etc. bits are now separate modules, so I'll continue working on them there.
It's configurable at runtime (see the Readme there). So you should be able to specify the different OpenWorm vector stores to it and use it as it is. I've tested this out with the NeuroML vector stores so far, and it works fine. See the vector-stores.json file for example.
The package also includes a CLI and Streamlit UI, which should also just work (also noted in Readme).
Do let me know if you run into any issues