vrr/agent-zero
2
0
Fork 0
mirror of https://github.com/agent0ai/agent-zero.git synced 2026-04-29 12:10:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refactor: Remove ModelProvider enum

Browse source
This commit is contained in:
linuztx 2025-07-16 18:00:20 +08:00
parent 37615c5a24
commit 4c45b5cf8c
5 changed files with 49 additions and 51 deletions
Download patch file Download diff file Expand all files Collapse all files

6
python/helpers/memory.py
View file

@ -129,7 +129,7 @@ class Memory:
**model_config.build_kwargs(),
)
embeddings_model_id = files.safe_file_name(
model_config.provider.name + "_" + model_config.name
model_config.provider + "_" + model_config.name
)
# here we setup the embeddings model with the chosen cache storage
@ -160,7 +160,7 @@ class Memory:
if files.exists(emb_set_file):
embedding_set = json.loads(files.read_file(emb_set_file))
if (
embedding_set["model_provider"] == model_config.provider.name
embedding_set["model_provider"] == model_config.provider
and embedding_set["model_name"] == model_config.name
):
# model matches
@ -200,7 +200,7 @@ class Memory:
meta_file_path,
json.dumps(
{
"model_provider": model_config.provider.name,
"model_provider": model_config.provider,
"model_name": model_config.name,
}
),