# Copilot 规则
1. 所有Python代码必须符合PEP8规范，函数名用蛇形命名，类名用大驼峰；
2. 所有接口必须添加参数校验，返回格式统一为{"code": 0, "msg": "success", "data": {}}；
3. 所有数据库操作必须用SQLAlchemy ORM，禁止拼接SQL字符串；
4. 所有的外部调用必须添加超时时间和重试机制，超时时间默认10秒，重试3次；
5. 代码必须添加必要的注释，复杂逻辑要加注释说明。

渲染错误: Mermaid 渲染失败: Parsing failed: Lexer error on line 2, column 21: unexpected character: ->[<- at offset: 38, skipped 5 characters. Lexer error on line 3, column 24: unexpected character: ->[<- at offset: 67, skipped 6 characters. Lexer error on line 4, column 20: unexpected character: ->[<- at offset: 101, skipped 5 characters. Lexer error on line 6, column 18: unexpected character: ->[<- at offset: 137, skipped 5 characters. Lexer error on line 7, column 20: unexpected character: ->[<- at offset: 162, skipped 1 characters. Lexer error on line 7, column 24: unexpected character: ->网<- at offset: 166, skipped 3 characters. Lexer error on line 8, column 21: unexpected character: ->[<- at offset: 201, skipped 6 characters. Lexer error on line 10, column 16: unexpected character: ->[<- at offset: 239, skipped 1 characters. Lexer error on line 10, column 22: unexpected character: ->层<- at offset: 245, skipped 2 characters. Lexer error on line 11, column 34: unexpected character: ->[<- at offset: 281, skipped 5 characters. Lexer error on line 11, column 44: unexpected character: ->]<- at offset: 291, skipped 1 characters. Lexer error on line 12, column 27: unexpected character: ->[<- at offset: 328, skipped 5 characters. Lexer error on line 12, column 37: unexpected character: ->]<- at offset: 338, skipped 1 characters. Lexer error on line 13, column 27: unexpected character: ->[<- at offset: 375, skipped 3 characters. Lexer error on line 13, column 35: unexpected character: ->]<- at offset: 383, skipped 1 characters. Lexer error on line 14, column 29: unexpected character: ->[<- at offset: 422, skipped 3 characters. Lexer error on line 14, column 37: unexpected character: ->]<- at offset: 430, skipped 1 characters. Lexer error on line 16, column 20: unexpected character: ->[<- at offset: 465, skipped 6 characters. Lexer error on line 17, column 20: unexpected character: ->[<- at offset: 491, skipped 1 characters. Lexer error on line 17, column 24: unexpected character: ->检<- at offset: 495, skipped 5 characters. Lexer error on line 18, column 24: unexpected character: ->[<- at offset: 537, skipped 7 characters. Lexer error on line 19, column 27: unexpected character: ->[<- at offset: 584, skipped 7 characters. Lexer error on line 20, column 23: unexpected character: ->[<- at offset: 627, skipped 1 characters. Lexer error on line 20, column 27: unexpected character: ->文<- at offset: 631, skipped 4 characters. Lexer error on line 22, column 15: unexpected character: ->[<- at offset: 668, skipped 5 characters. Lexer error on line 23, column 25: unexpected character: ->[<- at offset: 698, skipped 1 characters. Lexer error on line 23, column 29: unexpected character: ->工<- at offset: 702, skipped 3 characters. Lexer error on line 24, column 26: unexpected character: ->[<- at offset: 739, skipped 6 characters. Lexer error on line 25, column 24: unexpected character: ->[<- at offset: 777, skipped 1 characters. Lexer error on line 25, column 27: unexpected character: ->/<- at offset: 780, skipped 1 characters. Lexer error on line 25, column 30: unexpected character: ->工<- at offset: 783, skipped 3 characters. Lexer error on line 26, column 26: unexpected character: ->[<- at offset: 820, skipped 8 characters. Parse error on line 7, column 21: Expecting: one of these possible Token sequences: 1. [NEWLINE] 2. [EOF] but found: 'API' Parse error on line 7, column 28: Expecting token of type ':' but found `in`. Parse error on line 10, column 17: Expecting: one of these possible Token sequences: 1. [NEWLINE] 2. [EOF] but found: 'Agent' Parse error on line 10, column 24: Expecting token of type ':' but found ` `. Parse error on line 11, column 39: Expecting: one of these possible Token sequences: 1. [NEWLINE] 2. [EOF] but found: 'Agent' Parse error on line 11, column 46: Expecting token of type ':' but found `in`. Parse error on line 12, column 32: Expecting: one of these possible Token sequences: 1. [NEWLINE] 2. [EOF] but found: 'Agent' Parse error on line 12, column 39: Expecting token of type ':' but found `in`. Parse error on line 13, column 30: Expecting: one of these possible Token sequences: 1. [NEWLINE] 2. [EOF] but found: 'Agent' Parse error on line 13, column 37: Expecting token of type ':' but found `in`. Parse error on line 14, column 32: Expecting: one of these possible Token sequences: 1. [NEWLINE] 2. [EOF] but found: 'Agent' Parse error on line 14, column 39: Expecting token of type ':' but found `in`. Parse error on line 17, column 21: Expecting: one of these possible Token sequences: 1. [NEWLINE] 2. [EOF] but found: 'R' Parse error on line 17, column 30: Expecting token of type ':' but found `in`. Parse error on line 20, column 24: Expecting: one of these possible Token sequences: 1. [NEWLINE] 2. [EOF] but found: 'API' Parse error on line 20, column 32: Expecting token of type ':' but found `in`. Parse error on line 23, column 26: Expecting: one of these possible Token sequences: 1. [NEWLINE] 2. [EOF] but found: 'Git' Parse error on line 23, column 33: Expecting token of type ':' but found `in`. Parse error on line 25, column 25: Expecting: one of these possible Token sequences: 1. [NEWLINE] 2. [EOF] but found: 'CI' Parse error on line 25, column 28: Expecting token of type ':' but found `CD`. Parse error on line 25, column 34: Expecting: one of these possible Token sequences: 1. [NEWLINE] 2. [EOF] but found: 'in' Parse error on line 25, column 41: Expecting token of type ':' but found ` `. Parse error on line 28, column 13: Expecting token of type ':' but found `--`. Parse error on line 28, column 17: Expecting token of type 'ARROW_DIRECTION' but found `api`. Parse error on line 29, column 9: Expecting token of type ':' but found `--`. Parse error on line 29, column 13: Expecting token of type 'ARROW_DIRECTION' but found `api`. Parse error on line 30, column 9: Expecting token of type ':' but found `--`. Parse error on line 30, column 13: Expecting token of type 'ARROW_DIRECTION' but found `auth`. Parse error on line 31, column 10: Expecting token of type ':' but found `--`. Parse error on line 31, column 14: Expecting token of type 'ARROW_DIRECTION' but found `requirement_agent`. Parse error on line 32, column 23: Expecting token of type ':' but found `--`. Parse error on line 32, column 27: Expecting token of type 'ARROW_DIRECTION' but found `rag`. Parse error on line 33, column 9: Expecting token of type ':' but found `--`. Parse error on line 33, column 13: Expecting token of type 'ARROW_DIRECTION' but found `spec_db`. Parse error on line 34, column 9: Expecting token of type ':' but found `--`. Parse error on line 34, column 13: Expecting token of type 'ARROW_DIRECTION' but found `project_db`. Parse error on line 35, column 9: Expecting token of type ':' but found `--`. Parse error on line 35, column 13: Expecting token of type 'ARROW_DIRECTION' but found `api_db`. Parse error on line 36, column 23: Expecting token of type ':' but found `--`. Parse error on line 36, column 27: Expecting token of type 'ARROW_DIRECTION' but found `code_agent`. Parse error on line 37, column 16: Expecting token of type ':' but found `--`. Parse error on line 37, column 20: Expecting token of type 'ARROW_DIRECTION' but found `git_tool`. Parse error on line 38, column 16: Expecting token of type ':' but found `--`. Parse error on line 38, column 20: Expecting token of type 'ARROW_DIRECTION' but found `test_agent`. Parse error on line 39, column 16: Expecting token of type ':' but found `--`. Parse error on line 39, column 20: Expecting token of type 'ARROW_DIRECTION' but found `test_tool`. Parse error on line 40, column 16: Expecting token of type ':' but found `--`. Parse error on line 40, column 20: Expecting token of type 'ARROW_DIRECTION' but found `scan_tool`. Parse error on line 41, column 16: Expecting token of type ':' but found `--`. Parse error on line 41, column 20: Expecting token of type 'ARROW_DIRECTION' but found `deploy_agent`. Parse error on line 42, column 18: Expecting token of type ':' but found `--`. Parse error on line 42, column 22: Expecting token of type 'ARROW_DIRECTION' but found `ci_tool`. Parse error on line 43, column 18: Expecting token of type ':' but found `--`. Parse error on line 43, column 22: Expecting token of type 'ARROW_DIRECTION' but found `dev`.

from langchain.agents import AgentExecutor, create_openai_tools_agent
from langchain_openai import ChatOpenAI
from langchain_core.tools import tool
from langchain_core.prompts import ChatPromptTemplate
import os
import subprocess
import git

# 1. 定义工具集
@tool
def read_file(file_path: str) -> str:
    """读取指定路径的文件内容，用于获取项目上下文"""
    if not os.path.exists(file_path):
        return f"错误：文件{file_path}不存在"
    try:
        with open(file_path, 'r', encoding='utf-8') as f:
            return f.read()
    except Exception as e:
        return f"读取文件失败：{str(e)}"

@tool
def write_file(file_path: str, content: str) -> str:
    """将内容写入指定路径的文件，用于保存生成的代码"""
    try:
        os.makedirs(os.path.dirname(file_path), exist_ok=True)
        with open(file_path, 'w', encoding='utf-8') as f:
            f.write(content)
        return f"成功写入文件：{file_path}"
    except Exception as e:
        return f"写入文件失败：{str(e)}"

@tool
def run_pytest(test_path: str) -> str:
    """运行指定路径的pytest单元测试，返回测试结果"""
    try:
        result = subprocess.run(
            f"pytest {test_path} -v",
            shell=True,
            capture_output=True,
            text=True,
            timeout=300
        )
        return f"测试ExitCode：{result.returncode}\n标准输出：{result.stdout}\n错误输出：{result.stderr}"
    except subprocess.TimeoutExpired:
        return "测试运行超时，请检查代码是否有死循环"

@tool
def git_commit(commit_message: str) -> str:
    """提交当前修改到Git仓库，参数为提交信息"""
    try:
        repo = git.Repo(os.getcwd())
        repo.git.add(all=True)
        repo.index.commit(commit_message)
        return f"代码提交成功，提交信息：{commit_message}"
    except Exception as e:
        return f"代码提交失败：{str(e)}"

# 2. 初始化LLM和提示词
llm = ChatOpenAI(model="gpt-4o", temperature=0, api_key=os.getenv("OPENAI_API_KEY"))
prompt = ChatPromptTemplate.from_messages([
    ("system", """你是团队的资深Python开发工程师，负责自主完成FastAPI项目的常规需求开发，请严格遵守以下规则：
    1. 先读取项目的README、现有配置文件和相关代码，理解项目结构和规范；
    2. 生成代码必须符合团队的PEP8规范、安全规范，所有数据库操作用SQLAlchemy ORM；
    3. 代码生成完成后必须编写单元测试，运行测试通过后再提交代码；
    4. 如果遇到无法解决的问题，直接返回问题说明，不要生成错误的代码。"""),
    ("user", "{input}"),
    ("agent_scratchpad", "{agent_scratchpad}")
])

# 3. 创建Agent
tools = [read_file, write_file, run_pytest, git_commit]
agent = create_openai_tools_agent(llm, tools, prompt)
agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True, max_iterations=20)

# 4. 运行Agent处理需求
if __name__ == "__main__":
    requirement = """给当前电商项目添加优惠券功能：
    1. 优惠券类型：满减券，满100减10，满200减30，满500减100；
    2. 规则：每个用户每种优惠券只能领1张，有效期30天，过期不能使用，不能叠加使用；
    3. 功能：优惠券领取接口、查询用户优惠券列表接口、下单时核销优惠券接口；
    4. 生成对应的模型、接口、单测，测试通过后提交代码，提交信息为"feat: add coupon module"。"""
    result = agent_executor.invoke({"input": requirement})
    print("开发完成，结果：", result["output"])