Use X-SMG-Routing-Key header instead of json body and add tests (#15826)
This commit is contained in:
@@ -10,7 +10,6 @@ fn test_embedding_request_serialization_string_input() {
|
||||
user: Some("user-1".to_string()),
|
||||
dimensions: Some(128),
|
||||
rid: Some("rid-123".to_string()),
|
||||
routing_id: None,
|
||||
log_metrics: None,
|
||||
};
|
||||
|
||||
@@ -34,7 +33,6 @@ fn test_embedding_request_serialization_array_input() {
|
||||
user: None,
|
||||
dimensions: None,
|
||||
rid: None,
|
||||
routing_id: None,
|
||||
log_metrics: None,
|
||||
};
|
||||
|
||||
@@ -53,7 +51,6 @@ fn test_embedding_generation_request_trait_string() {
|
||||
user: None,
|
||||
dimensions: None,
|
||||
rid: None,
|
||||
routing_id: None,
|
||||
log_metrics: None,
|
||||
};
|
||||
assert!(!req.is_stream());
|
||||
@@ -70,7 +67,6 @@ fn test_embedding_generation_request_trait_array() {
|
||||
user: None,
|
||||
dimensions: None,
|
||||
rid: None,
|
||||
routing_id: None,
|
||||
log_metrics: None,
|
||||
};
|
||||
assert_eq!(req.extract_text_for_routing(), "hello world");
|
||||
@@ -85,7 +81,6 @@ fn test_embedding_generation_request_trait_non_text() {
|
||||
user: None,
|
||||
dimensions: None,
|
||||
rid: None,
|
||||
routing_id: None,
|
||||
log_metrics: None,
|
||||
};
|
||||
assert_eq!(req.extract_text_for_routing(), "");
|
||||
@@ -100,7 +95,6 @@ fn test_embedding_generation_request_trait_mixed_array_ignores_nested() {
|
||||
user: None,
|
||||
dimensions: None,
|
||||
rid: None,
|
||||
routing_id: None,
|
||||
log_metrics: None,
|
||||
};
|
||||
// Only top-level string elements are extracted
|
||||
|
||||
@@ -17,7 +17,6 @@ fn test_rerank_request_serialization() {
|
||||
return_documents: true,
|
||||
rid: Some(StringOrArray::String("req-123".to_string())),
|
||||
user: Some("user-456".to_string()),
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
let serialized = to_string(&request).unwrap();
|
||||
@@ -60,7 +59,6 @@ fn test_rerank_request_validation_success() {
|
||||
return_documents: true,
|
||||
rid: None,
|
||||
user: None,
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
assert!(request.validate().is_ok());
|
||||
@@ -76,7 +74,6 @@ fn test_rerank_request_validation_empty_query() {
|
||||
return_documents: true,
|
||||
rid: None,
|
||||
user: None,
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
let result = request.validate();
|
||||
@@ -93,7 +90,6 @@ fn test_rerank_request_validation_whitespace_query() {
|
||||
return_documents: true,
|
||||
rid: None,
|
||||
user: None,
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
let result = request.validate();
|
||||
@@ -110,7 +106,6 @@ fn test_rerank_request_validation_empty_documents() {
|
||||
return_documents: true,
|
||||
rid: None,
|
||||
user: None,
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
let result = request.validate();
|
||||
@@ -127,7 +122,6 @@ fn test_rerank_request_validation_top_k_zero() {
|
||||
return_documents: true,
|
||||
rid: None,
|
||||
user: None,
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
let result = request.validate();
|
||||
@@ -144,7 +138,6 @@ fn test_rerank_request_validation_top_k_greater_than_docs() {
|
||||
return_documents: true,
|
||||
rid: None,
|
||||
user: None,
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
// This should pass but log a warning
|
||||
@@ -161,7 +154,6 @@ fn test_rerank_request_effective_top_k() {
|
||||
return_documents: true,
|
||||
rid: None,
|
||||
user: None,
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
assert_eq!(request.effective_top_k(), 2);
|
||||
@@ -177,7 +169,6 @@ fn test_rerank_request_effective_top_k_none() {
|
||||
return_documents: true,
|
||||
rid: None,
|
||||
user: None,
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
assert_eq!(request.effective_top_k(), 3);
|
||||
@@ -399,7 +390,6 @@ fn test_rerank_request_generation_request_trait() {
|
||||
return_documents: true,
|
||||
rid: None,
|
||||
user: None,
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
assert_eq!(request.get_model(), Some("test-model"));
|
||||
@@ -418,7 +408,6 @@ fn test_rerank_request_very_long_query() {
|
||||
return_documents: true,
|
||||
rid: None,
|
||||
user: None,
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
assert!(request.validate().is_ok());
|
||||
@@ -435,7 +424,6 @@ fn test_rerank_request_many_documents() {
|
||||
return_documents: true,
|
||||
rid: None,
|
||||
user: None,
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
assert!(request.validate().is_ok());
|
||||
@@ -455,7 +443,6 @@ fn test_rerank_request_special_characters() {
|
||||
return_documents: true,
|
||||
rid: Some(StringOrArray::String("req-🚀-123".to_string())),
|
||||
user: Some("user-🎉-456".to_string()),
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
assert!(request.validate().is_ok());
|
||||
@@ -474,7 +461,6 @@ fn test_rerank_request_rid_array() {
|
||||
"req2".to_string(),
|
||||
])),
|
||||
user: None,
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
assert!(request.validate().is_ok());
|
||||
@@ -529,7 +515,6 @@ fn test_full_rerank_workflow() {
|
||||
return_documents: true,
|
||||
rid: Some(StringOrArray::String("req-123".to_string())),
|
||||
user: Some("user-456".to_string()),
|
||||
routing_id: None,
|
||||
};
|
||||
|
||||
// Validate request
|
||||
|
||||
Reference in New Issue
Block a user