apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-06 14:22:46 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #1107 from better629/main

Browse source 
fix provider failed ut
This commit is contained in:
Alexander Wu 2024-03-26 10:26:11 +08:00 • committed by GitHub
commit db7aebaedf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
metagpt/provider/human_provider.py
View file

@ -17,7 +17,7 @@ class HumanProvider(BaseLLM):
"""
def __init__(self, config: LLMConfig):
pass
self.config = config
def ask(self, msg: str, timeout=USE_CONFIG_TIMEOUT) -> str:
logger.info("It's your turn, please type in your response. You may also refer to the context below")

3
tests/metagpt/provider/test_base_llm.py
View file

@ -11,6 +11,7 @@ import pytest
from metagpt.configs.llm_config import LLMConfig
from metagpt.provider.base_llm import BaseLLM
from metagpt.schema import Message
from tests.metagpt.provider.mock_llm_config import mock_llm_config
from tests.metagpt.provider.req_resp_const import (
default_resp_cont,
get_part_chat_completion,
@ -22,7 +23,7 @@ name = "GPT"
class MockBaseLLM(BaseLLM):
def __init__(self, config: LLMConfig = None):
pass
self.config = config or mock_llm_config
def completion(self, messages: list[dict], timeout=3):
return get_part_chat_completion(name)