在AI辅助写作中插入代码，可以通过以下几种方式实现：

1. 直接插入：将代码直接复制粘贴到文档中。为了确保代码的可读性，可以使用等宽字体（如Courier New）来显示代码。

2. 使用代码块：如果你是在支持Markdown语法的编辑器中写作，可以使用代码块来插入代码。代码块通常通过三个反引号（```)开始和结束，可以在反引号后面指定编程语言，以启用语法高亮。例如：
```
```python
def hello_world():
print("Hello, world!")
```
```

3. 图片形式：将代码截图或转换为图片格式，然后将图片插入文档中。这种方式适合于少量代码或需要特别格式化的代码片段。

4. 链接引用：如果代码量较大，可以将代码上传到GitHub、GitLab等代码托管平台，然后在文档中提供链接，引导读者查看。

选择哪种方式取决于你的具体需求和文档的最终用途。