27、如何根据代码库代码生成新的代码

// 中间件：记录请求日志
func LoggerMiddleware() gin.HandlerFunc {
    return func(c *gin.Context) {
        start := time.Now()
        path := c.Request.URL.Path
        method := c.Request.Method

        c.Next()

        latency := time.Since(start)
        statusCode := c.Writer.Status()
        fmt.Printf("[%s] %s %d %v\n", method, path, statusCode, latency)
    }
}

// 中间件：跨域请求处理
func CorsMiddleware() gin.HandlerFunc {
    return func(c *gin.Context) {
        c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
        c.Writer.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
        c.Writer.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization")

        if c.Request.Method == "OPTIONS" {
            c.AbortWithStatus(204)
            return
        }

        c.Next()
    }
}

// 中间件：请求超时控制
func TimeoutMiddleware(timeout time.Duration) gin.HandlerFunc {
    return func(c *gin.Context) {
        ctx, cancel := context.WithTimeout(c.Request.Context(), timeout)
        defer cancel()

        c.Request = c.Request.WithContext(ctx)

        done := make(chan bool, 1)
        go func() {
            c.Next()
            done <- true
        }()

        select {
        case <-done:
            return
        case <-ctx.Done():
            c.AbortWithStatusJSON(http.StatusRequestTimeout, gin.H{
                "error": "Request timeout",
            })
        }
    }
}

import os
from langchain_community.document_loaders import TextLoader
from langchain_text_splitters import CharacterTextSplitter
from llm import QdrantVecStoreFromDocs

def load_code(ext:str,dir_path:str):
    if not os.path.exists(dir_path):
        print(f"文件夹{dir_path}不存在")
        return

    files=[]

    for file in os.listdir(dir_path):
        if file.endswith(ext):
            print(f"加载文件{file}")
            files.append(os.path.join(dir_path,file))

    all_docs=[]
    code_text_splitter = CharacterTextSplitter(separator="\n",chunk_size=500,chunk_overlap=100,length_function=len)

    for file in files:
        loader=TextLoader(file, encoding='utf-8').load()
        docs=code_text_splitter.split_documents(loader)
        for doc in docs:
            doc.metadata["source"]=file
            all_docs.append(doc)
    
    QdrantVecStoreFromDocs(all_docs,"code")

if __name__ == '__main__':
    load_code(".go","D:\\workspace\\python\\class23\\langgraph\\code05\\middleware")

SYSTEM
你是一个 go 语言编程专家，擅长根据问题以及代码库的代码进行代码生成。
使用上下文来生成代码。你只需输出golang代码，无需任何解释和说明。不要将代码放到 ```go ``` 中。

上下文：
{context}

HUMAN
问题：{question}

该工具可用于生成中间件函数，参数需传入具体的生成代码的需求，例如：跨域中间件

class State(TypedDict):
    main: str
    models: list[str]
    routes: list[str]
    handlers: list[str]
    middleware: list[str]

middleware_prompt = """
#中间件
创建用于跨域的中间件函数
"""

def middleware_node(state):
    message=llm.invoke([SystemMessage(content=systemMessage),HumanMessage(content=middleware_prompt)])
    for tool_call in message.tool_calls:
        tool_name = tool_call["name"]
        get_tool = tools_names[tool_name]
        result = get_tool.invoke(tool_call["args"])
        state["middleware"].append(result)
    return state

func main() {
        r := gin.Default()

        r.Use(CorsMiddleware())

        r.GET("/version", version_handler)
        r.GET("/users", users_handler)

        r.Run(":8080")
}

1.创建gin对象
2.中间件的代码为{middleware}，请提取其函数名称，并使用r.Use注册这些中间件
3.拥有路由代码
{routes}
handler代码已经生成，无需再进行处理
4.启动端口为8080

graph = sg.compile()
code = graph.invoke({"main":"", "routes":[], "handlers":[], "models":[], "middleware":[]})

print(code["models"][0])
print(code["middleware"][0])
print(code["main"])
for handler in code["handlers"]:
    print(handler)