scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-10-09 03:20:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(core): add methods to update default LLM and embedding model names

Browse source
This commit is contained in:
jinli.yl 2026-02-20 01:04:04 +08:00
parent b130cf3350
commit 3408e8dcf1
2 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
reme/__init__.py
View file

@ -20,7 +20,7 @@ __all__ = [
"ReMeFs",
]
__version__ = "0.3.0.0b2"
__version__ = "0.3.0.0b3"
"""

8
reme/core/application.py
View file

@ -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."""