apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-25 08:46:48 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

use model_name in embedding

Browse source
This commit is contained in:
seehi 2024-05-24 17:18:56 +08:00
parent db26406b95
commit 5f1ca3ca7e
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/metagpt/rag/factories/test_embedding.py
View file

@ -66,7 +66,7 @@ class TestRAGEmbeddingFactory:
@pytest.mark.parametrize(
"model, embed_batch_size, expected_params",
[("test_model", 100, {"model": "test_model", "embed_batch_size": 100}), (None, None, {})],
[("test_model", 100, {"model_name": "test_model", "embed_batch_size": 100}), (None, None, {})],
)
def test_try_set_model_and_batch_size(self, mock_config, model, embed_batch_size, expected_params):
# Mock