Trae生成“测试用例自动生成” 的应用实例

背景：使用生成一个通过ai输入需求生成用例的应用，要基于python代码编写trae 生成一个自动化用例编写的应用，内置封装好的AI模型

目的：熟悉下AI工具的应用范围，了解AI知识的冰山一角，对AI编程有个大致的认识

执行步骤：

1、安装trae，编辑栏输入“生成一个通过ai输入需求生成用例的应用，要基于python代码编写”

2、生成应用的文件目录

3、替换.py文件里相关的ai的key和url

封装好的ai模型数据：

def analyze_code():

data = request.json

prompt_type = data.get('prompt_type')

code = data.get('code')

if not code:

return jsonify({"error": "请输入代码内容"}), 400

prompt = f"{prompt_type}:\n{code}"

print(prompt)

#获取输入内容

————————————

#通过调接口实现ai模型调用

headers = {

'Authorization': 'Bearer e***,  #aikey

'Content-Type': 'application/json'

}

data = {

"model": "volcengine/DeepSeek-R1", #模型类型

"messages": [

{

"role": "user",

"content": prompt   #输入的内容

}

],

"max_tokens": 500,

"temperature": 0.2,

"chat_template_kwargs": {

"enable_thinking": False

}

}

response = requests.request("POST", "http://url", headers=headers, json=data)

date = response.text

4、调试，执行成功