# mentat

The AI Coding Assistant

https://github.com/AbanteAI/mentat

## Usage

python 3.10


准备一个 openai gpt4 key，新建 .env文件，输入：

```
OPENAI_API_KEY=sk-rju8J2X

```

自己的项目文件夹路径，如 /workspaces/mentat 执行：

```
python3 -m venv .venv
source .venv/bin/activat

git clone https://github.com/AbanteAI/mentat.git
cd mentat
pip install -e .

mentat path/to/directory --exclude exclude_me.py dir1/dir2 **/*.ts
mentat /workspaces/mentat/setup.py
```
完了之后就会把自己的代码post到openai，用户命令行输入命令，调用openai执行相应的修改代码命令。

## 源码分析

输入命令 `mentat /workspaces/mentat/setup.py` 会执行如下命令：

```
mentat=mentat.app:run_cli

# 带参数运行
    run(
        expand_paths(paths),
        expand_paths(exclude_paths),
        no_code_map,
        diff,
        pr_diff,
    )
    
    
```

第一个参数为文件路径，或者文件夹。后面参数为排除路径等。

```
        setup_api_key()
        loop(paths, exclude_paths, cost_tracker, no_code_map, diff, pr_diff)
```

先设置 key，然后循环：
```
    while True:
        if need_user_request:
            user_response = user_input_manager.collect_user_input()
            conv.add_user_message(user_response)

        file_edits = conv.get_model_response(parser, config)
        file_edits = [
            file_edit
            for file_edit in file_edits
            if file_edit.is_valid(code_file_manager, config)
        ]
        if file_edits:
            need_user_request = get_user_feedback_on_edits(
                config,
                conv,
                code_context,
                user_input_manager,
                code_file_manager,
                file_edits,
            )
        else:
            need_user_request = True
```
这里循环等待用户输入，将用户输入 `conv.add_user_message(user_response)` 加入conv对象。那么继续深入这个 Conversation 类，这个是调用 openai 接口。



**提取词：**

和其他项目类似，这个项目也设置了自己的提取词：目录在 mentat/prompts ，通过这些提取词，实现自定义的命令。