code2prompt入门指南 - 将代码库转换为LLM提示的强大CLI工具

Ray

code2prompt:将代码库转换为LLM提示的利器

code2prompt是一个强大的命令行工具,可以将整个代码库转换为单个LLM(大型语言模型)提示。它可以生成包含源代码树结构和所有代码的格式良好的Markdown提示,并提供提示模板和token计数等功能。本文将全面介绍code2prompt的主要特性和使用方法,帮助开发者快速掌握这个提高AI辅助开发效率的实用工具。

主要特性

code2prompt具有以下主要功能:

快速从任何规模的代码库生成LLM提示
使用Handlebars模板自定义提示生成
遵守.gitignore规则
使用glob模式过滤和排除文件
显示生成提示的token数量
可选择在生成的提示中包含Git diff输出(暂存文件)
自动将生成的提示复制到剪贴板
将生成的提示保存到输出文件
按名称或路径排除文件和文件夹
为源代码块添加行号

安装方法

有以下几种安装code2prompt的方法:

从Releases页面下载最新的二进制文件
使用cargo安装:
```
cargo install code2prompt
```

从源代码构建:

git clone https://github.com/mufeedvh/code2prompt.git
cd code2prompt/
cargo build --release

使用示例

以下是code2prompt的一些常见用法:

从代码库目录生成提示:
```
code2prompt path/to/codebase
```

使用自定义Handlebars模板文件:

code2prompt path/to/codebase -t path/to/template.hbs

使用glob模式过滤文件:

code2prompt path/to/codebase --include="*.rs,*.toml"

排除特定文件:

code2prompt path/to/codebase --exclude="*.txt,*.md"

显示生成提示的token数量:
```
code2prompt path/to/codebase --tokens
```

将生成的提示保存到输出文件:

code2prompt path/to/codebase --output=output.txt

内置模板

code2prompt提供了一些内置模板用于常见用例:

document-the-code.hbs:生成用于记录代码的提示
find-security-vulnerabilities.hbs:生成用于查找潜在安全漏洞的提示
clean-up-code.hbs:生成用于清理和改进代码质量的提示
fix-bugs.hbs:生成用于修复代码中的bug的提示
write-github-pull-request.hbs:创建GitHub拉取请求描述
write-github-readme.hbs:生成高质量的README文件
write-git-commit.hbs:生成Git提交消息
improve-performance.hbs:生成用于改进代码性能的提示

用户定义变量

code2prompt支持在Handlebars模板中使用用户定义的变量。在提示生成过程中,用户将被要求输入这些变量的值,从而实现基于用户输入的进一步自定义。

总结

code2prompt是一个功能强大且易于使用的工具,可以大大简化从代码库生成LLM提示的过程。它为AI辅助开发提供了便利,有助于提高代码分析、生成和其他任务的效率。无论是个人开发者还是大型团队,都可以从code2prompt中受益,提升开发流程的智能化水平。

希望本文能帮助您快速上手code2prompt,充分发挥这个实用工具的潜力。如果您觉得code2prompt有用,不妨给项目一个star,支持作者继续改进这个工具。

code2prompt screenshot