feat: support Space rerank models (#2358)

Co-authored-by: chan <dadachann@users.noreply.github.com>
2026-08-09 12:40:59 +00:00 · 2026-07-26 16:37:34 +08:00
parent 38e35d328a
commit dd8d1007a1
7 changed files with 171 additions and 5 deletions
@@ -693,7 +693,7 @@ class TestSpaceServiceGetModels:
                            'uuid': 'uuid-2',
                            'model_id': 'model-2',
                            'provider': 'provider-2',
-                            'category': 'chat',
+                            'category': 'rerank',
                            'status': 'active',
                        },
                    ]
@@ -714,6 +714,7 @@ class TestSpaceServiceGetModels:

        # Verify
        assert len(result) == 2
+        assert result[1].category == 'rerank'

    async def test_get_models_api_error(self):
        """Raises ValueError on API error."""