mirror of
https://github.com/imartinez/privateGPT.git
synced 2025-04-27 03:11:11 +00:00
fc13368bc7
* Support for Google Gemini LLMs and Embeddings Initial support for Gemini, enables usage of Google LLMs and embedding models (see settings-gemini.yaml) Install via poetry install --extras "llms-gemini embeddings-gemini" Notes: * had to bump llama-index-core to later version that supports Gemini * poetry --no-update did not work: Gemini/llama_index seem to require more (transient) updates to make it work... * fix: crash when gemini is not selected * docs: add gemini llm --------- Co-authored-by: Javier Martinez <javiermartinezalvarez98@gmail.com>
11 lines
153 B
YAML
11 lines
153 B
YAML
llm:
|
|
mode: gemini
|
|
|
|
embedding:
|
|
mode: gemini
|
|
|
|
gemini:
|
|
api_key: ${GOOGLE_API_KEY:}
|
|
model: models/gemini-pro
|
|
embedding_model: models/embedding-001
|