cache_config.py 531 B

12345678910111213141516
  1. from typing import Optional
  2. from embedchain.config.base_config import BaseConfig
  3. from embedchain.helpers.json_serializable import register_deserializable
  4. @register_deserializable
  5. class CacheConfig(BaseConfig):
  6. def __init__(
  7. self,
  8. similarity_threshold: Optional[float] = 0.5,
  9. ):
  10. if similarity_threshold < 0 or similarity_threshold > 1:
  11. raise ValueError(f"similarity_threshold {similarity_threshold} should be between 0 and 1")
  12. self.similarity_threshold = similarity_threshold