vrr/open-notebook
2
0
Fork 0
mirror of https://github.com/lfnovo/open-notebook.git synced 2026-05-02 21:30:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add gemini support

Browse source
This commit is contained in:
LUIS NOVO 2024-10-26 05:41:06 -03:00
parent 7648caca7b
commit aaa7831ab1
4 changed files with 28 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
open_notebook/llm_router.py
View file

@ -1,5 +1,6 @@
from open_notebook.llms import (
AnthropicLanguageModel,
GeminiLanguageModel,
LiteLLMLanguageModel,
OllamaLanguageModel,
OpenAILanguageModel,
@ -17,6 +18,7 @@ PROVIDER_CLASS_MAP = {
"vertexai": VertexAILanguageModel,
"anthropic": AnthropicLanguageModel,
"openai": OpenAILanguageModel,
"gemini": GeminiLanguageModel,
}