cc/ds教学，计算机小白笔记（2.2）

代码是AI写的，用的CC接入DS-v4-pro，代码在前文基础上更新，功能包括：Agent框架，文件工具，联网搜索

本篇文章仅作为个人学习笔记记录

文件工具

完整代码如下：

""" 文件工具 Agent —— 让 AI 能读写你的文件 ======================================== 基于 04-agent-framework.py 的框架，新增两个文件工具。 展示框架的扩展能力：想加工具？写函数 → add_tool → 搞定。 """ import sys import os sys.stdout.reconfigure(encoding="utf-8") # 直接复用框架！ from importlib.util import spec_from_file_location, module_from_spec spec = spec_from_file_location("framework", os.path.join(os.path.dirname(__file__), "04-agent-framework.py")) framework = module_from_spec(spec) spec.loader.exec_module(framework) Agent = framework.Agent TOOL_REGISTRY = framework.TOOL_REGISTRY.copy() # 继承基础工具 # ============================================================ # 新增工具1：读文件 # ============================================================ def read_file(path: str) -> str: """读取指定路径的文件内容""" if not os.path.exists(path): return f"文件不存在：{path}" try: with open(path, "r", encoding="utf-8") as f: content = f.read() # 限制返回长度，避免 token 爆炸 if len(content) > 3000: content = content[:3000] + "\n...(内容已截断)" return content except Exception as e: return f"读取失败：{e}" # ============================================================ # 新增工具2：写文件 # ============================================================ def write_file(path: str, content: str) -> str: """写入内容到文件。会自动创建不存在的目录。""" try: dir_path = os.path.dirname(path) if dir_path and not os.path.exists(dir_path): os.makedirs(dir_path) with open(path, "w", encoding="utf-8") as f: f.write(content) return f"文件已写入：{path}（{len(content)} 个字符）" except Exception as e: return f"写入失败：{e}" # ============================================================ # 注册新工具 # ============================================================ TOOL_REGISTRY.update({ "read_file": { "func": read_file, "description": "读取电脑上的文件内容。当需要查看某个文件时使用。", "parameters": { "type": "object", "properties": { "path": {"type": "string", "description": "文件的完整路径"}, }, "required": ["path"], }, }, "write_file": { "func": write_file, "description": "将内容写入文件。用户要求保存、记录、导出时使用。", "parameters": { "type": "object", "properties": { "path": {"type": "string", "description": "要写入的文件路径"}, "content": {"type": "string", "description": "要写入的内容"}, }, "required": ["path", "content"], }, }, }) # ============================================================ # 启动 # ============================================================ if __name__ == "__main__": # 指定一个工作目录给 Agent，避免它乱翻 work_dir = os.path.join(os.path.dirname(__file__), "agent_workspace") os.makedirs(work_dir, exist_ok=True) agent = Agent(system_prompt=f"""你是一个能用工具的 AI 助手，可以用中文回答问题。 你可以： - 运行 Python 代码做计算 - 查看当前时间 - 读写电脑上的文件 文件操作的默认目录是：{work_dir} 用户说"记下来"时，你就用 write_file 保存到那个目录。 用户说"看一下"时，你就用 read_file 读取。""") for name, info in TOOL_REGISTRY.items(): agent.add_tool(name, info["func"], info["description"], info["parameters"]) print("=" * 50) print("文件 Agent —— 我能帮你读写文件了！") print(f"工作目录：{work_dir}") print("试试：") print(" '帮我写一首诗保存到 poem.txt'") print(" '帮我记一下：明天下午3点开会'") print(" '看看 poem.txt 里写了什么'") print("输入 'quit' 退出，'clear' 清空记忆") print("=" * 50) while True: user_input = input("\n你: ") if user_input.lower() == "quit": print("再见！") break if user_input.lower() == "clear": agent.clear_memory() print("(记忆已清空)") continue reply = agent.chat(user_input) print(f"AI: {reply}")

一、动态加载

使用普通import复用其他.py文件只需要直接import文件名（不带.py后缀）即可

1.从importlib.util导入两个辅助函数，spec_from_file_location创建一个“模块规范”ModuleSpec，描述了模块的元数据。module_from_spec根据模块规范创建一个空的模块对象。

# 直接复用框架！ from importlib.util import spec_from_file_location, module_from_spec

2.spec = spec_from_file_location("framework", os.path.join(os.path.dirname(__file__), "04-agent-framework.py"))

spec_from_file_location的第一个参数"framework"是给这个模块起的名字（可以任意，不一定和文件名相同）。第二个参数是文件的完整路径，通过os.path.dirname(__file__)获取当前脚本所在的目录（__file__：当前脚本文件的路径），然后拼接上"04-agent-framework.py"。这个函数返回一个ModuleSpec对象，里面记录了“如何加载这个文件作为一个模块”。

framework = module_from_spec(spec)根据上面创建的规范，创建一个空的模块对象

spec.loader是模块对应的加载器，.exec_module(framework)是真正执行04-agent-framework.py文件中的代码，并把执行结果填充到framework这个模块对象中

spec = spec_from_file_location("framework", os.path.join(os.path.dirname(__file__), "04-agent-framework.py")) framework = module_from_spec(spec) spec.loader.exec_module(framework)

3.调用上面代码导入的framework模块，framework.Agent获取框架中的类 ，.copy()方法复制字典，避免修改原注册表

Agent = framework.Agent TOOL_REGISTRY = framework.TOOL_REGISTRY.copy() # 继承基础工具

二、文件读取

def read_file(path: str) -> str: """读取指定路径的文件内容""" if not os.path.exists(path): # 1. 检查文件是否存在 return f"文件不存在：{path}" try: with open(path, "r", encoding="utf-8") as f: # 2. 打开文件 content = f.read() # 3. 读取内容 # 4. 限制返回长度，避免 token 爆炸 if len(content) > 3000: content = content[:3000] + "\n...(内容已截断)" return content except Exception as e: # 5. 异常处理 return f"读取失败：{e}"

1.with open()是上下文管理器，自动关闭文件

as f将打开的文件对象赋值给变量f

with 表达式 [as 变量]:调用该对象的_enter_()方法，其返回值（如果有）赋值给as后面的变量。无论块内是否发生异常，都会在退出时调用上下文管理器的_exit_()方法，执行清理工作。如果_exit_返回True，则异常被压制（不再向外抛出）；否则异常会继续传播。

def read_file(path: str) -> str: """读取指定路径的文件内容""" if not os.path.exists(path): # 1. 检查文件是否存在 return f"文件不存在：{path}" try: with open(path, "r", encoding="utf-8") as f: # 2. 打开文件 content = f.read() # 3. 读取内容 # 4. 限制返回长度，避免 token 爆炸 if len(content) > 3000: content = content[:3000] + "\n...(内容已截断)" return content except Exception as e: # 5. 异常处理 return f"读取失败：{e}"

2.f.read()一次性读取文件的全部内容，返回一个字符串。

content = f.read()

三、文件写入工具

1.os.makedirs()递归创建多级目录，os.makedirs(path, exist_ok=True)递归创建所有缺失目录同时在路径已存在的时候不会报错

def write_file(path: str, content: str) -> str: """写入内容到文件。会自动创建不存在的目录。""" try: dir_path = os.path.dirname(path) # 1. 获取目录路径 # 2. 如果目录不存在则创建 if dir_path and not os.path.exists(dir_path): os.makedirs(dir_path) # 递归创建目录 # 3. 写入文件 with open(path, "w", encoding="utf-8") as f: f.write(content) # 写入内容 return f"文件已写入：{path}（{len(content)} 个字符）" except Exception as e: return f"写入失败：{e}"

2.if dir_path and not os.path.exists(dir_path):中dir_path首先判断你给出的路径名称是不是空的，然后and not os.path.exists(dir_path):判断路径是不是已经存在了

if dir_path and not os.path.exists(dir_path):

四、工具注册

TOOL_REGISTRY.update({ "read_file": { "func": read_file, "description": "读取电脑上的文件内容。当需要查看某个文件时使用。", "parameters": { "type": "object", "properties": { "path": { "type": "string", "description": "文件的完整路径" }, }, "required": ["path"], # 必需参数 }, }, "write_file": {...} # 类似结构 })

五、主程序配置

1.创建工作目录

work_dir = os.path.join(os.path.dirname(__file__), "agent_workspace") os.makedirs(work_dir, exist_ok=True)

2.name和info是键值对，info也是字典，这个循环可以使主程序自动加载工具

for name, info in TOOL_REGISTRY.items(): agent.add_tool(name, info["func"], info["description"], info["parameters"])