基于 Python 开发可调用工具的 AI Agent 实现电脑配置评价
AI Agent 凭借强大的任务处理能力,成为开发人员手中的得力工具。本文演示如何使用 Python 打造能够调用工具的 AI Agent,实现根据电脑信息对配置进行专业评价的功能。
一、项目创建与目录结构
1.1 项目创建
使用 UV 进行项目初始化。
uv init demo
进入项目文件夹并安装必要的包,例如安装 psutil。
uv add psutil
安装的包会记录在 pyproject.toml 中:
[project]
name = "demo"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
"psutil>=7.0.0",
"pydantic-ai>=0.2.16",
"python-dotenv>=1.1.0",
"setuptools>=80.9.0",
]
1.2 目录结构
项目目录结构如下:
demo
├── .venv
├── .env
├── .gitignore
├── .python-version
├── main.py
├── pyproject.toml
├── README.md
├── tools.py
└── uv.lock
主要创建了以下文件:
main.py:项目主程序文件,负责 AI Agent 的运行逻辑,包括模型初始化、交互及结果处理。tools.py:存放读取电脑配置信息的工具函数。.env:存储敏感信息(如OPENROUTER_API_KEY),保证 API 密钥安全。
二、功能实现与代码解析
2.1 读取电脑配置信息(tools.py)
定义 read_pc_info 函数,用于获取电脑的详细配置信息,依赖 platform 和 psutil 库。
import platform
import psutil
def read_pc_info():
info = []
# CPU 信息
info.append()
info.append()
mem = psutil.virtual_memory()
info.append()
partitions = psutil.disk_partitions()
p partitions:
p.fstype:
usage = psutil.disk_usage(p.mountpoint)
info.append()
info.append()
.join(info)


