mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-29 02:23:44 +02:00
Implement logging strategy (#444)
* Logging strategy and convert all prints() to logging invocations
This commit is contained in:
parent
3e0651222b
commit
dd70aade11
117 changed files with 1216 additions and 667 deletions
|
|
@ -248,9 +248,13 @@ class TestDocumentRagIntegration:
|
|||
@pytest.mark.asyncio
|
||||
async def test_document_rag_verbose_logging(self, mock_embeddings_client,
|
||||
mock_doc_embeddings_client, mock_prompt_client,
|
||||
capsys):
|
||||
caplog):
|
||||
"""Test DocumentRAG verbose logging functionality"""
|
||||
# Arrange
|
||||
import logging
|
||||
|
||||
# Arrange - Configure logging to capture debug messages
|
||||
caplog.set_level(logging.DEBUG)
|
||||
|
||||
document_rag = DocumentRag(
|
||||
embeddings_client=mock_embeddings_client,
|
||||
doc_embeddings_client=mock_doc_embeddings_client,
|
||||
|
|
@ -261,14 +265,14 @@ class TestDocumentRagIntegration:
|
|||
# Act
|
||||
await document_rag.query("test query for verbose logging")
|
||||
|
||||
# Assert
|
||||
captured = capsys.readouterr()
|
||||
assert "Initialised" in captured.out
|
||||
assert "Construct prompt..." in captured.out
|
||||
assert "Compute embeddings..." in captured.out
|
||||
assert "Get docs..." in captured.out
|
||||
assert "Invoke LLM..." in captured.out
|
||||
assert "Done" in captured.out
|
||||
# Assert - Check for new logging messages
|
||||
log_messages = caplog.text
|
||||
assert "DocumentRag initialized" in log_messages
|
||||
assert "Constructing prompt..." in log_messages
|
||||
assert "Computing embeddings..." in log_messages
|
||||
assert "Getting documents..." in log_messages
|
||||
assert "Invoking LLM..." in log_messages
|
||||
assert "Query processing complete" in log_messages
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@pytest.mark.slow
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue