TRAE advanced features and 自定义

# example: 自定义提示模板

# basic提示模板
{
  "name": "python_function_docstring",
  "description": "生成Pythonfunction 详细documentationstring",
  "template": " for 以 under Pythonfunction生成详细 documentationstring, includingfunctionsdescribes, parameter说明, return value说明 and example: \n\n{code}"
}

# usingexample
def calculate_area(radius):
    return 3.14159 * radius ** 2

# TRAE根据自定义模板生成 documentationstring
def calculate_area(radius):
    """计算圆 面积
    
    parameter:
        radius: 圆 半径, 必须 for 正数
    
    返回:
        圆 面积, 结果保留5位 small 数
    
    example:
        >>> calculate_area(5)
        78.53975
    """
    return 3.14159 * radius ** 2

# example:  on  under 文提示定制

#  on  under 文提示configuration
{
  "name": "web_development_context",
  "description": "WebDevelopment on  under 文提示",
  "context": {
    "framework": "React",
    "language": "JavaScript",
    "style_guide": "Airbnb",
    "preferred_libraries": ["axios", "react-router-dom", "styled-components"],
    "coding_conventions": {
      "component_naming": "PascalCase",
      "function_naming": "camelCase",
      "use_hooks": true,
      "use_typescript": false
    }
  }
}

# usingexample: 根据 on  under 文生成Reactcomponent
# 输入: "creation一个userinformationcomponent"
# TRAE根据 on  under 文提示生成 code: 

import React from 'react';
import styled from 'styled-components';

const UserCard = styled.div`
  background-color: #fff;
  border-radius: 8px;
  box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
  padding: 16px;
  max-width: 300px;
`;

const UserName = styled.h2`
  margin: 0 0 8px 0;
  font-size: 18px;
  color: #333;
`;

const UserEmail = styled.p`
  margin: 0;
  font-size: 14px;
  color: #666;
`;

const UserInfo = ({ user }) => {
  return (
    
      {user.name}
      {user.email}
    
  );
};

export default UserInfo;

# example: 微调TRAEmodel

# 1. 准备训练data
training_data = [
  {
    "input": "def calculate_area(radius):",
    "output": "    return 3.14159 * radius ** 2"
  },
  {
    "input": "def process_data(data):",
    "output": "    return [item for item in data if item['status'] == 'active']"
  },
  # 更 many 训练data...
]

# 2. 保存训练data to JSONfile
import json
with open('training_data.json', 'w') as f:
    json.dump(training_data, f)

# 3. run微调commands
# trae fine-tune --data training_data.json --model trae-base --output trae-custom

# 4. configurationTRAEusing微调 after  model
#  in .trae/config.jsonin添加: 
# {
#   "model": {
#     "name": "trae-custom",
#     "path": "./models/trae-custom"
#   }
# }

# example: Development一个 simple  TRAE插件

# plugin.py
import trae
from trae.plugin import Plugin

class HelloPlugin(Plugin):
    """ simple  Hello插件"""
    
    name = "hello"
    description = "一个 simple  Hello插件"
    version = "1.0.0"
    
    def __init__(self):
        super().__init__()
        self.register_command("hello", self.hello_command)
    
    def hello_command(self, args):
        """processinghellocommands"""
        if args:
            name = args[0]
            return f"Hello, {name}! Welcome to TRAE plugin development!"
        else:
            return "Hello! Welcome to TRAE plugin development!"
    
    def on_file_open(self, file_path):
        """file打开eventprocessing"""
        self.log(f"File opened: {file_path}")
    
    def on_code_change(self, file_path, changes):
        """code变更eventprocessing"""
        self.log(f"Code changed in {file_path}: {len(changes)} changes")

# register插件
trae.register_plugin(HelloPlugin)

# using插件
# 1. installation插件: trae plugin install ./plugin.py
# 2. run插件commands: trae plugin run hello --name Alice

# example: TRAEconfigurationfile

{
  "general": {
    "language": "zh-CN",
    "theme": "light",
    "auto_save": true
  },
  "completion": {
    "enable": true,
    "max_suggestions": 5,
    "show_documentation": true,
    "show_deprecated": false
  },
  "code_fix": {
    "enable": true,
    "auto_apply": false,
    "severity_threshold": "medium"
  },
  "documentation": {
    "enable": true,
    "format": "google",
    "include_examples": true
  },
  "model": {
    "name": "trae-base",
    "temperature": 0.7,
    "max_tokens": 1000
  },
  "plugins": {
    "enabled": ["hello", "code_quality"],
    "disabled": ["experimental"]
  }
}

# example: usingTRAE API

import requests

# APIconfiguration
API_URL = "http://localhost:8080/api/v1"
API_KEY = "your_api_key"

# code补全request
def get_code_completion(code, language="python"):
    headers = {
        "Authorization": f"Bearer {API_KEY}",
        "Content-Type": "application/json"
    }
    data = {
        "code": code,
        "language": language,
        "max_suggestions": 3
    }
    response = requests.post(f"{API_URL}/completion", headers=headers, json=data)
    if response.status_code == 200:
        return response.json()
    else:
        print(f"Error: {response.status_code} - {response.text}")
        return None

# usingexample
code = "def calculate_area(radius):\n    # TODO: Implement area calculation"
result = get_code_completion(code)
if result:
    for i, suggestion in enumerate(result["suggestions"]):
        print(f"Suggestion {i+1}:\n{suggestion}\n")