17 lines
531 B
Python
17 lines
531 B
Python
from typing import Optional
|
|
|
|
from embedchain.config.base_config import BaseConfig
|
|
from embedchain.helpers.json_serializable import register_deserializable
|
|
|
|
|
|
@register_deserializable
|
|
class CacheConfig(BaseConfig):
|
|
def __init__(
|
|
self,
|
|
similarity_threshold: Optional[float] = 0.5,
|
|
):
|
|
if similarity_threshold < 0 or similarity_threshold > 1:
|
|
raise ValueError(f"similarity_threshold {similarity_threshold} should be between 0 and 1")
|
|
|
|
self.similarity_threshold = similarity_threshold
|