
import google.protobuf
from packaging.version import Version
if Version(google.protobuf.__version__).major >= 5:
  # -*- coding: utf-8 -*-
  # Generated by the protocol buffer compiler.  DO NOT EDIT!
  # source: prompt_optimization.proto
  # Protobuf Python Version: 5.26.0
  """Generated protocol buffer code."""
  from google.protobuf import descriptor as _descriptor
  from google.protobuf import descriptor_pool as _descriptor_pool
  from google.protobuf import symbol_database as _symbol_database
  from google.protobuf.internal import builder as _builder
  # @@protoc_insertion_point(imports)

  _sym_db = _symbol_database.Default()


  from . import jobs_pb2 as jobs__pb2
  from .scalapb import scalapb_pb2 as scalapb_dot_scalapb__pb2


  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x19prompt_optimization.proto\x12\x06mlflow\x1a\njobs.proto\x1a\x15scalapb/scalapb.proto\"6\n\x18PromptOptimizationJobTag\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"\x90\x01\n\x1bPromptOptimizationJobConfig\x12-\n\x0eoptimizer_type\x18\x01 \x01(\x0e\x32\x15.mlflow.OptimizerType\x12\x12\n\ndataset_id\x18\x02 \x01(\t\x12\x0f\n\x07scorers\x18\x03 \x03(\t\x12\x1d\n\x15optimizer_config_json\x18\x04 \x01(\t\"\xe1\x04\n\x15PromptOptimizationJob\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\x0e\n\x06run_id\x18\x02 \x01(\t\x12\x1f\n\x05state\x18\x03 \x01(\x0b\x32\x10.mlflow.JobState\x12\x15\n\rexperiment_id\x18\x04 \x01(\t\x12\x19\n\x11source_prompt_uri\x18\x05 \x01(\t\x12\x1c\n\x14optimized_prompt_uri\x18\x06 \x01(\t\x12\x33\n\x06\x63onfig\x18\x07 \x01(\x0b\x32#.mlflow.PromptOptimizationJobConfig\x12\x1d\n\x15\x63reation_timestamp_ms\x18\x08 \x01(\x03\x12\x1f\n\x17\x63ompletion_timestamp_ms\x18\t \x01(\x03\x12.\n\x04tags\x18\n \x03(\x0b\x32 .mlflow.PromptOptimizationJobTag\x12Q\n\x13initial_eval_scores\x18\x0b \x03(\x0b\x32\x34.mlflow.PromptOptimizationJob.InitialEvalScoresEntry\x12M\n\x11\x66inal_eval_scores\x18\x0c \x03(\x0b\x32\x32.mlflow.PromptOptimizationJob.FinalEvalScoresEntry\x1a\x38\n\x16InitialEvalScoresEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x1a\x36\n\x14\x46inalEvalScoresEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01*g\n\rOptimizerType\x12\x1e\n\x1aOPTIMIZER_TYPE_UNSPECIFIED\x10\x00\x12\x17\n\x13OPTIMIZER_TYPE_GEPA\x10\x01\x12\x1d\n\x19OPTIMIZER_TYPE_METAPROMPT\x10\x02\x42\x1e\n\x14org.mlflow.api.proto\x90\x01\x01\xe2?\x02\x10\x01')

  _globals = globals()
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
  _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'prompt_optimization_pb2', _globals)
  if not _descriptor._USE_C_DESCRIPTORS:
    _globals['DESCRIPTOR']._loaded_options = None
    _globals['DESCRIPTOR']._serialized_options = b'\n\024org.mlflow.api.proto\220\001\001\342?\002\020\001'
    _globals['_PROMPTOPTIMIZATIONJOB_INITIALEVALSCORESENTRY']._loaded_options = None
    _globals['_PROMPTOPTIMIZATIONJOB_INITIALEVALSCORESENTRY']._serialized_options = b'8\001'
    _globals['_PROMPTOPTIMIZATIONJOB_FINALEVALSCORESENTRY']._loaded_options = None
    _globals['_PROMPTOPTIMIZATIONJOB_FINALEVALSCORESENTRY']._serialized_options = b'8\001'
    _globals['_OPTIMIZERTYPE']._serialized_start=887
    _globals['_OPTIMIZERTYPE']._serialized_end=990
    _globals['_PROMPTOPTIMIZATIONJOBTAG']._serialized_start=72
    _globals['_PROMPTOPTIMIZATIONJOBTAG']._serialized_end=126
    _globals['_PROMPTOPTIMIZATIONJOBCONFIG']._serialized_start=129
    _globals['_PROMPTOPTIMIZATIONJOBCONFIG']._serialized_end=273
    _globals['_PROMPTOPTIMIZATIONJOB']._serialized_start=276
    _globals['_PROMPTOPTIMIZATIONJOB']._serialized_end=885
    _globals['_PROMPTOPTIMIZATIONJOB_INITIALEVALSCORESENTRY']._serialized_start=773
    _globals['_PROMPTOPTIMIZATIONJOB_INITIALEVALSCORESENTRY']._serialized_end=829
    _globals['_PROMPTOPTIMIZATIONJOB_FINALEVALSCORESENTRY']._serialized_start=831
    _globals['_PROMPTOPTIMIZATIONJOB_FINALEVALSCORESENTRY']._serialized_end=885
  # @@protoc_insertion_point(module_scope)

else:
  # -*- coding: utf-8 -*-
  # Generated by the protocol buffer compiler.  DO NOT EDIT!
  # source: prompt_optimization.proto
  """Generated protocol buffer code."""
  from google.protobuf.internal import enum_type_wrapper
  from google.protobuf import descriptor as _descriptor
  from google.protobuf import descriptor_pool as _descriptor_pool
  from google.protobuf import message as _message
  from google.protobuf import reflection as _reflection
  from google.protobuf import symbol_database as _symbol_database
  # @@protoc_insertion_point(imports)

  _sym_db = _symbol_database.Default()


  from . import jobs_pb2 as jobs__pb2
  from .scalapb import scalapb_pb2 as scalapb_dot_scalapb__pb2


  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x19prompt_optimization.proto\x12\x06mlflow\x1a\njobs.proto\x1a\x15scalapb/scalapb.proto\"6\n\x18PromptOptimizationJobTag\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"\x90\x01\n\x1bPromptOptimizationJobConfig\x12-\n\x0eoptimizer_type\x18\x01 \x01(\x0e\x32\x15.mlflow.OptimizerType\x12\x12\n\ndataset_id\x18\x02 \x01(\t\x12\x0f\n\x07scorers\x18\x03 \x03(\t\x12\x1d\n\x15optimizer_config_json\x18\x04 \x01(\t\"\xe1\x04\n\x15PromptOptimizationJob\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\x0e\n\x06run_id\x18\x02 \x01(\t\x12\x1f\n\x05state\x18\x03 \x01(\x0b\x32\x10.mlflow.JobState\x12\x15\n\rexperiment_id\x18\x04 \x01(\t\x12\x19\n\x11source_prompt_uri\x18\x05 \x01(\t\x12\x1c\n\x14optimized_prompt_uri\x18\x06 \x01(\t\x12\x33\n\x06\x63onfig\x18\x07 \x01(\x0b\x32#.mlflow.PromptOptimizationJobConfig\x12\x1d\n\x15\x63reation_timestamp_ms\x18\x08 \x01(\x03\x12\x1f\n\x17\x63ompletion_timestamp_ms\x18\t \x01(\x03\x12.\n\x04tags\x18\n \x03(\x0b\x32 .mlflow.PromptOptimizationJobTag\x12Q\n\x13initial_eval_scores\x18\x0b \x03(\x0b\x32\x34.mlflow.PromptOptimizationJob.InitialEvalScoresEntry\x12M\n\x11\x66inal_eval_scores\x18\x0c \x03(\x0b\x32\x32.mlflow.PromptOptimizationJob.FinalEvalScoresEntry\x1a\x38\n\x16InitialEvalScoresEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x1a\x36\n\x14\x46inalEvalScoresEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01*g\n\rOptimizerType\x12\x1e\n\x1aOPTIMIZER_TYPE_UNSPECIFIED\x10\x00\x12\x17\n\x13OPTIMIZER_TYPE_GEPA\x10\x01\x12\x1d\n\x19OPTIMIZER_TYPE_METAPROMPT\x10\x02\x42\x1e\n\x14org.mlflow.api.proto\x90\x01\x01\xe2?\x02\x10\x01')

  _OPTIMIZERTYPE = DESCRIPTOR.enum_types_by_name['OptimizerType']
  OptimizerType = enum_type_wrapper.EnumTypeWrapper(_OPTIMIZERTYPE)
  OPTIMIZER_TYPE_UNSPECIFIED = 0
  OPTIMIZER_TYPE_GEPA = 1
  OPTIMIZER_TYPE_METAPROMPT = 2


  _PROMPTOPTIMIZATIONJOBTAG = DESCRIPTOR.message_types_by_name['PromptOptimizationJobTag']
  _PROMPTOPTIMIZATIONJOBCONFIG = DESCRIPTOR.message_types_by_name['PromptOptimizationJobConfig']
  _PROMPTOPTIMIZATIONJOB = DESCRIPTOR.message_types_by_name['PromptOptimizationJob']
  _PROMPTOPTIMIZATIONJOB_INITIALEVALSCORESENTRY = _PROMPTOPTIMIZATIONJOB.nested_types_by_name['InitialEvalScoresEntry']
  _PROMPTOPTIMIZATIONJOB_FINALEVALSCORESENTRY = _PROMPTOPTIMIZATIONJOB.nested_types_by_name['FinalEvalScoresEntry']
  PromptOptimizationJobTag = _reflection.GeneratedProtocolMessageType('PromptOptimizationJobTag', (_message.Message,), {
    'DESCRIPTOR' : _PROMPTOPTIMIZATIONJOBTAG,
    '__module__' : 'prompt_optimization_pb2'
    # @@protoc_insertion_point(class_scope:mlflow.PromptOptimizationJobTag)
    })
  _sym_db.RegisterMessage(PromptOptimizationJobTag)

  PromptOptimizationJobConfig = _reflection.GeneratedProtocolMessageType('PromptOptimizationJobConfig', (_message.Message,), {
    'DESCRIPTOR' : _PROMPTOPTIMIZATIONJOBCONFIG,
    '__module__' : 'prompt_optimization_pb2'
    # @@protoc_insertion_point(class_scope:mlflow.PromptOptimizationJobConfig)
    })
  _sym_db.RegisterMessage(PromptOptimizationJobConfig)

  PromptOptimizationJob = _reflection.GeneratedProtocolMessageType('PromptOptimizationJob', (_message.Message,), {

    'InitialEvalScoresEntry' : _reflection.GeneratedProtocolMessageType('InitialEvalScoresEntry', (_message.Message,), {
      'DESCRIPTOR' : _PROMPTOPTIMIZATIONJOB_INITIALEVALSCORESENTRY,
      '__module__' : 'prompt_optimization_pb2'
      # @@protoc_insertion_point(class_scope:mlflow.PromptOptimizationJob.InitialEvalScoresEntry)
      })
    ,

    'FinalEvalScoresEntry' : _reflection.GeneratedProtocolMessageType('FinalEvalScoresEntry', (_message.Message,), {
      'DESCRIPTOR' : _PROMPTOPTIMIZATIONJOB_FINALEVALSCORESENTRY,
      '__module__' : 'prompt_optimization_pb2'
      # @@protoc_insertion_point(class_scope:mlflow.PromptOptimizationJob.FinalEvalScoresEntry)
      })
    ,
    'DESCRIPTOR' : _PROMPTOPTIMIZATIONJOB,
    '__module__' : 'prompt_optimization_pb2'
    # @@protoc_insertion_point(class_scope:mlflow.PromptOptimizationJob)
    })
  _sym_db.RegisterMessage(PromptOptimizationJob)
  _sym_db.RegisterMessage(PromptOptimizationJob.InitialEvalScoresEntry)
  _sym_db.RegisterMessage(PromptOptimizationJob.FinalEvalScoresEntry)

  if _descriptor._USE_C_DESCRIPTORS == False:

    DESCRIPTOR._options = None
    DESCRIPTOR._serialized_options = b'\n\024org.mlflow.api.proto\220\001\001\342?\002\020\001'
    _PROMPTOPTIMIZATIONJOB_INITIALEVALSCORESENTRY._options = None
    _PROMPTOPTIMIZATIONJOB_INITIALEVALSCORESENTRY._serialized_options = b'8\001'
    _PROMPTOPTIMIZATIONJOB_FINALEVALSCORESENTRY._options = None
    _PROMPTOPTIMIZATIONJOB_FINALEVALSCORESENTRY._serialized_options = b'8\001'
    _OPTIMIZERTYPE._serialized_start=887
    _OPTIMIZERTYPE._serialized_end=990
    _PROMPTOPTIMIZATIONJOBTAG._serialized_start=72
    _PROMPTOPTIMIZATIONJOBTAG._serialized_end=126
    _PROMPTOPTIMIZATIONJOBCONFIG._serialized_start=129
    _PROMPTOPTIMIZATIONJOBCONFIG._serialized_end=273
    _PROMPTOPTIMIZATIONJOB._serialized_start=276
    _PROMPTOPTIMIZATIONJOB._serialized_end=885
    _PROMPTOPTIMIZATIONJOB_INITIALEVALSCORESENTRY._serialized_start=773
    _PROMPTOPTIMIZATIONJOB_INITIALEVALSCORESENTRY._serialized_end=829
    _PROMPTOPTIMIZATIONJOB_FINALEVALSCORESENTRY._serialized_start=831
    _PROMPTOPTIMIZATIONJOB_FINALEVALSCORESENTRY._serialized_end=885
  # @@protoc_insertion_point(module_scope)

