def _escape(value): if value is None: return None return str(value).replace("'", "''") # 客户侧可见字段(不含 listing_status、is_active 等运营字段) _CUSTOMER_MODEL_COLUMNS = """ id, llmid, provider, model_name, display_name, model_type, context_length, input_token_price, output_token_price, cache_hit_input_price, billing_method, billing_unit, capabilities, limitations, highlights, description, sort_order """ def _customer_listed_conditions(ns): """已上架且启用的模型;支持按厂商、模型类别筛选""" conditions = ["listing_status = 1", "is_active = 1"] if ns.get('provider'): conditions.append("provider = '%s'" % _escape(ns.get('provider'))) if ns.get('model_type'): conditions.append("model_type = '%s'" % _escape(ns.get('model_type'))) return ' AND '.join(conditions) async def model_management_customer_search(ns={}): """ 客户查看模型列表:仅已上架且启用的模型。 可选参数: provider (str) 厂商,精确匹配筛选 model_type (str) 模型类别,精确匹配筛选 current_page (int) 页码,默认 1 page_size (int) 每页条数,默认 10 返回 data: provider_list 当前可见模型中的厂商列表(去重) model_type_list 当前可见模型中的模型类别列表(去重) filter_total 当前筛选条件下的模型数量 model_list 模型列表 page_size, current_page 调用示例见 model_management_customer_search.dspy """ page_size = int(ns.get('page_size', 1000)) current_page = int(ns.get('current_page', 1)) offset = (current_page - 1) * page_size where_clause = _customer_listed_conditions(ns) listed_base = "listing_status = 1 AND is_active = 1" db = DBPools() async with db.sqlorContext('kboss') as sor: try: provider_sql = """ SELECT DISTINCT provider FROM model_management WHERE %s AND provider IS NOT NULL AND provider != '' ORDER BY provider; """ % listed_base model_type_sql = """ SELECT DISTINCT model_type FROM model_management WHERE %s AND model_type IS NOT NULL AND model_type != '' ORDER BY model_type; """ % listed_base count_sql = """ SELECT COUNT(*) AS total_count FROM model_management WHERE %s; """ % where_clause find_sql = """ SELECT %s FROM model_management WHERE %s ORDER BY sort_order ASC LIMIT %s OFFSET %s; """ % (_CUSTOMER_MODEL_COLUMNS, where_clause, page_size, offset) provider_rows = await sor.sqlExe(provider_sql, {}) model_type_rows = await sor.sqlExe(model_type_sql, {}) filter_total = (await sor.sqlExe(count_sql, {}))[0]['total_count'] model_list = await sor.sqlExe(find_sql, {}) return { 'status': True, 'msg': 'customer model search success', 'data': { 'provider_list': [r['provider'] for r in provider_rows], 'model_type_list': [r['model_type'] for r in model_type_rows], 'filter_total': filter_total, 'page_size': page_size, 'current_page': current_page, 'model_list': model_list, }, } except Exception as e: return {'status': False, 'msg': 'customer model search failed, %s' % str(e)} ret = await model_management_customer_search(params_kw) return ret