feat(threshold): add basic threshold design

unit-mesh · Dec 19, 2023 · d3418ae · d3418ae
1 parent 14b740c
commit d3418ae
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -88,6 +88,11 @@ Unit Eval prompt should keep the same structure as the AutoDev prompt. Prompt ex
 
 ### Code quality pipeline
 
+Before Check:
+
+- FileSize: 64k
+- Complexity: 1000
+
 ![Code Quality Workflow](https://unitmesh.cc/uniteval/code-quality-workflow.png)
 
 ### Extendable customize quality thresholds

diff --git a/docs/quality/threshold.md b/docs/quality/threshold.md
@@ -8,7 +8,7 @@ permalink: /quality/threshold
 
 Default threshold:
 
-- File Size: 64kb, about 640 ~ 1280 lines,
+- File Size: 64kb, about 640 ~ 1280 lines, maybe some comments, like chinese
 - Complexity: 100,
 
 

diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/worker/WorkerContext.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/worker/WorkerContext.kt
@@ -4,7 +4,9 @@ import cc.unitmesh.pick.builder.BuilderConfig
 import cc.unitmesh.pick.prompt.CodeContextStrategy
 import cc.unitmesh.pick.prompt.CompletionBuilderType
 import cc.unitmesh.quality.CodeQualityType
+import kotlinx.serialization.Serializable
 
+@Serializable
 data class WorkerContext(
     val codeContextStrategies: List<CodeContextStrategy>,
     val qualityTypes: List<CodeQualityType>,
@@ -13,3 +15,8 @@ data class WorkerContext(
     val completionTypes: List<CompletionBuilderType>,
     val maxCompletionInOneFile: Int,
 )
+@Serializable
+data class QualityThreshold(
+    val complexity: Long = 100,
+    val fileSize: Long = 1024 * 64,
+)
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/worker/WorkerManager.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/worker/WorkerManager.kt
@@ -33,7 +33,7 @@ class WorkerManager(workerContext: WorkerContext) {
 
         val language = job.fileSummary.language.toSupportLanguage()
         val worker = workers[language] ?: return
-        logger.info("add file:" + job.fileSummary.location)
+//        logger.info("add file:" + job.fileSummary.location)
         worker.addJob(job)
     }