youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-06-08 09:27:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix : features get null vector_space

Browse Source
This commit is contained in:
hjlarry
2026-05-25 12:04:29 +08:00
parent 9ddd98a265
commit 3cabe9058b
2 changed files with 34 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/services/billing_service.py
View File

@ -192,6 +192,9 @@ class BillingService:
params["exclude_vector_space"] = "true"
billing_info = cls._send_request("GET", "/subscription/info", params=params)
if exclude_vector_space and billing_info.get("vector_space") is None:
# Unset proto message fields can be serialized as null; the light billing contract treats it as absent.
billing_info.pop("vector_space", None)
return _billing_info_adapter.validate_python(billing_info)
@classmethod

31
api/tests/unit_tests/services/test_billing_service.py
View File

@ -343,6 +343,37 @@ class TestBillingServiceSubscriptionInfo:
params={"tenant_id": tenant_id, "exclude_vector_space": "true"},
)
def test_get_info_exclude_vector_space_normalizes_null_field(self, mock_send_request):
"""When billing serializes skipped vector_space as null, get_info treats it as absent."""
# Arrange
tenant_id = "tenant-123"
expected_response = {
"enabled": True,
"subscription": {"plan": "professional", "interval": "month", "education": False},
"members": {"size": 1, "limit": 50},
"apps": {"size": 1, "limit": 200},
"vector_space": None,
"knowledge_rate_limit": {"limit": 1000},
"documents_upload_quota": {"size": 0, "limit": 1000},
"annotation_quota_limit": {"size": 0, "limit": 5000},
"docs_processing": "top-priority",
"can_replace_logo": True,
"model_load_balancing_enabled": True,
"knowledge_pipeline_publish_enabled": True,
}
mock_send_request.return_value = expected_response
# Act
result = BillingService.get_info(tenant_id, exclude_vector_space=True)
# Assert
assert "vector_space" not in result
mock_send_request.assert_called_once_with(
"GET",
"/subscription/info",
params={"tenant_id": tenant_id, "exclude_vector_space": "true"},
)
def test_get_vector_space_success(self, mock_send_request):
"""Test successful retrieval of vector-space usage and limit."""
# Arrange