增加模型信息的记录入库字段
This commit is contained in:
@ -38,6 +38,13 @@ def _prepare_table_schema(value: Any) -> str:
|
||||
return json.dumps(value, ensure_ascii=False)
|
||||
|
||||
|
||||
def _prepare_model_params(params: Dict[str, Any] | None) -> str | None:
|
||||
if not params:
|
||||
return None
|
||||
serialized, _ = _serialize_json(params)
|
||||
return serialized
|
||||
|
||||
|
||||
def _collect_common_columns(request: TableSnippetUpsertRequest) -> Dict[str, Any]:
|
||||
logger.debug(
|
||||
"Collecting common columns for table_id=%s version_ts=%s action_type=%s",
|
||||
@ -53,6 +60,8 @@ def _collect_common_columns(request: TableSnippetUpsertRequest) -> Dict[str, Any
|
||||
"callback_url": str(request.callback_url),
|
||||
"table_schema_version_id": request.table_schema_version_id,
|
||||
"table_schema": _prepare_table_schema(request.table_schema),
|
||||
"model": request.model,
|
||||
"model_provider": request.model_provider,
|
||||
}
|
||||
|
||||
payload.update(
|
||||
@ -72,6 +81,8 @@ def _collect_common_columns(request: TableSnippetUpsertRequest) -> Dict[str, Any
|
||||
}
|
||||
)
|
||||
|
||||
payload["model_params"] = _prepare_model_params(request.model_params)
|
||||
|
||||
if request.llm_usage is not None:
|
||||
llm_usage_json, _ = _serialize_json(request.llm_usage)
|
||||
if llm_usage_json is not None:
|
||||
|
||||
Reference in New Issue
Block a user