MCPcopy Index your code
hub / github.com/dataease/SQLBot / save_sql_data

Method save_sql_data

backend/apps/chat/task/llm.py:1101–1116  ·  view source on GitHub ↗
(self, session: Session, data_obj: Dict[str, Any])

Source from the content-addressed store, hash-verified

1099 return save_error_message(session=session, record_id=self.record.id, message=message)
1100
1101 def save_sql_data(self, session: Session, data_obj: Dict[str, Any]):
1102 try:
1103 data_result = data_obj.get('data')
1104 limit = 1000
1105 if data_result:
1106 data_result = prepare_for_orjson(data_result)
1107 if data_result and len(data_result) > limit and self.enable_sql_row_limit:
1108 data_obj['data'] = data_result[:limit]
1109 data_obj['limit'] = limit
1110 else:
1111 data_obj['data'] = data_result
1112 data_obj['datasource'] = self.ds.id
1113 return save_sql_exec_data(session=session, record_id=self.record.id,
1114 data=orjson.dumps(data_obj).decode())
1115 except Exception as e:
1116 raise e
1117
1118 def finish(self, session: Session):
1119 return finish_record(session=session, record_id=self.record.id)

Callers 1

run_taskMethod · 0.95

Calls 3

prepare_for_orjsonFunction · 0.90
save_sql_exec_dataFunction · 0.90
getMethod · 0.65

Tested by

no test coverage detected