Canner / WrenAI

🚀 Open-source SQL AI Agent for Text-to-SQL. Make Text2SQL Easy! 🙌
https://getwren.ai/oss
GNU Affero General Public License v3.0
1.73k stars 155 forks source link

Implement GoogleAIGeminiGenerator as a new custom LLM #469

Open cyyeh opened 2 months ago

cyyeh commented 2 months ago

Currently we officially support several types of LLMs, such as OpenAI, OpenAI api-compatible and Ollama. We also released a documentation on how to write your custom LLM: https://docs.getwren.ai/installation/custom_llm#adding-a-custom-llm-or-document-store-to-wren-ai

I hope this issue can attract the community members who are interested to contribute to GoogleAIGeminiGenerator LLM based on the guide provided above!

If you would like to contribute, please comment below and let others know. If you have any issues for the implementation, welcome to raise below and discuss with us. Thank you 💯