feat(models): 组织、用户、角色、客户、会话与消息模型
按组织边界定义数据关系,补齐仓储层与通用工具,迁移脚本可重复执行。
This commit is contained in:
1 parent
977783c126
commit
9eb8613c05
126 files changed
+17436
No files matched your search
@@ -0,0 +1,116 @@
|
||||
"""
|
||||
Copyright 2024-2026 ReplyPier
|
||||
|
||||
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.
|
||||
"""
|
||||
|
||||
from sqlalchemy.orm import Session
|
||||
from app.models.ai_config import AIConfig, AIModelType, DEFAULT_AI_SETTINGS
|
||||
from app.core.security import encrypt_api_key, decrypt_api_key
|
||||
from typing import Optional
|
||||
|
||||
|
||||
class AIConfigRepository:
|
||||
def __init__(self, db: Session):
|
||||
self.db = db
|
||||
|
||||
def create_config(self, org_id: str, model_type: str, model_name: str, api_key: str, base_url: Optional[str] = None) -> AIConfig:
|
||||
"""Create a new AI configuration"""
|
||||
try:
|
||||
# Encrypt API key before storing
|
||||
encrypted_key = encrypt_api_key(api_key)
|
||||
|
||||
# Create new config
|
||||
config = AIConfig(
|
||||
organization_id=org_id,
|
||||
model_type=AIModelType(model_type.upper()),
|
||||
model_name=model_name,
|
||||
encrypted_api_key=encrypted_key,
|
||||
is_active=True
|
||||
)
|
||||
if base_url:
|
||||
# No dedicated column: base_url lives in the JSON settings blob.
|
||||
# The column default is only applied by SQLAlchemy at flush time,
|
||||
# so start from the same default explicitly rather than {}.
|
||||
config.settings = {**DEFAULT_AI_SETTINGS, "base_url": base_url}
|
||||
|
||||
# Deactivate any existing configs
|
||||
existing_configs = self.db.query(AIConfig).filter(
|
||||
AIConfig.organization_id == org_id,
|
||||
AIConfig.is_active == True
|
||||
).all()
|
||||
for existing in existing_configs:
|
||||
existing.is_active = False
|
||||
|
||||
self.db.add(config)
|
||||
self.db.commit()
|
||||
self.db.refresh(config)
|
||||
return config
|
||||
|
||||
except Exception as e:
|
||||
self.db.rollback()
|
||||
raise e
|
||||
|
||||
def get_active_config(self, org_id: str) -> Optional[AIConfig]:
|
||||
"""Get active AI configuration for an organization"""
|
||||
config = self.db.query(AIConfig).filter(
|
||||
AIConfig.organization_id == org_id,
|
||||
AIConfig.is_active == True
|
||||
).first()
|
||||
|
||||
if not config:
|
||||
return None
|
||||
|
||||
# Return dict with decrypted API key
|
||||
return config
|
||||
|
||||
def update_config(self, config_id: int, **kwargs) -> Optional[AIConfig]:
|
||||
"""Update an existing AI configuration"""
|
||||
config = self.db.query(AIConfig).filter(
|
||||
AIConfig.id == config_id).first()
|
||||
if not config:
|
||||
return None
|
||||
|
||||
for key, value in kwargs.items():
|
||||
if key == 'api_key':
|
||||
if value is not None:
|
||||
config.encrypted_api_key = encrypt_api_key(value)
|
||||
elif key == 'base_url':
|
||||
# None means "no change" (mirrors api_key above). An explicit
|
||||
# empty string clears it. Merged into settings JSON — no
|
||||
# dedicated column — so the rest of settings survives.
|
||||
if value is None:
|
||||
continue
|
||||
current_settings = dict(config.settings or DEFAULT_AI_SETTINGS)
|
||||
if value:
|
||||
current_settings['base_url'] = value
|
||||
else:
|
||||
current_settings.pop('base_url', None)
|
||||
config.settings = current_settings
|
||||
else:
|
||||
setattr(config, key, value)
|
||||
|
||||
self.db.commit()
|
||||
self.db.refresh(config)
|
||||
return config
|
||||
|
||||
def deactivate_config(self, config_id: int) -> bool:
|
||||
"""Deactivate an AI configuration"""
|
||||
config = self.db.query(AIConfig).filter(
|
||||
AIConfig.id == config_id).first()
|
||||
if not config:
|
||||
return False
|
||||
|
||||
config.is_active = False
|
||||
self.db.commit()
|
||||
return True
|
||||
Reference in new issue
Block a user