klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Feature] Add support for NVIDIA AI LLMs and embedding models (#1293)

Browse Source
This commit is contained in:
Deshraj Yadav
2024-02-29 23:56:25 -08:00
committed by GitHub
parent 6518c0c06b
commit c77a75dfb5
18 changed files with 195 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
embedchain/config/base_app_config.py
View File

@@ -36,7 +36,6 @@ class BaseAppConfig(BaseConfig, JSONSerializable):
defaults to None
:type collection_name: Optional[str], optional
"""
self._setup_logging(log_level)
self.id = id
self.collect_metrics = True if (collect_metrics is True or collect_metrics is None) else False
self.collection_name = collection_name
@@ -52,12 +51,6 @@ class BaseAppConfig(BaseConfig, JSONSerializable):
logging.warning("DEPRECATION WARNING: Please supply the collection name to the database config.")
return
def _setup_logging(self, debug_level):
level = logging.WARNING # Default level
if debug_level is not None:
level = getattr(logging, debug_level.upper(), None)
if not isinstance(level, int):
raise ValueError(f"Invalid log level: {debug_level}")
logging.basicConfig(format="%(asctime)s [%(name)s] [%(levelname)s] %(message)s", level=level)
def _setup_logging(self, log_level):
logging.basicConfig(format="%(asctime)s [%(name)s] [%(levelname)s] %(message)s", level=log_level)
self.logger = logging.getLogger(__name__)