TBNilles/LEANN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use logger instead of print (#186)

Browse Source 
Signed-off-by: droctothorpe <mythicalsunlight@gmail.com>
This commit is contained in:
Alex
2025-12-10 16:48:57 -05:00
committed by GitHub
parent a0bbf831db
commit 3629ccf8f7
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
packages/leann-core/src/leann/api.py
View File

@@ -1251,15 +1251,15 @@ class LeannChat:
"Please provide the best answer you can based on this context and your knowledge." "Please provide the best answer you can based on this context and your knowledge."
) )
print("The context provided to the LLM is:") logger.info("The context provided to the LLM is:")
print(f"{'Relevance':<10} | {'Chunk id':<10} | {'Content':<60} | {'Source':<80}") logger.info(f"{'Relevance':<10} | {'Chunk id':<10} | {'Content':<60} | {'Source':<80}")
print("-" * 150) logger.info("-" * 150)
for r in results: for r in results:
chunk_relevance = f"{r.score:.3f}" chunk_relevance = f"{r.score:.3f}"
chunk_id = r.id chunk_id = r.id
chunk_content = r.text[:60] chunk_content = r.text[:60]
chunk_source = r.metadata.get("source", "")[:80] chunk_source = r.metadata.get("source", "")[:80]
print( logger.info(
f"{chunk_relevance:<10} | {chunk_id:<10} | {chunk_content:<60} | {chunk_source:<80}" f"{chunk_relevance:<10} | {chunk_id:<10} | {chunk_content:<60} | {chunk_source:<80}"
) )
ask_time = time.time() ask_time = time.time()