Package com.tnsai.llm.routing

Class TaskClassifier.Builder

java.lang.Object

com.tnsai.llm.routing.TaskClassifier.Builder

Enclosing class:

TaskClassifier

public static class TaskClassifier.Builder
extends Object

Builder for TaskClassifier.

Constructor Summary

Constructors

Constructor	Description
Builder()

Method Summary

Modifier and Type	Method	Description
TaskClassifier.Builder	addKeywords(TaskType type, Set<String> keywords)	Adds custom keywords for a task type.
TaskClassifier.Builder	addPattern(TaskType type, Pattern pattern)	Adds a custom pattern for a task type.
TaskClassifier	build()
TaskClassifier.Builder	setDefaultType(TaskType defaultType)	Sets the default task type when classification confidence is too low.
TaskClassifier.Builder	setMinConfidenceThreshold(double threshold)	Sets the minimum confidence threshold.

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

Builder

public Builder()

Method Details

addKeywords

public TaskClassifier.Builder addKeywords(TaskType type,
 Set<String> keywords)

Adds custom keywords for a task type.

addPattern

public TaskClassifier.Builder addPattern(TaskType type,
 Pattern pattern)

Adds a custom pattern for a task type.

setDefaultType

public TaskClassifier.Builder setDefaultType(TaskType defaultType)

Sets the default task type when classification confidence is too low.

setMinConfidenceThreshold

public TaskClassifier.Builder setMinConfidenceThreshold(double threshold)

Sets the minimum confidence threshold. Below this threshold, default type is used.

build

public TaskClassifier build()