fixing ruff suggestions and tests needed updating.
This commit is contained in:
@@ -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.",
|
||||
|
||||
Reference in New Issue
Block a user