def init_from_proto()

in evalbench/dataset/evalinput.py [0:0]

• 20 lines of code
• 13 McCabe index (conditional complexity)

        def init_from_proto(self, proto: eval_request_pb2.EvalInputRequest):  # type: ignore
            """Initializes an EvalInputRequest from eval_request_pb2 proto."""

            request = MessageToDict(proto)
            return self(
                id=str(request.get("id") or -1),
                query_type=request.get("queryType") or "",
                database=request.get("database") or "",
                nl_prompt=request.get("nlPrompt") or "",
                dialects=request.get("dialects") or [],
                golden_sql=_get_dialect_based_sql(request.get("goldenSql")),
                eval_query=_get_dialect_based_sql(request.get("evalQuery")),
                setup_sql=_get_dialect_based_sql(request.get("setupSql")),
                cleanup_sql=_get_dialect_based_sql(request.get("cleanupSql")),
                tags=request.get("tags") or [],
                other=request.get("other") or {},
                sql_generator_error=request.get("sqlGeneratorError") or "",
                sql_generator_time=request.get("sqlGeneratorTime") or 0,
                generated_sql=request.get("generatedSql") or "",
                job_id=request.get("jobId") or "",
                trace_id=request.get("traceId") or "",
            )