fixbug: role option, cost_manager argments

This commit is contained in:
莘权 马 2023-08-22 18:49:39 +08:00
parent a2e9797d4e
commit 8eaf22dd62
4 changed files with 11 additions and 5 deletions

View file

@ -26,9 +26,11 @@ DESC = """
class CustomerService(Sales):
def __init__(
self,
options,
cost_manager,
name="Xiaomei",
profile="Human customer service",
desc=DESC,
store=None
):
super().__init__(name, profile, desc=desc, store=store)
super().__init__(options=options, cost_manager=cost_manager, name=name, profile=profile, desc=desc, store=store)

View file

@ -22,6 +22,8 @@ class Report(BaseModel):
class Researcher(Role):
def __init__(
self,
options,
cost_manager,
name: str = "David",
profile: str = "Researcher",
goal: str = "Gather information and conduct research",
@ -29,7 +31,7 @@ class Researcher(Role):
language: str = "en-us",
**kwargs,
):
super().__init__(name, profile, goal, constraints, **kwargs)
super().__init__(options=options, cost_manager=cost_manager, name=name, profile=profile, goal=goal, constraints=constraints, **kwargs)
self._init_actions([CollectLinks(name), WebBrowseAndSummarize(name), ConductResearch(name)])
self.language = language
if language not in ("en-us", "zh-cn"):

View file

@ -13,6 +13,8 @@ from metagpt.tools import SearchEngineType
class Sales(Role):
def __init__(
self,
options,
cost_manager,
name="Xiaomei",
profile="Retail sales guide",
desc="I am a sales guide in retail. My name is Xiaomei. I will answer some customer questions next, and I "
@ -23,7 +25,7 @@ class Sales(Role):
"professional guide",
store=None
):
super().__init__(name, profile, desc=desc)
super().__init__(options=options, cost_manager=cost_manager, name=name, profile=profile, desc=desc)
self._set_store(store)
def _set_store(self, store):

View file

@ -13,9 +13,9 @@ from metagpt.tools import SearchEngineType
class Searcher(Role):
def __init__(self, name='Alice', profile='Smart Assistant', goal='Provide search services for users',
def __init__(self, options, cost_manager, name='Alice', profile='Smart Assistant', goal='Provide search services for users',
constraints='Answer is rich and complete', engine=SearchEngineType.SERPAPI_GOOGLE, **kwargs):
super().__init__(name, profile, goal, constraints, **kwargs)
super().__init__(options=options, cost_manager=cost_manager, name=name, profile=profile, goal=goal, constraints=constraints, **kwargs)
self._init_actions([SearchAndSummarize(engine=engine)])
def set_search_func(self, search_func):