wt002 commited on
Commit
b88e3ed
·
verified ·
1 Parent(s): 1c46468

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +7 -3
agent.py CHANGED
@@ -388,9 +388,13 @@ def create_documents(data_source: str, data: List[dict]) -> List[Document]:
388
  "level": str(item.get("Level", "")),
389
  "file_name": str(item.get("file_name", ""))
390
  }
391
-
392
- docs.append(Document(page_content=content, metadata=metadata))
393
-
 
 
 
 
394
  return docs
395
 
396
  # 4. Vector Store Integration
 
388
  "level": str(item.get("Level", "")),
389
  "file_name": str(item.get("file_name", ""))
390
  }
391
+
392
+ # Make sure that the content and metadata are valid
393
+ if content.strip(): # Only append non-empty content
394
+ docs.append(Document(page_content=content, metadata=metadata))
395
+ else:
396
+ print(f"Skipping invalid entry: {item}")
397
+
398
  return docs
399
 
400
  # 4. Vector Store Integration