Package com.tnsai.llm.routing

Class TaskBasedRouter.Builder

java.lang.Object

com.tnsai.llm.routing.TaskBasedRouter.Builder

Enclosing class:

TaskBasedRouter

public static class TaskBasedRouter.Builder
extends Object

Builder for TaskBasedRouter.

Constructor Summary

Constructors

Constructor	Description
Builder()

Method Summary

Modifier and Type	Method	Description
TaskBasedRouter	build()
TaskBasedRouter.Builder	classifier(TaskClassifier classifier)	Uses a custom TaskClassifier.
TaskBasedRouter.Builder	confidenceThreshold(double threshold)	Sets the minimum confidence threshold for task classification.
TaskBasedRouter.Builder	defaultClient(com.tnsai.llm.LLMClient client)	Sets the default client for unmatched or low-confidence tasks.
TaskBasedRouter.Builder	forTask(TaskType type, com.tnsai.llm.LLMClient client)	Assigns a client for a specific task type.
TaskBasedRouter.Builder	withCommonSetup(com.tnsai.llm.LLMClient coding, com.tnsai.llm.LLMClient reasoning, com.tnsai.llm.LLMClient fast, com.tnsai.llm.LLMClient general)	Convenience method to set up common task mappings.

Methods inherited from class java.lang.Object

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

Constructor Details

Builder

public Builder()

Method Details

forTask

public TaskBasedRouter.Builder forTask(TaskType type,
 com.tnsai.llm.LLMClient client)

Assigns a client for a specific task type.

defaultClient

public TaskBasedRouter.Builder defaultClient(com.tnsai.llm.LLMClient client)

Sets the default client for unmatched or low-confidence tasks.

classifier

public TaskBasedRouter.Builder classifier(TaskClassifier classifier)

Uses a custom TaskClassifier.

confidenceThreshold

public TaskBasedRouter.Builder confidenceThreshold(double threshold)

Sets the minimum confidence threshold for task classification. Below this threshold, default client is used.

withCommonSetup

public TaskBasedRouter.Builder withCommonSetup(com.tnsai.llm.LLMClient coding,
 com.tnsai.llm.LLMClient reasoning,
 com.tnsai.llm.LLMClient fast,
 com.tnsai.llm.LLMClient general)

Convenience method to set up common task mappings.

Parameters:

coding - Client for coding tasks

reasoning - Client for math/reasoning tasks

fast - Client for simple/fast tasks

general - Default client

build

public TaskBasedRouter build()