让AI给你写代码（四）—— 初步利用LangChain Agent根据输入生成，保存，执行

最新推荐文章于 2025-11-07 15:38:57 发布

原创

最新推荐文章于 2025-11-07 15:38:57 发布 · 1.3k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#langchain #agent #python

要进一步提升智能编码助手的效率，我觉得需要做到两点
1）进一步让主人聚焦于设计输入以及结果验证的循环
2）进一步让智能编码助手聚焦于代码实现和程序流程（保存、打开，修订、执行、合并…）
正好接触到LLM的LangChain的框架，那么初步体验一把利用其Agent实现代码生成，保存与执行
LangChain的中文官网

参考借鉴链接：阿里通义千问结合Langchain基于程序运行结果回答问题
链接中有一篇介绍，要求AI计算给定阶数（文中是10阶）斐波那契数列，代码生成和执行已经有了，所以我的诉求，看上去挺简单，增加一个保存有效代码的功能就OK了
初步改造如下，新增存盘的函数和相关调用：

#**新增存盘函数**
def saveFile(replyMessages: str):
    print('... to save the file ...')
    
    python_path="/home/cfets/eclipse-workspace/TestAI/testchain/"
    
    now_time = time.strftime('%Y-%m-%d_%H:%M:%S', time.localtime())
    i = random.randint(1, 100)
    code_file = python_path + "pyAITest_" + now_time + '_' + str(i) + ".py"
    # 保存至文件
    res_content=False
    
    try:
        with open(code_file, 'w') as f:
            f.write(replyMessages)
            res_content =True
    except Exception as e:
        print('Error: %s' % e)
        
    return res_content

@tool
def py_repl_tool(code: str):
    """Returns the result of execution."""

    _, after = code.split("```py")
    realcode = code.split("```")[0]
     
    # **新增保存文件**
    if saveFile(realcode)