search

soverant / lattice

MIT License
0 stars 0 forks source link

Retrieval Mechanism #6

Open mehdibalouchi opened 1 month ago

mehdibalouchi commented 1 month ago

Code to transform the code graph into database relations. Should create necessary indices

parent issue: #5

amrhssn commented 1 month ago

@mehdibalouchi I have implemented two functions for chatting with Together's LLMs and computing embeddings. You can find it here.

I used the following two models for chat and embedding. I needed them for issue #8 to prepare a setup to start working on the Retrieve module.

together_llm_model_name = "meta-llama/Llama-3-70b-chat-hf"

together_embedding_model_name = "togethercomputer/m2-bert-80M-32k-retrieval"

Feel free to modify them in any way 🙌🏻