|
|
|
|
|
|
|
|
""" |
|
|
@Time : 2023/5/12 00:30 |
|
|
@Author : alexanderwu |
|
|
@Modified From : https://github.com/geekan/MetaGPT/blob/main/metagpt/software_company.py |
|
|
""" |
|
|
from pydantic import BaseModel, Field |
|
|
|
|
|
from .roles import Role |
|
|
from .actions import Requirement |
|
|
from .environment import Environment |
|
|
|
|
|
import cfg |
|
|
from .system.logs import logger |
|
|
from .system.schema import Message |
|
|
from .system.utils.common import NoMoneyException |
|
|
|
|
|
|
|
|
class Explorer(BaseModel): |
|
|
environment: Environment = Field(default_factory=Environment) |
|
|
investment: float = Field(default=10.0) |
|
|
|
|
|
class Config: |
|
|
arbitrary_types_allowed = True |
|
|
|
|
|
def hire(self, roles: list[Role]): |
|
|
self.environment.add_roles(roles) |
|
|
|
|
|
def invest(self, investment: float): |
|
|
self.investment = investment |
|
|
cfg.MAX_BUDGET = investment |
|
|
logger.info(f'Investment: ${investment}.') |
|
|
|
|
|
def _check_balance(self): |
|
|
if cfg.TOTAL_COST > cfg.MAX_BUDGET: |
|
|
raise NoMoneyException(cfg.TOTAL_COST, f'Insufficient funds: {cfg.MAX_BUDGET}') |
|
|
|
|
|
async def start_project(self, idea=None, llm_api_key=None, proxy=None, serpapi_key=None, task_id=None, alg_msg_queue=None): |
|
|
self.environment.llm_api_key = llm_api_key |
|
|
self.environment.proxy = proxy |
|
|
self.environment.task_id = task_id |
|
|
self.environment.alg_msg_queue = alg_msg_queue |
|
|
self.environment.serpapi_key = serpapi_key |
|
|
|
|
|
await self.environment.publish_message(Message(role="Question/Task", content=idea, cause_by=Requirement)) |
|
|
|
|
|
def _save(self): |
|
|
logger.info(self.json()) |
|
|
|
|
|
async def run(self, n_round=3): |
|
|
while n_round > 0: |
|
|
|
|
|
n_round -= 1 |
|
|
logger.debug(f"{n_round=}") |
|
|
self._check_balance() |
|
|
await self.environment.run() |
|
|
return self.environment.history |
|
|
|
