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

add x.ai support

Browse source
This commit is contained in:
LUIS NOVO 2024-11-11 16:49:50 -03:00
parent 2e2a4947b3
commit ac2ea9e554
3 changed files with 32 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
open_notebook/models/__init__.py
View file

@ -17,6 +17,7 @@ from open_notebook.models.llms import (
OpenRouterLanguageModel,
VertexAILanguageModel,
VertexAnthropicLanguageModel,
XAILanguageModel,
)
from open_notebook.models.speech_to_text_models import (
OpenAISpeechToTextModel,
@ -44,6 +45,7 @@ MODEL_CLASS_MAP: Dict[str, ProviderMap] = {
"anthropic": AnthropicLanguageModel,
"openai": OpenAILanguageModel,
"gemini": GeminiLanguageModel,
"xai": XAILanguageModel,
},
"embedding": {
"openai": OpenAIEmbeddingModel,