88 lines
4.5 KiB
Python
88 lines
4.5 KiB
Python
# Copyright 2026 zhaoxi826
|
|
#
|
|
# 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 typing import List
|
|
|
|
from kilostar.core.postgres_database.model.provider import Provider
|
|
from sqlmodel import select
|
|
from kilostar.core.postgres_database.database_exception import database_exception
|
|
|
|
|
|
class ProviderDatabase:
|
|
"""ProviderDatabase 核心组件类。
|
|
这是一个模型/服务提供商适配器类，屏蔽了外部不同供应商（如 OpenAI、Anthropic 等）的底层 API 差异。它负责标准化参数组装、网络请求发送、鉴权处理以及响应结构的反序列化。"""
|
|
|
|
def __init__(self, async_session_maker):
|
|
self.async_session_maker = async_session_maker
|
|
|
|
@database_exception
|
|
async def get_provider(self) -> List[Provider]:
|
|
"""检索并获取特定的 provider 数据集合或实例对象。
|
|
根据提供的查询条件或上下文凭证，从数据库、缓存或第三方服务中读取对应的资源状态。
|
|
Returns: (List[Provider]): 经过筛选、排序或分页处理后的实体对象列表集合。"""
|
|
async with self.async_session_maker() as session:
|
|
statement = select(Provider)
|
|
results = await session.execute(statement)
|
|
results = results.scalars().all()
|
|
providers = [
|
|
Provider(
|
|
provider_title=provider.provider_title,
|
|
provider_url=provider.provider_url,
|
|
provider_apikey=provider.provider_apikey,
|
|
provider_models=provider.provider_models,
|
|
provider_type=provider.provider_type,
|
|
)
|
|
for provider in results
|
|
]
|
|
return providers
|
|
|
|
@database_exception
|
|
async def add_provider(self, **kwargs) -> None:
|
|
"""创建并持久化新的 provider 实体。
|
|
接收构建参数，执行必要的数据校验与默认值填充后，将新记录安全地写入底层存储或系统注册表中。
|
|
Returns: (None): 经由当前业务模型加工处理后所输出的具体数据实例或领域模型对象。"""
|
|
async with self.async_session_maker() as session:
|
|
provider = Provider(**kwargs)
|
|
session.add(provider)
|
|
await session.commit()
|
|
|
|
@database_exception
|
|
async def delete_provider(self, provider_id: str) -> None:
|
|
"""安全地移除或注销 provider。
|
|
执行物理删除或逻辑删除操作，并妥善清理相关的关联数据及占用资源。
|
|
Args: provider_id (str): 目标对象的唯一全局标识符 (UUID/ULID)，用于在数据库表或缓存结构中精准匹配该 provider 实例。
|
|
Returns: (None): 经由当前业务模型加工处理后所输出的具体数据实例或领域模型对象。"""
|
|
async with self.async_session_maker() as session:
|
|
provider = await session.get(Provider, provider_id)
|
|
if provider is not None:
|
|
session.delete(provider)
|
|
await session.commit()
|
|
|
|
@database_exception
|
|
async def update_provider(self, provider_id: str, **kwargs) -> Provider:
|
|
"""对现有的 provider 进行状态更新或属性覆盖。
|
|
基于增量变更原则，合并最新的配置或数据，并触发相关依赖组件的缓存刷新或事件通知。
|
|
Args: provider_id (str): 目标对象的唯一全局标识符 (UUID/ULID)，用于在数据库表或缓存结构中精准匹配该 provider 实例。
|
|
Returns: (Provider): 经由当前业务模型加工处理后所输出的具体数据实例或领域模型对象。"""
|
|
async with self.async_session_maker() as session:
|
|
provider = await session.get(Provider, provider_id)
|
|
if provider is not None:
|
|
for key, value in kwargs.items():
|
|
setattr(provider, key, value)
|
|
session.add(provider)
|
|
await session.commit()
|
|
await session.refresh(provider)
|
|
return provider
|
|
return None
|