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

update debug_code ut

Browse source 
update sd_tool_usage example
This commit is contained in:
stellahsr 2024-01-11 21:50:35 +08:00
parent a98edada1a
commit af26fe06cf
5 changed files with 68 additions and 65 deletions
Download patch file Download diff file Expand all files Collapse all files

41
examples/sd_tool_usage.py
View file

@ -3,38 +3,17 @@
# @Author : stellahong (stellahong@fuzhi.ai)
# @Desc :
import asyncio
from metagpt.const import METAGPT_ROOT
from metagpt.actions.write_analysis_code import WriteCodeWithTools
from metagpt.plan.planner import Planner
from metagpt.actions.execute_code import ExecutePyCode
from metagpt.roles.code_interpreter import CodeInterpreter
sd_url = 'http://106.75.10.65:19094/sdapi/v1/txt2img'
requirement = f"i have a text2image tool, generate a girl image use it, sd_url={sd_url}"
async def main(requirement: str = ""):
code_interpreter = CodeInterpreter(use_tools=True, goal=requirement)
await code_interpreter.run(requirement)
if __name__ == "__main__":
code_interpreter = CodeInterpreter(use_tools=True, goal=requirement)
asyncio.run(code_interpreter.run(requirement))
# planner = Planner(
# goal="i have a sdt2i tool, generate a girl image use it, sd_url='http://106.75.10.65:19094/sdapi/v1/txt2img'",
# auto_run=True)
# asyncio.run(planner.update_plan())
# schema_path = METAGPT_ROOT / "metagpt/tools/functions/schemas"
# #
# prompt = "1girl, beautiful"
# planner = Planner(
# goal="i have a sdt2i tool, generate a girl image use it, sd_url='http://106.75.10.65:19094/sdapi/v1/txt2img'",
# auto_run=True)
# asyncio.run(planner.update_plan())
# planner.plan.current_task.task_type = "sd"
# planner.plan.current_task.instruction = "Use the sdt2i tool with the provided API endpoint to generate the girl image."
# executor = ExecutePyCode()
#
# tool_context, code = asyncio.run(WriteCodeWithTools(schema_path=schema_path).run(
# context=f"task prompt: {prompt}",
# plan=planner.plan,
# column_info="",
# ))
# print(code)
# asyncio.run(executor.run(code))
sd_url = 'http://106.75.10.65:19094'
requirement = f"I want to generate an image of a beautiful girl using the stable diffusion text2image tool, sd_url={sd_url}"
asyncio.run(main(requirement))