test: 补充知识处理、检索与 Agent 回归用例

覆盖切块、向量检索、工具调用和历史预算；模型相关用例使用桩。

backend/tests/repo/test_ai_configrepo.py

@@ -0,0 +1,207 @@
+"""
+Copyright 2024-2026 ChatterMate
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+import pytest
+from app.models.ai_config import AIConfig, AIModelType
+from app.repositories.ai_config import AIConfigRepository
+from app.core.security import decrypt_api_key
+from uuid import UUID
+from app.models.organization import Organization
+
+@pytest.fixture
+def ai_config_repo(db):
+    """Create an AI config repository instance"""
+    return AIConfigRepository(db)
+
+def test_create_config(ai_config_repo, test_organization_id):
+    """Test creating a new AI configuration"""
+    model_type = "openai"
+    model_name = "gpt-4"
+    api_key = "test-api-key-123"
+
+    config = ai_config_repo.create_config(
+        org_id=test_organization_id,
+        model_type=model_type,
+        model_name=model_name,
+        api_key=api_key
+    )
+
+    assert config is not None
+    assert config.organization_id == test_organization_id
+    assert config.model_type == AIModelType.OPENAI
+    assert config.model_name == model_name
+    assert config.is_active is True
+    # Verify API key is encrypted
+    assert config.encrypted_api_key != api_key
+    # Verify API key can be decrypted
+    decrypted_key = decrypt_api_key(config.encrypted_api_key)
+    assert decrypted_key == api_key
+
+def test_create_config_deactivates_existing(ai_config_repo, test_organization_id):
+    """Test that creating a new config deactivates existing ones"""
+    # Create first config
+    config1 = ai_config_repo.create_config(
+        org_id=test_organization_id,
+        model_type="openai",
+        model_name="gpt-4",
+        api_key="key1"
+    )
+
+    # Create second config
+    config2 = ai_config_repo.create_config(
+        org_id=test_organization_id,
+        model_type="anthropic",
+        model_name="claude-2",
+        api_key="key2"
+    )
+
+    # Verify first config is deactivated
+    assert config1.is_active is False
+    # Verify second config is active
+    assert config2.is_active is True
+
+def test_get_active_config(ai_config_repo, test_organization_id):
+    """Test retrieving active AI configuration"""
+    # Create a config
+    original_config = ai_config_repo.create_config(
+        org_id=test_organization_id,
+        model_type="openai",
+        model_name="gpt-4",
+        api_key="test-key"
+    )
+
+    # Get active config
+    active_config = ai_config_repo.get_active_config(test_organization_id)
+    assert active_config is not None
+    assert active_config.id == original_config.id
+    assert active_config.model_type == original_config.model_type
+    assert active_config.model_name == original_config.model_name
+
+def test_get_active_config_none_exists(ai_config_repo, test_organization_id):
+    """Test retrieving active config when none exists"""
+    config = ai_config_repo.get_active_config(test_organization_id)
+    assert config is None
+
+def test_update_config(ai_config_repo, test_organization_id):
+    """Test updating an AI configuration"""
+    # Create initial config
+    config = ai_config_repo.create_config(
+        org_id=test_organization_id,
+        model_type="openai",
+        model_name="gpt-4",
+        api_key="old-key"
+    )
+
+    # Update config
+    new_model_name = "gpt-4-turbo"
+    new_api_key = "new-key"
+    updated_config = ai_config_repo.update_config(
+        config_id=config.id,
+        model_name=new_model_name,
+        api_key=new_api_key
+    )
+
+    assert updated_config is not None
+    assert updated_config.id == config.id
+    assert updated_config.model_name == new_model_name
+    # Verify new API key is encrypted and can be decrypted
+    decrypted_key = decrypt_api_key(updated_config.encrypted_api_key)
+    assert decrypted_key == new_api_key
+
+def test_update_nonexistent_config(ai_config_repo):
+    """Test updating a non-existent configuration"""
+    result = ai_config_repo.update_config(
+        config_id=999,
+        model_name="new-model"
+    )
+    assert result is None
+
+def test_deactivate_config(ai_config_repo, test_organization_id):
+    """Test deactivating an AI configuration"""
+    # Create config
+    config = ai_config_repo.create_config(
+        org_id=test_organization_id,
+        model_type="openai",
+        model_name="gpt-4",
+        api_key="test-key"
+    )
+
+    # Deactivate config
+    success = ai_config_repo.deactivate_config(config.id)
+    assert success is True
+
+    # Verify config is deactivated
+    active_config = ai_config_repo.get_active_config(test_organization_id)
+    assert active_config is None
+
+def test_deactivate_nonexistent_config(ai_config_repo):
+    """Test deactivating a non-existent configuration"""
+    success = ai_config_repo.deactivate_config(999)
+    assert success is False
+
+def test_multiple_organizations_configs(ai_config_repo, db):
+    """Test handling configs for multiple organizations"""
+    # Create test organizations
+    org1 = Organization(
+        name="Test Organization 1",
+        domain="test1.com",
+        timezone="UTC"
+    )
+    db.add(org1)
+    
+    org2 = Organization(
+        name="Test Organization 2",
+        domain="test2.com",
+        timezone="UTC"
+    )
+    db.add(org2)
+    db.commit()
+    db.refresh(org1)
+    db.refresh(org2)
+
+    # Create configs for both organizations
+    config1 = ai_config_repo.create_config(
+        org_id=org1.id,
+        model_type="openai",
+        model_name="gpt-4",
+        api_key="key1"
+    )
+
+    config2 = ai_config_repo.create_config(
+        org_id=org2.id,
+        model_type="anthropic",
+        model_name="claude-2",
+        api_key="key2"
+    )
+
+    # Verify both configs are active
+    active1 = ai_config_repo.get_active_config(org1.id)
+    active2 = ai_config_repo.get_active_config(org2.id)
+
+    assert active1.id == config1.id
+    assert active2.id == config2.id
+    assert active1.is_active is True
+    assert active2.is_active is True
+
+def test_create_config_invalid_model_type(ai_config_repo, test_organization_id):
+    """Test creating config with invalid model type"""
+    with pytest.raises(ValueError):
+        ai_config_repo.create_config(
+            org_id=test_organization_id,
+            model_type="invalid",
+            model_name="model",
+            api_key="key"
+        )