diff --git a/reme/__init__.py b/reme/__init__.py index 100fa70f..537864d0 100644 --- a/reme/__init__.py +++ b/reme/__init__.py @@ -20,7 +20,7 @@ __all__ = [ "ReMeFs", ] -__version__ = "0.3.0.0b2" +__version__ = "0.3.0.0b3" """ diff --git a/reme/core/application.py b/reme/core/application.py index 2ac457e4..c1792f5a 100644 --- a/reme/core/application.py +++ b/reme/core/application.py @@ -169,6 +169,10 @@ class Application: """Get an LLM instance by name.""" return self.service_context.llms.get(name) + def update_default_llm_name(self, name: str): + """Update the default LLM name.""" + self.default_llm.model_name = name + @property def default_embedding_model(self) -> BaseEmbeddingModel: """Get the default embedding model instance.""" @@ -178,6 +182,10 @@ class Application: """Get an embedding model instance by name.""" return self.service_context.embedding_models.get(name) + def update_default_embedding_name(self, name: str): + """Update the default embedding model name.""" + self.default_embedding_model.model_name = name + @property def default_vector_store(self) -> BaseVectorStore: """Get the default vector store instance."""