UnitGen

Architecture

Design Philosophy

Unique Prompt

AutoDev prompt

Write unit test for following code.${context.coc}${context.framework}${context.related_model}```${context.language}${context.selection}```

Unit Picker prompt

Instruction(    instruction="Complete${it.language} code, return rest code, no explaining",    output= it.output,    input="""    |```${it.language}    |${it.relatedCode}    |```    |    |Code:    |```${it.language}    |${it.beforeCursor}    |```""".trimMargin())

UnitGen prompt

Complete ${language} code, return rest code, no explaining```${language}${relatedCode}```Code:```${language}${beforeCursor}```

Code quality pipeline

Extendable customize quality thresholds

enumclassCodeQualityType {BadSmell,TestBadSmell,JavaController,JavaRepository,JavaService,}

data classBsThresholds(valbsLongParasLength:Int =5,valbsIfSwitchLength:Int =8,valbsLargeLength:Int =20,valbsMethodLength:Int =30,valbsIfLinesLength:Int =3,)

val apis= apiAnalyser.toContainerServices()val ruleset=RuleSet(RuleType.SQL_SMELL,"normal",UnknownColumnSizeRule(),LimitTableNameLengthRule()// more rules)val issues=WebApiRuleVisitor(apis).visitor(listOf(ruleset))// if issues are not empty, then the code has bad smell

Quick Start

use CLI

Generate Instructions

use Java API

dependencies {    implementation("cc.unitmesh:unit-picker:0.1.5")    implementation("cc.unitmesh:code-quality:0.1.5")}

publicclassApp {publicstaticvoidmain(String[]args) {List<InstructionType>builderTypes =newArrayList<>();builderTypes.add(InstructionType.RELATED_CODE_COMPLETION);List<CodeQualityType>codeQualityTypes =newArrayList<>();codeQualityTypes.add(CodeQualityType.BadSmell);codeQualityTypes.add(CodeQualityType.JavaService);PickerOptionpickerOption =newPickerOption("https://github.com/unit-mesh/unit-gen-testing","master","java",".",builderTypes,codeQualityTypes,newBuilderConfig()        );SimpleCodePickersimpleCodePicker =newSimpleCodePicker(pickerOption);List<Instruction>output =simpleCodePicker.blockingExecute();// handle output in here    }}

Thanks to

LICENSE