apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-20 15:38:09 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

rename ConfigFactory to RAGConfigRegistry

Browse source
This commit is contained in:
seehi 2024-03-15 15:36:10 +08:00
parent e58cef6f18
commit cb9543b2b9
5 changed files with 13 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/metagpt/rag/factories/test_base.py
View file

@ -1,6 +1,6 @@
import pytest
from metagpt.rag.factories.base import ConfigFactory, GenericFactory
from metagpt.rag.factories.base import GenericFactory, RAGConfigRegistry
class TestGenericFactory:
@ -55,7 +55,7 @@ class DummyConfig:
self.name = name
class TestConfigFactory:
class TestRAGConfigRegistry:
@pytest.fixture
def config_creators(self):
return {
@ -64,7 +64,7 @@ class TestConfigFactory:
@pytest.fixture
def config_factory(self, config_creators):
return ConfigFactory(creators=config_creators)
return RAGConfigRegistry(creators=config_creators)
def test_get_instance_success(self, config_factory):
# Test successful retrieval of an instance
@ -85,18 +85,18 @@ class TestConfigFactory:
def test_val_from_config_or_kwargs_priority(self):
# Test that the value from the config object has priority over kwargs
config = DummyConfig(name="ConfigName")
result = ConfigFactory._val_from_config_or_kwargs("name", config, name="KwargsName")
result = RAGConfigRegistry._val_from_config_or_kwargs("name", config, name="KwargsName")
assert result == "ConfigName"
def test_val_from_config_or_kwargs_fallback_to_kwargs(self):
# Test fallback to kwargs when config object does not have the value
config = DummyConfig(name=None)
result = ConfigFactory._val_from_config_or_kwargs("name", config, name="KwargsName")
result = RAGConfigRegistry._val_from_config_or_kwargs("name", config, name="KwargsName")
assert result == "KwargsName"
def test_val_from_config_or_kwargs_key_error(self):
# Test KeyError when the key is not found in both config object and kwargs
config = DummyConfig(name=None)
with pytest.raises(KeyError) as exc_info:
ConfigFactory._val_from_config_or_kwargs("missing_key", config)
RAGConfigRegistry._val_from_config_or_kwargs("missing_key", config)
assert "The key 'missing_key' is required but not provided" in str(exc_info.value)