12 lines
453 B
Python
12 lines
453 B
Python
|
class KnowledgeRuntimeConfig:
|
||
|
|
DEFAULT_ROUTES = {
|
||
|
|
"faq": ("fin_faq_collection", 3),
|
||
|
|
"product_inquiry": ("fin_product_collection", 5),
|
||
|
|
"policy_explain": ("fin_policy_collection", 5),
|
||
|
|
}
|
||
|
|
|
||
|
|
def __init__(self, *, vector_dim: int = 1024, similarity_threshold: float = 0.70) -> None:
|
||
|
|
self.routes = dict(self.DEFAULT_ROUTES)
|
||
|
|
self.vector_dim = vector_dim
|
||
|
|
self.similarity_threshold = similarity_threshold
|