MetaGPT/metagpt/team.py
geekan 22288a342d 1. 动作优化
1. SummarizeCode动作:用于基于代码进行总结,思考bug、逻辑、todo
  2. CodeReview动作优化:目前强制要求回答问题,有更高的成功率了
    1. 增加了LGTM/LBTM的回答,在LGTM时会及时停止,不重写代码
    2. 目前增加了设置中的参数code_review_k_times,与reflexion类似,设置为2
    3. 仍然有概率发生指令不遵循,尤其是会有比较高的概率发生同时review多个代码文件,还没想好怎么解决 #FIXME
  3. 增加了env到Action结构中,现在可以直接调用环境接口了
  4. WriteDesign:去除了对project_name的纠正代码,现在引导下可以一次生成对
    1. 修改了提示词中的##格式,改为了JSON格式
2. 数据结构
  1. Document的标准化:Env->Repo->Document,其中Document/Asset/Code都是Document
    1. 原用于检索的Document改为IndexableDocument
  2. Repo结构引入:用于Document装载与元数据装载
  3. RepoParser引入:写了一个简单的AST parser(后续可能要换tree-sitter),给出了整库symbol
  4. Env中增加了set/get/set_doc/get_doc接口,用于set/get单个变量或者一个Document。这个逻辑后续或许会进一步简化
3. 配置优化
  1. 默认更换为gpt-4-1106-preview,以获得最好的效果与成本
  2. 提供~/.metagpt作为配置最高优先级目录,从中读取config.yaml
  3. workspace可以灵活指定了,在config中配置
  4. project_name可以由命令行指定,并且改为由ProductManager生成
4. metagpt作为默认命令行,而非python startup.py
metagpt --help

metagpt --project-name game_2048 "make a 2048 game"
metagpt "make a 2048 game"

metagpt --project-name game_2048 --inc "将2048改为4096"

metagpt --project-name game_2048 --auto-inc "make a 2048 game"
  1. 使用新的METAGPT_ROOT生成方式,而非寻找git,以便cli安装
  2. 命令行由fire换为了typer,它会带来相对更好的体验
  3. project_name可以灵活指定了,在metagpt命令行输入中配置
5. 其他
  1. 现在支持多国语言了,中文已测试
  2. BossRequirement -> UserRequirement
  3. 大量错误文本的修正,增加了可读性
  4. 中量提示词优化,稍微提升了一些准确率
  5. 暂时屏蔽了LongtermMemory相关逻辑,这个逻辑底层调用了langchain的FAISS,会带来~5秒加载耗时
  6. 修复了安装包中的部分描述错误
  7. 去除了config中在openai_proxy设定时对base的重复修改,这个修改应该在openai初始化时发生
  8. 修复了JSON在中文存储时的特定问题,ensure_ascii=False
2023-11-27 15:48:12 +08:00

68 lines
No EOL
2.3 KiB
Python

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
@Time : 2023/5/12 00:30
@Author : alexanderwu
@File : software_company.py
"""
from pydantic import BaseModel, Field
from metagpt.actions import UserRequirement
from metagpt.config import CONFIG
from metagpt.environment import Environment
from metagpt.logs import logger
from metagpt.roles import Role
from metagpt.schema import Message
from metagpt.utils.common import NoMoneyException
class Team(BaseModel):
"""
Team: Possesses one or more roles (agents), SOP (Standard Operating Procedures), and a platform for instant messaging,
dedicated to perform any multi-agent activity, such as collaboratively writing executable code.
"""
env: Environment = Field(default_factory=Environment)
investment: float = Field(default=10.0)
idea: str = Field(default="")
class Config:
arbitrary_types_allowed = True
def hire(self, roles: list[Role]):
"""Hire roles to cooperate"""
self.env.add_roles(roles)
def invest(self, investment: float):
"""Invest company. raise NoMoneyException when exceed max_budget."""
self.investment = investment
CONFIG.max_budget = investment
logger.info(f'Investment: ${investment}.')
def _check_balance(self):
if CONFIG.total_cost > CONFIG.max_budget:
raise NoMoneyException(CONFIG.total_cost, f'Insufficient funds: {CONFIG.max_budget}')
def run_project(self, idea, send_to: str = "", project_name: str = "", inc: bool = False):
"""Start a project from publishing user requirement."""
self.idea = idea
# If user set project_name, then use it.
if project_name:
path = CONFIG.workspace_path / project_name
self.env.load_existing_repo(path, inc=inc)
# Human requirement.
self.env.publish_message(Message(role="Human", content=idea, cause_by=UserRequirement, send_to=send_to))
def _save(self):
logger.info(self.json(ensure_ascii=False))
async def run(self, n_round=3):
"""Run company until target round or no money"""
while n_round > 0:
# self._save()
n_round -= 1
logger.debug(f"{n_round=}")
self._check_balance()
await self.env.run()
return self.env.history