mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-06-14 15:25:17 +02:00
处理冲突
This commit is contained in:
commit
340e1486d2
9 changed files with 214 additions and 7 deletions
27
tests/data/config/config2.yaml
Normal file
27
tests/data/config/config2.yaml
Normal file
|
|
@ -0,0 +1,27 @@
|
|||
llm:
|
||||
api_type: "openai" # or azure / ollama / groq etc.
|
||||
base_url: "YOUR_gpt-3.5-turbo_BASE_URL"
|
||||
api_key: "YOUR_gpt-3.5-turbo_API_KEY"
|
||||
model: "gpt-3.5-turbo" # or gpt-3.5-turbo
|
||||
# proxy: "YOUR_gpt-3.5-turbo_PROXY" # for LLM API requests
|
||||
# timeout: 600 # Optional. If set to 0, default value is 300.
|
||||
# Details: https://azure.microsoft.com/en-us/pricing/details/cognitive-services/openai-service/
|
||||
pricing_plan: "" # Optional. Use for Azure LLM when its model name is not the same as OpenAI's
|
||||
|
||||
models:
|
||||
"YOUR_MODEL_NAME_1": # model: "gpt-4-turbo" # or gpt-3.5-turbo
|
||||
api_type: "openai" # or azure / ollama / groq etc.
|
||||
base_url: "YOUR_MODEL_1_BASE_URL"
|
||||
api_key: "YOUR_MODEL_1_API_KEY"
|
||||
# proxy: "YOUR_MODEL_1_PROXY" # for LLM API requests
|
||||
# timeout: 600 # Optional. If set to 0, default value is 300.
|
||||
# Details: https://azure.microsoft.com/en-us/pricing/details/cognitive-services/openai-service/
|
||||
pricing_plan: "" # Optional. Use for Azure LLM when its model name is not the same as OpenAI's
|
||||
"YOUR_MODEL_NAME_2": # model: "gpt-4-turbo" # or gpt-3.5-turbo
|
||||
api_type: "openai" # or azure / ollama / groq etc.
|
||||
base_url: "YOUR_MODEL_2_BASE_URL"
|
||||
api_key: "YOUR_MODEL_2_API_KEY"
|
||||
proxy: "YOUR_MODEL_2_PROXY" # for LLM API requests
|
||||
# timeout: 600 # Optional. If set to 0, default value is 300.
|
||||
# Details: https://azure.microsoft.com/en-us/pricing/details/cognitive-services/openai-service/
|
||||
pricing_plan: "" # Optional. Use for Azure LLM when its model name is not the same as OpenAI's
|
||||
0
tests/metagpt/configs/__init__.py
Normal file
0
tests/metagpt/configs/__init__.py
Normal file
34
tests/metagpt/configs/test_models_config.py
Normal file
34
tests/metagpt/configs/test_models_config.py
Normal file
|
|
@ -0,0 +1,34 @@
|
|||
import pytest
|
||||
|
||||
from metagpt.actions.talk_action import TalkAction
|
||||
from metagpt.configs.models_config import ModelsConfig
|
||||
from metagpt.const import METAGPT_ROOT, TEST_DATA_PATH
|
||||
from metagpt.utils.common import aread, awrite
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_models_configs(context):
|
||||
default_model = ModelsConfig.default()
|
||||
assert default_model is not None
|
||||
|
||||
models = ModelsConfig.from_yaml_file(TEST_DATA_PATH / "config/config2.yaml")
|
||||
assert models
|
||||
|
||||
default_models = ModelsConfig.default()
|
||||
backup = ""
|
||||
if not default_models.models:
|
||||
backup = await aread(filename=METAGPT_ROOT / "config/config2.yaml")
|
||||
test_data = await aread(filename=TEST_DATA_PATH / "config/config2.yaml")
|
||||
await awrite(filename=METAGPT_ROOT / "config/config2.yaml", data=test_data)
|
||||
|
||||
try:
|
||||
action = TalkAction(context=context, i_context="who are you?", llm_name_or_type="YOUR_MODEL_NAME_1")
|
||||
assert action.private_llm.config.model == "YOUR_MODEL_NAME_1"
|
||||
assert context.config.llm.model != "YOUR_MODEL_NAME_1"
|
||||
finally:
|
||||
if backup:
|
||||
await awrite(filename=METAGPT_ROOT / "config/config2.yaml", data=backup)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
pytest.main([__file__, "-s"])
|
||||
Loading…
Add table
Add a link
Reference in a new issue