fixing ruff suggestions and tests needed updating.
All checks were successful
CI / Lint (push) Successful in 1m39s
CI / Test (push) Successful in 1m37s
CI / Release (push) Successful in 6s
CI / Notify (push) Successful in 1s

This commit is contained in:
2026-02-18 07:37:13 -05:00
parent 24a4098c9a
commit a1cf87909d
3 changed files with 40 additions and 17 deletions

View File

@@ -53,7 +53,7 @@ def sample_reranked():
def mock_nats_message():
"""Create a mock NATS message."""
msg = MagicMock()
msg.subject = "ai.chat.request"
msg.subject = "ai.chat.user.test-user-1.message"
msg.reply = "ai.chat.response.test-123"
return msg
@@ -63,7 +63,13 @@ def mock_chat_request():
"""Sample chat request payload."""
return {
"request_id": "test-request-123",
"query": "What is machine learning?",
"user_id": "test-user-1",
"username": "testuser",
"message": "What is machine learning?",
"premium": True,
"enable_rag": True,
"enable_reranker": True,
"enable_streaming": False,
"collection": "test_collection",
"enable_tts": False,
"system_prompt": None,
@@ -75,7 +81,13 @@ def mock_chat_request_with_tts():
"""Sample chat request with TTS enabled."""
return {
"request_id": "test-request-456",
"query": "Tell me about AI",
"user_id": "test-user-2",
"username": "testuser2",
"message": "Tell me about AI",
"premium": True,
"enable_rag": True,
"enable_reranker": True,
"enable_streaming": False,
"collection": "documents",
"enable_tts": True,
"system_prompt": "You are a helpful assistant.",