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

support ollama, vertex and gemini embedding models

Browse source
This commit is contained in:
LUIS NOVO 2024-10-30 15:19:16 -03:00
parent 2f07f0ab49
commit f64897fbf8
6 changed files with 122 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
open_notebook/models/__init__.py
View file

@ -1,5 +1,10 @@
from open_notebook.domain.models import Model
from open_notebook.models.embedding_models import OpenAIEmbeddingModel
from open_notebook.models.embedding_models import (
GeminiEmbeddingModel,
OllamaEmbeddingModel,
OpenAIEmbeddingModel,
VertexEmbeddingModel,
)
from open_notebook.models.llms import (
AnthropicLanguageModel,
GeminiLanguageModel,
@ -26,6 +31,9 @@ MODEL_CLASS_MAP = {
},
"embedding": {
"openai": OpenAIEmbeddingModel,
"gemini": GeminiEmbeddingModel,
"vertexai": VertexEmbeddingModel,
"ollama": OllamaEmbeddingModel,
},
"speech_to_text": {
"openai": OpenAISpeechToTextModel,