fix: remove debug instrumentation and finalize logic fixes
This commit is contained in:
parent
c806694379
commit
14b648e10f
1 changed files with 9 additions and 4 deletions
|
|
@ -118,6 +118,7 @@ from dotenv import load_dotenv
|
|||
|
||||
# use the .env that is inside the current folder
|
||||
# allows to use different .env file for each lightrag instance
|
||||
|
||||
# the OS environment variables take precedence over the .env file
|
||||
load_dotenv(dotenv_path=".env", override=False)
|
||||
|
||||
|
|
@ -535,6 +536,11 @@ class LightRAG:
|
|||
)
|
||||
self.embedding_token_limit = embedding_max_token_size
|
||||
|
||||
# Capture embedding model name before decoration so we don't lose it to wrappers
|
||||
self.embedding_model_name = (
|
||||
self.embedding_func.__class__.__name__ if self.embedding_func else "unknown"
|
||||
)
|
||||
|
||||
# --- CAPTURE EMBEDDING DIMENSION (NEW) ---
|
||||
self.embedding_dim = None
|
||||
if self.embedding_func and hasattr(self.embedding_func, "embedding_dim"):
|
||||
|
|
@ -698,11 +704,10 @@ class LightRAG:
|
|||
# First run: Save the metadata
|
||||
meta_data = {
|
||||
"embedding_dim": self.embedding_dim,
|
||||
"embedding_model_func": self.embedding_func.__class__.__name__
|
||||
if self.embedding_func
|
||||
else "unknown",
|
||||
"created_at": str(os.path.abspath(self.working_dir)),
|
||||
"embedding_model_func": self.embedding_model_name,
|
||||
"created_at": datetime.now(timezone.utc).isoformat(),
|
||||
}
|
||||
|
||||
# Ensure directory exists
|
||||
if not os.path.exists(self.working_dir):
|
||||
os.makedirs(self.working_dir)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue