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

fix conflict

Browse source
This commit is contained in:
程茂宇 2023-07-24 14:24:41 +08:00
commit 4975baa890
110 changed files with 950 additions and 248 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tests/metagpt/actions/mock.py
View file

@ -510,4 +510,3 @@ Process finished with exit code 1'''
MEILI_CODE_REFINED = """
"""

3
tests/metagpt/actions/test_action.py
View file

@ -5,9 +5,6 @@
@Author : alexanderwu
@File : test_action.py
"""
import pytest
from metagpt.logs import logger
from metagpt.actions import Action, WritePRD, WriteTest

3
tests/metagpt/actions/test_action_output.py
View file

@ -5,9 +5,10 @@
@Author : chengmaoyu
@File : test_action_output
"""
from metagpt.actions import ActionOutput
from typing import List, Tuple
from metagpt.actions import ActionOutput
t_dict = {"Required Python third-party packages": "\"\"\"\nflask==1.1.2\npygame==2.0.1\n\"\"\"\n",
"Required Other language third-party packages": "\"\"\"\nNo third-party packages required for other languages.\n\"\"\"\n",
"Full API spec": "\"\"\"\nopenapi: 3.0.0\ninfo:\n title: Web Snake Game API\n version: 1.0.0\npaths:\n /game:\n get:\n summary: Get the current game state\n responses:\n '200':\n description: A JSON object of the game state\n post:\n summary: Send a command to the game\n requestBody:\n required: true\n content:\n application/json:\n schema:\n type: object\n properties:\n command:\n type: string\n responses:\n '200':\n description: A JSON object of the updated game state\n\"\"\"\n",

3
tests/metagpt/actions/test_debug_error.py
View file

@ -6,6 +6,7 @@
@File : test_debug_error.py
"""
import pytest
from metagpt.actions.debug_error import DebugError
@ -13,12 +14,10 @@ from metagpt.actions.debug_error import DebugError
async def test_debug_error():
code = "def add(a, b):\n return a - b"
error = "AssertionError: Expected add(1, 1) to equal 2 but got 0"
fixed_code = "def add(a, b):\n return a + b"
debug_error = DebugError("debug_error")
result = await debug_error.run(code, error)
prompt = f"以下是一段Python代码:\n\n{code}\n\n执行时发生了以下错误:\n\n{error}\n\n请尝试修复这段代码中的错误。"
# mock_llm.ask.assert_called_once_with(prompt)
assert len(result) > 0

5
tests/metagpt/actions/test_design_api.py
View file

@ -7,11 +7,8 @@
"""
import pytest
from metagpt.logs import logger
from metagpt.actions.design_api import WriteDesign
from metagpt.llm import LLM
from metagpt.roles.architect import Architect
from metagpt.logs import logger
from tests.metagpt.actions.mock import PRD_SAMPLE

4
tests/metagpt/actions/test_design_api_review.py
View file

@ -24,12 +24,12 @@ API列表:
3. next(): 跳到播放列表的下一首歌曲。
4. previous(): 跳到播放列表的上一首歌曲。
"""
api_review = "API设计看起来非常合理，满足了PRD中的所有需求。"
_ = "API设计看起来非常合理，满足了PRD中的所有需求。"
design_api_review = DesignReview("design_api_review")
result = await design_api_review.run(prd, api_design)
prompt = f"以下是产品需求文档(PRD):\n\n{prd}\n\n以下是基于这个PRD设计的API列表:\n\n{api_design}\n\n请审查这个API设计是否满足PRD的需求，以及是否符合良好的设计实践。"
_ = f"以下是产品需求文档(PRD):\n\n{prd}\n\n以下是基于这个PRD设计的API列表:\n\n{api_design}\n\n请审查这个API设计是否满足PRD的需求，以及是否符合良好的设计实践。"
# mock_llm.ask.assert_called_once_with(prompt)
assert len(result) > 0

2
tests/metagpt/actions/test_project_management.py
View file

@ -6,8 +6,6 @@
@File : test_project_management.py
"""
from metagpt.actions.project_management import WriteTasks, AssignTasks
class TestCreateProjectPlan:
pass

2
tests/metagpt/actions/test_run_code.py
View file

@ -6,6 +6,7 @@
@File : test_run_code.py
"""
import pytest
from metagpt.actions.run_code import RunCode
@ -35,4 +36,3 @@ result = add(1, '2')
result = await run_code.run(code)
assert "TypeError: unsupported operand type(s) for +" in result

191
tests/metagpt/actions/test_ui_design.py Normal file
View file

@ -0,0 +1,191 @@
# -*- coding: utf-8 -*-
# @Date : 2023/7/22 02:40
# @Author : stellahong (stellahong@fuzhi.ai)
#
from tests.metagpt.roles.ui_role import UIDesign
llm_resp= '''
# UI Design Description
```The user interface for the snake game will be designed in a way that is simple, clean, and intuitive. The main elements of the game such as the game grid, snake, food, score, and game over message will be clearly defined and easy to understand. The game grid will be centered on the screen with the score displayed at the top. The game controls will be intuitive and easy to use. The design will be modern and minimalist with a pleasing color scheme.```
## Selected Elements
Game Grid: The game grid will be a rectangular area in the center of the screen where the game will take place. It will be defined by a border and will have a darker background color.
Snake: The snake will be represented by a series of connected blocks that move across the grid. The color of the snake will be different from the background color to make it stand out.
Food: The food will be represented by small objects that are a different color from the snake and the background. The food will be randomly placed on the grid.
Score: The score will be displayed at the top of the screen. The score will increase each time the snake eats a piece of food.
Game Over: When the game is over, a message will be displayed in the center of the screen. The player will be given the option to restart the game.
## HTML Layout
```html
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Snake Game</title>
<link rel="stylesheet" href="styles.css">
</head>
<body>
<div class="score">Score: 0</div>
<div class="game-grid">
<!-- Snake and food will be dynamically generated here using JavaScript -->
</div>
<div class="game-over">Game Over</div>
</body>
</html>
```
## CSS Styles (styles.css)
```css
body {
display: flex;
flex-direction: column;
justify-content: center;
align-items: center;
height: 100vh;
margin: 0;
background-color: #f0f0f0;
}
.score {
font-size: 2em;
margin-bottom: 1em;
}
.game-grid {
width: 400px;
height: 400px;
display: grid;
grid-template-columns: repeat(20, 1fr);
grid-template-rows: repeat(20, 1fr);
gap: 1px;
background-color: #222;
border: 1px solid #555;
}
.snake-segment {
background-color: #00cc66;
}
.food {
background-color: #cc3300;
}
.control-panel {
display: flex;
justify-content: space-around;
width: 400px;
margin-top: 1em;
}
.control-button {
padding: 1em;
font-size: 1em;
border: none;
background-color: #555;
color: #fff;
cursor: pointer;
}
.game-over {
position: absolute;
top: 50%;
left: 50%;
transform: translate(-50%, -50%);
font-size: 3em;
'''
def test_ui_design_parse_css():
ui_design_work = UIDesign(name="UI design action")
css = '''
body {
display: flex;
flex-direction: column;
justify-content: center;
align-items: center;
height: 100vh;
margin: 0;
background-color: #f0f0f0;
}
.score {
font-size: 2em;
margin-bottom: 1em;
}
.game-grid {
width: 400px;
height: 400px;
display: grid;
grid-template-columns: repeat(20, 1fr);
grid-template-rows: repeat(20, 1fr);
gap: 1px;
background-color: #222;
border: 1px solid #555;
}
.snake-segment {
background-color: #00cc66;
}
.food {
background-color: #cc3300;
}
.control-panel {
display: flex;
justify-content: space-around;
width: 400px;
margin-top: 1em;
}
.control-button {
padding: 1em;
font-size: 1em;
border: none;
background-color: #555;
color: #fff;
cursor: pointer;
}
.game-over {
position: absolute;
top: 50%;
left: 50%;
transform: translate(-50%, -50%);
font-size: 3em;
'''
assert ui_design_work.parse_css_code(context=llm_resp)==css
def test_ui_design_parse_html():
ui_design_work = UIDesign(name="UI design action")
html = '''
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Snake Game</title>
<link rel="stylesheet" href="styles.css">
</head>
<body>
<div class="score">Score: 0</div>
<div class="game-grid">
<!-- Snake and food will be dynamically generated here using JavaScript -->
</div>
<div class="game-over">Game Over</div>
</body>
</html>
'''
assert ui_design_work.parse_css_code(context=llm_resp)==html

5
tests/metagpt/actions/test_write_code.py
View file

@ -6,10 +6,11 @@
@File : test_write_code.py
"""
import pytest
from metagpt.logs import logger
from metagpt.actions.write_code import WriteCode
from tests.metagpt.actions.mock import WRITE_CODE_PROMPT_SAMPLE, TASKS_2
from metagpt.llm import LLM
from metagpt.logs import logger
from tests.metagpt.actions.mock import TASKS_2, WRITE_CODE_PROMPT_SAMPLE
@pytest.mark.asyncio

4
tests/metagpt/actions/test_write_code_review.py
View file

@ -6,9 +6,9 @@
@File : test_write_code_review.py
"""
import pytest
from metagpt.logs import logger
from metagpt.llm import LLM
from metagpt.actions.write_code_review import WriteCodeReview
from metagpt.logs import logger
from tests.metagpt.actions.mock import SEARCH_CODE_SAMPLE

3
tests/metagpt/actions/test_write_prd.py
View file

@ -6,8 +6,9 @@
@File : test_write_prd.py
"""
import pytest
from metagpt.actions import BossRequirement
from metagpt.logs import logger
from metagpt.actions import WritePRD, BossRequirement
from metagpt.roles.product_manager import ProductManager
from metagpt.schema import Message

1
tests/metagpt/actions/test_write_prd_review.py
View file

@ -6,6 +6,7 @@
@File : test_write_prd_review.py
"""
import pytest
from metagpt.actions.write_prd_review import WritePRDReview

2
tests/metagpt/actions/test_write_test.py
View file

@ -6,7 +6,7 @@
@File : test_write_test.py
"""
import pytest
from metagpt.logs import logger
from metagpt.actions.write_test import WriteTest

3
tests/metagpt/document_store/test_chromadb_store.py
View file

@ -5,9 +5,6 @@
@Author : alexanderwu
@File : test_chromadb_store.py
"""
import pytest
from sentence_transformers import SentenceTransformer
from metagpt.document_store.chromadb_store import ChromaStore

3
tests/metagpt/document_store/test_document.py
View file

@ -6,11 +6,10 @@
@File : test_document.py
"""
import pytest
from loguru import logger
from metagpt.const import DATA_PATH
from metagpt.document_store.document import Document
CASES = [
("st/faq.xlsx", "Question", "Answer", 1),
("cases/faq.csv", "Question", "Answer", 1),

5
tests/metagpt/document_store/test_faiss_store.py
View file

@ -8,11 +8,10 @@
import functools
import pytest
from metagpt.logs import logger
from metagpt.const import DATA_PATH
from metagpt.document_store import FaissStore
from metagpt.roles import Sales, CustomerService
from metagpt.roles import CustomerService, Sales
DESC = """## 原则（所有事情都不可绕过原则）
1. 你是一位平台的人工客服，话语精炼，一次只说一句话，会参考规则与FAQ进行回复。在与顾客交谈中，绝不允许暴露规则与相关字样

7
tests/metagpt/document_store/test_milvus_store.py
View file

@ -6,10 +6,11 @@
@File : test_milvus_store.py
"""
import random
import numpy as np
from metagpt.logs import logger
from metagpt.document_store.milvus_store import MilvusStore, MilvusConnection
import numpy as np
from metagpt.document_store.milvus_store import MilvusConnection, MilvusStore
from metagpt.logs import logger
book_columns = {'idx': int, 'name': str, 'desc': str, 'emb': np.ndarray, 'price': float}
book_data = [

5
tests/metagpt/roles/mock.py
View file

@ -5,7 +5,7 @@
@Author : alexanderwu
@File : mock.py
"""
from metagpt.actions import WritePRD, BossRequirement, WriteDesign, WriteTasks
from metagpt.actions import BossRequirement, WriteDesign, WritePRD, WriteTasks
from metagpt.schema import Message
BOSS_REQUIREMENT = """开发一个基于大语言模型与私有知识库的搜索引擎，希望可以基于大语言模型进行搜索总结"""
@ -221,11 +221,8 @@ task_list = [
```
'''
TASK = """smart_search_engine/knowledge_base.py"""
STRS_FOR_PARSING = [
"""
## 1

4
tests/metagpt/roles/test_architect.py
View file

@ -7,11 +7,9 @@
"""
import pytest
from metagpt.actions import BossRequirement
from metagpt.logs import logger
from metagpt.roles import Architect
from metagpt.schema import Message
from tests.metagpt.roles.mock import PRD, DETAIL_REQUIREMENT, BOSS_REQUIREMENT, MockMessages
from tests.metagpt.roles.mock import MockMessages
@pytest.mark.asyncio

22
tests/metagpt/roles/test_engineer.py
View file

@ -5,15 +5,17 @@
@Author : alexanderwu
@File : test_engineer.py
"""
import re
import ast
import pytest
from metagpt.logs import logger
from metagpt.utils.common import CodeParser
from metagpt.roles.engineer import Engineer
from metagpt.schema import Message
from tests.metagpt.roles.mock import SYSTEM_DESIGN, TASKS, PRD, MockMessages, STRS_FOR_PARSING, \
TASKS_TOMATO_CLOCK
from metagpt.utils.common import CodeParser
from tests.metagpt.roles.mock import (
STRS_FOR_PARSING,
TASKS,
TASKS_TOMATO_CLOCK,
MockMessages,
)
@pytest.mark.asyncio
@ -63,6 +65,9 @@ def test_parse_file_list():
assert isinstance(tasks, list)
assert target_list == tasks
file_list = CodeParser.parse_file_list("Task list", TASKS_TOMATO_CLOCK, lang="python")
logger.info(file_list)
target_code = """task_list = [
"smart_search_engine/knowledge_base.py",
@ -85,8 +90,3 @@ def test_parse_code():
logger.info(code)
assert isinstance(code, str)
assert target_code == code
def test_parse_file_list():
file_list = CodeParser.parse_file_list("Task list", TASKS_TOMATO_CLOCK, lang="python")
logger.info(file_list)

6
tests/metagpt/roles/test_product_manager.py
View file

@ -6,12 +6,10 @@
@File : test_product_manager.py
"""
import pytest
from metagpt.logs import logger
from metagpt.actions import BossRequirement
from metagpt.logs import logger
from metagpt.roles import ProductManager
from metagpt.schema import Message
from tests.metagpt.roles.mock import DETAIL_REQUIREMENT, BOSS_REQUIREMENT, MockMessages
from tests.metagpt.roles.mock import MockMessages
@pytest.mark.asyncio

4
tests/metagpt/roles/test_project_manager.py
View file

@ -6,10 +6,10 @@
@File : test_project_manager.py
"""
import pytest
from metagpt.logs import logger
from metagpt.roles import ProjectManager
from metagpt.schema import Message
from tests.metagpt.roles.mock import SYSTEM_DESIGN, MockMessages
from tests.metagpt.roles.mock import MockMessages
@pytest.mark.asyncio

1
tests/metagpt/roles/test_qa_engineer.py
View file

@ -5,4 +5,3 @@
@Author : alexanderwu
@File : test_qa_engineer.py
"""

22
tests/metagpt/roles/test_ui.py Normal file
View file

@ -0,0 +1,22 @@
# -*- coding: utf-8 -*-
# @Date : 2023/7/22 02:40
# @Author : stellahong (stellahong@fuzhi.ai)
#
from metagpt.software_company import SoftwareCompany
from metagpt.roles import ProductManager
from tests.metagpt.roles.ui_role import UI
def test_add_ui():
ui = UI()
assert ui.profile == "UI Design"
async def test_ui_role(idea: str, investment: float = 3.0, n_round: int = 5):
"""Run a startup. Be a boss."""
company = SoftwareCompany()
company.hire([ProductManager(), UI()])
company.invest(investment)
company.start_project(idea)
await company.run(n_round=n_round)

276
tests/metagpt/roles/ui_role.py Normal file
View file

@ -0,0 +1,276 @@
# -*- coding: utf-8 -*-
# @Date : 2023/7/15 16:40
# @Author : stellahong (stellahong@fuzhi.ai)
# @Desc :
import re
import os
from importlib import import_module
from functools import wraps
from metagpt.logs import logger
from metagpt.actions import Action, ActionOutput
from metagpt.roles import ProductManager, Role
from metagpt.schema import Message
from metagpt.const import WORKSPACE_ROOT
from metagpt.actions import WritePRD
from metagpt.software_company import SoftwareCompany
from metagpt.tools.sd_engine import SDEngine
PROMPT_TEMPLATE = '''
# Context
{context}
## Format example
{format_example}
-----
Role: You are a UserInterface Designer; the goal is to finish a UI design according to PRD, give a design description, and select specified elements and UI style.
Requirements: Based on the context, fill in the following missing information, provide detailed HTML and CSS code
Attention: Use '##' to split sections, not '#', and '## <SECTION_NAME>' SHOULD WRITE BEFORE the code and triple quote.
## UI Design Description:Provide as Plain text, place the design objective here
## Selected Elements:Provide as Plain text, up to 5 specified elements, clear and simple
## HTML Layout:Provide as Plain text, use standard HTML code
## CSS Styles (styles.css):Provide as Plain text,use standard css code
## Anything UNCLEAR:Provide as Plain text. Make clear here.
'''
FORMAT_EXAMPLE = '''
## UI Design Description
```Snake games are classic and addictive games with simple yet engaging elements. Here are the main elements commonly found in snake games ```
## Selected Elements
Game Grid: The game grid is a rectangular...
Snake: The player controls a snake that moves across the grid...
Food: Food items (often represented as small objects or differently colored blocks)
Score: The player's score increases each time the snake eats a piece of food. The longer the snake becomes, the higher the score.
Game Over: The game ends when the snake collides with itself or an obstacle. At this point, the player's final score is displayed, and they are given the option to restart the game.
## HTML Layout
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Snake Game</title>
<link rel="stylesheet" href="styles.css">
</head>
<body>
<div class="game-grid">
<!-- Snake will be dynamically generated here using JavaScript -->
</div>
<div class="food">
<!-- Food will be dynamically generated here using JavaScript -->
</div>
</body>
</html>
## CSS Styles (styles.css)
body {
display: flex;
justify-content: center;
align-items: center;
height: 100vh;
margin: 0;
background-color: #f0f0f0;
}
.game-grid {
width: 400px;
height: 400px;
display: grid;
grid-template-columns: repeat(20, 1fr); /* Adjust to the desired grid size */
grid-template-rows: repeat(20, 1fr);
gap: 1px;
background-color: #222;
border: 1px solid #555;
}
.game-grid div {
width: 100%;
height: 100%;
background-color: #444;
}
.snake-segment {
background-color: #00cc66; /* Snake color */
}
.food {
width: 100%;
height: 100%;
background-color: #cc3300; /* Food color */
position: absolute;
}
/* Optional styles for a simple game over message */
.game-over {
position: absolute;
top: 50%;
left: 50%;
transform: translate(-50%, -50%);
font-size: 24px;
font-weight: bold;
color: #ff0000;
display: none;
}
## Anything UNCLEAR
There are no unclear points.
'''
OUTPUT_MAPPING = {
"UI Design Description": (str, ...),
"Selected Elements": (str, ...),
"HTML Layout": (str, ...),
"CSS Styles (styles.css)": (str, ...),
"Anything UNCLEAR": (str, ...),
}
def load_engine(func):
"""Decorator to load an engine by file name and engine name."""
@wraps(func)
def wrapper(*args, **kwargs):
file_name, engine_name = func(*args, **kwargs)
engine_file = import_module(file_name, package='metagpt')
ip_module_cls = getattr(engine_file, engine_name)
try:
engine = ip_module_cls()
except:
engine = None
return engine
return wrapper
def parse(func):
"""Decorator to parse information using regex pattern."""
@wraps(func)
def wrapper(*args, **kwargs):
context, pattern = func(*args, **kwargs)
match = re.search(pattern, context, re.DOTALL)
if match:
text_info = match.group(1)
logger.info(text_info)
else:
text_info = context
logger.info("未找到匹配的内容")
return text_info
return wrapper
class UIDesign(Action):
"""Class representing the UI Design action."""
def __init__(self, name, context=None, llm=None):
super().__init__(name, context, llm) # 需要调用LLM进一步丰富UI设计的prompt
@parse
def parse_requirement(self, context: str):
"""Parse UI Design draft from the context using regex."""
pattern = r"## UI Design draft.*?\n(.*?)## Anything UNCLEAR"
return context, pattern
@parse
def parse_ui_elements(self, context: str):
"""Parse Selected Elements from the context using regex."""
pattern = r"## Selected Elements.*?\n(.*?)## HTML Layout"
return context, pattern
@parse
def parse_css_code(self, context: str):
pattern = r"```css.*?\n(.*?)## Anything UNCLEAR"
return context, pattern
@parse
def parse_html_code(self, context: str):
pattern = r"```html.*?\n(.*?)```"
return context, pattern
async def draw_icons(self, context, *args, **kwargs):
"""Draw icons using SDEngine."""
engine = SDEngine()
icon_prompts = self.parse_ui_elements(context)
icons = icon_prompts.split("\n")
icons = [s for s in icons if len(s.strip()) > 0]
prompts_batch = []
for icon_prompt in icons:
# fixme: 添加icon lora
prompt = engine.construct_payload(icon_prompt + ".<lora:WZ0710_AW81e-3_30e3b128d64T32_goon0.5>")
prompts_batch.append(prompt)
await engine.run_t2i(prompts_batch)
logger.info("Finish icon design using StableDiffusion API")
async def _save(self, css_content, html_content):
save_dir = WORKSPACE_ROOT / "resources" / 'codes'
if not os.path.exists(save_dir):
os.makedirs(save_dir, exist_ok=True)
# Save CSS and HTML content to files
css_file_path = save_dir / f"ui_design.css"
html_file_path = save_dir / f"ui_design.html"
with open(css_file_path, 'w') as css_file:
css_file.write(css_content)
with open(html_file_path, 'w') as html_file:
html_file.write(html_content)
async def run(self, requirements: list[Message], *args, **kwargs) -> ActionOutput:
"""Run the UI Design action."""
# fixme: update prompt (根据需求细化prompt）
context = requirements[-1].content
ui_design_draft = self.parse_requirement(context=context)
# todo: parse requirements str
prompt = PROMPT_TEMPLATE.format(context=ui_design_draft, format_example=FORMAT_EXAMPLE)
logger.info(prompt)
ui_describe = await self._aask_v1(prompt, "ui_design", OUTPUT_MAPPING)
logger.info(ui_describe.content)
logger.info(ui_describe.instruct_content)
css = self.parse_css_code(context=ui_describe.content)
html = self.parse_html_code(context=ui_describe.content)
await self._save(css_content=css, html_content=html)
await self.draw_icons(ui_describe.content)
return ui_describe
class UI(Role):
"""Class representing the UI Role."""
def __init__(self, name="Catherine", profile="UI Design",
goal="Finish a workable and good User Interface design based on a product design",
constraints="Give clear layout description and use standard icons to finish the design",
skills=["SD"]):
super().__init__(name, profile, goal, constraints)
self.load_skills(skills)
self._init_actions([UIDesign])
self._watch([WritePRD])
@load_engine
def load_sd_engine(self):
"""Load the SDEngine."""
file_name = ".tools.sd_engine"
engine_name = "SDEngine"
return file_name, engine_name
def load_skills(self, skills):
"""Load skills for the UI Role."""
# todo: 添加其他出图engine
for skill in skills:
if skill == "SD":
self.sd_engine = self.load_sd_engine()
logger.info(f"load skill engine {self.sd_engine}")

6
tests/metagpt/test_environment.py
View file

@ -8,12 +8,12 @@
import pytest
from metagpt.actions import BossRequirement
from metagpt.environment import Environment
from metagpt.logs import logger
from metagpt.manager import Manager
from metagpt.environment import Environment
from metagpt.roles import ProductManager, Architect, Role
from metagpt.roles import Architect, ProductManager, Role
from metagpt.schema import Message
from metagpt.actions import BossRequirement
@pytest.fixture

3
tests/metagpt/test_gpt.py
View file

@ -7,6 +7,7 @@
"""
import pytest
from metagpt.logs import logger
@ -36,7 +37,7 @@ class TestGPT:
@pytest.mark.asyncio
async def test_llm_api_costs(self, llm_api):
answer = await llm_api.aask('hello chatgpt')
await llm_api.aask('hello chatgpt')
costs = llm_api.get_costs()
logger.info(costs)
assert costs.total_cost > 0

4
tests/metagpt/test_llm.py
View file

@ -7,6 +7,7 @@
"""
import pytest
from metagpt.llm import LLM
@ -26,8 +27,7 @@ async def test_llm_aask_batch(llm):
@pytest.mark.asyncio
async def test_llm_aask(llm):
async def test_llm_acompletion(llm):
hello_msg = [{'role': 'user', 'content': 'hello'}]
assert len(await llm.acompletion(hello_msg)) > 0
assert len(await llm.acompletion_batch([hello_msg])) > 0

2
tests/metagpt/test_message.py
View file

@ -7,7 +7,7 @@
"""
import pytest
from metagpt.schema import Message, UserMessage, SystemMessage, AIMessage, RawMessage
from metagpt.schema import AIMessage, Message, RawMessage, SystemMessage, UserMessage
def test_message():

4
tests/metagpt/test_schema.py
View file

@ -5,7 +5,7 @@
@Author : alexanderwu
@File : test_schema.py
"""
from metagpt.schema import UserMessage, SystemMessage, AIMessage, Message
from metagpt.schema import AIMessage, Message, SystemMessage, UserMessage
def test_messages():
@ -18,4 +18,4 @@ def test_messages():
]
text = str(msgs)
roles = ['user', 'system', 'assistant', 'QA']
assert all([i in text for i in roles])
assert all([i in text for i in roles])

1
tests/metagpt/test_software_company.py
View file

@ -6,6 +6,7 @@
@File : test_software_company.py
"""
import pytest
from metagpt.logs import logger
from metagpt.software_company import SoftwareCompany

8
tests/metagpt/tools/test_prompt_generator.py
View file

@ -7,8 +7,14 @@
"""
import pytest
from metagpt.tools.prompt_writer import GPTPromptGenerator, EnronTemplate, BEAGECTemplate, WikiHowTemplate
from metagpt.logs import logger
from metagpt.tools.prompt_writer import (
BEAGECTemplate,
EnronTemplate,
GPTPromptGenerator,
WikiHowTemplate,
)
@pytest.mark.usefixtures("llm_api")

25
tests/metagpt/tools/test_sd_tool.py Normal file
View file

@ -0,0 +1,25 @@
# -*- coding: utf-8 -*-
# @Date : 2023/7/22 02:40
# @Author : stellahong (stellahong@fuzhi.ai)
#
import os
from metagpt.tools.sd_engine import SDEngine, WORKSPACE_ROOT
def test_sd_engine_init():
sd_engine = SDEngine()
assert sd_engine.payload["seed"] == -1
def test_sd_engine_generate_prompt():
sd_engine = SDEngine()
sd_engine.construct_payload(prompt="test")
assert sd_engine.payload["prompt"] == "test"
async def test_sd_engine_run_t2i():
sd_engine = SDEngine()
await sd_engine.run_t2i(prompts=["test"])
img_path = WORKSPACE_ROOT / "resources" / "SD_Output" / "output_0.png"
assert os.path.exists(img_path) == True

1
tests/metagpt/tools/test_search_engine.py
View file

@ -7,6 +7,7 @@
"""
import pytest
from metagpt.logs import logger
from metagpt.tools.search_engine import SearchEngine

8
tests/metagpt/tools/test_search_engine_meilisearch.py
View file

@ -5,11 +5,13 @@
@Author : alexanderwu
@File : test_search_engine_meilisearch.py
"""
import time
import pytest
import subprocess
import time
import pytest
from metagpt.logs import logger
from metagpt.tools.search_engine_meilisearch import MeilisearchEngine, DataSource
from metagpt.tools.search_engine_meilisearch import DataSource, MeilisearchEngine
MASTER_KEY = '116Qavl2qpCYNEJNv5-e0RC9kncev1nr1gt7ybEGVLk'

3
tests/metagpt/tools/test_summarize.py
View file

@ -7,9 +7,6 @@
"""
import pytest
from metagpt.logs import logger
from metagpt.tools.search_engine import SearchEngine
CASES = [
"""# 上下文

1
tests/metagpt/tools/test_translate.py
View file

@ -7,6 +7,7 @@
"""
import pytest
from metagpt.logs import logger
from metagpt.tools.translator import Translator

5
tests/metagpt/tools/test_ut_generator.py
View file

@ -6,9 +6,8 @@
@File : test_ut_generator.py
"""
from metagpt.tools.ut_writer import UTGenerator
from metagpt.const import SWAGGER_PATH, UT_PY_PATH, API_QUESTIONS_PATH
from metagpt.tools.ut_writer import YFT_PROMPT_PREFIX
from metagpt.const import API_QUESTIONS_PATH, SWAGGER_PATH, UT_PY_PATH
from metagpt.tools.ut_writer import YFT_PROMPT_PREFIX, UTGenerator
class TestUTWriter:

1
tests/metagpt/utils/test_code_parser.py
View file

@ -7,6 +7,7 @@
"""
import pytest
from metagpt.utils.common import CodeParser
t_text = '''

4
tests/metagpt/utils/test_common.py
View file

@ -6,8 +6,10 @@
@File : test_common.py
"""
import pytest
import os
import pytest
from metagpt.const import get_project_root

7
tests/metagpt/utils/test_custom_aio_session.py
View file

@ -5,17 +5,12 @@
@Author : alexanderwu
@File : test_custom_aio_session.py
"""
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import pytest
from metagpt.logs import logger
from metagpt.provider.openai_api import OpenAIGPTAPI
from metagpt.utils.custom_aio_session import CustomAioSession
async def try_hello(api):
batch = [[{'role': 'user', 'content': 'hello'}],]
batch = [[{'role': 'user', 'content': 'hello'}]]
results = await api.acompletion_batch_text(batch)
return results

5
tests/metagpt/utils/test_output_parser.py
View file

@ -5,11 +5,10 @@
@Author : chengmaoyu
@File : test_output_parser.py
"""
from typing import List, Tuple
import pytest
from typing import List, Tuple
import re
import ast
from metagpt.utils.common import OutputParser

1
tests/metagpt/utils/test_read_docx.py
View file

@ -6,7 +6,6 @@
@File : test_read_docx.py
"""
import pytest
from metagpt.const import PROJECT_ROOT
from metagpt.utils.read_document import read_docx

2
tests/metagpt/utils/test_token_counter.py
View file

@ -66,4 +66,4 @@ def test_count_string_tokens_gpt_4():
"""Test that the string tokens are counted correctly."""
string = "Hello, world!"
assert count_string_tokens(string, model_name="gpt-4-0314") == 4
assert count_string_tokens(string, model_name="gpt-4-0314") == 4