googleapis_proto_ex

defmodule Google.Ai.Generativelanguage.V1alpha.TunedModel.State do
  @moduledoc false

  use Protobuf, enum: true, protoc_gen_elixir_version: "0.15.0", syntax: :proto3

  field(:STATE_UNSPECIFIED, 0)
  field(:CREATING, 1)
  field(:ACTIVE, 2)
  field(:FAILED, 3)
end

defmodule Google.Ai.Generativelanguage.V1alpha.TunedModel do
  @moduledoc false

  use Protobuf, protoc_gen_elixir_version: "0.15.0", syntax: :proto3

  oneof(:source_model, 0)

  field(:tuned_model_source, 3,
    type: Google.Ai.Generativelanguage.V1alpha.TunedModelSource,
    json_name: "tunedModelSource",
    oneof: 0,
    deprecated: false
  )

  field(:base_model, 4, type: :string, json_name: "baseModel", oneof: 0, deprecated: false)
  field(:name, 1, type: :string, deprecated: false)
  field(:display_name, 5, type: :string, json_name: "displayName", deprecated: false)
  field(:description, 6, type: :string, deprecated: false)
  field(:temperature, 11, proto3_optional: true, type: :float, deprecated: false)
  field(:top_p, 12, proto3_optional: true, type: :float, json_name: "topP", deprecated: false)
  field(:top_k, 13, proto3_optional: true, type: :int32, json_name: "topK", deprecated: false)

  field(:state, 7,
    type: Google.Ai.Generativelanguage.V1alpha.TunedModel.State,
    enum: true,
    deprecated: false
  )

  field(:create_time, 8,
    type: Google.Protobuf.Timestamp,
    json_name: "createTime",
    deprecated: false
  )

  field(:update_time, 9,
    type: Google.Protobuf.Timestamp,
    json_name: "updateTime",
    deprecated: false
  )

  field(:tuning_task, 10,
    type: Google.Ai.Generativelanguage.V1alpha.TuningTask,
    json_name: "tuningTask",
    deprecated: false
  )

  field(:reader_project_numbers, 14,
    repeated: true,
    type: :int64,
    json_name: "readerProjectNumbers",
    deprecated: false
  )
end

defmodule Google.Ai.Generativelanguage.V1alpha.TunedModelSource do
  @moduledoc false

  use Protobuf, protoc_gen_elixir_version: "0.15.0", syntax: :proto3

  field(:tuned_model, 1, type: :string, json_name: "tunedModel", deprecated: false)
  field(:base_model, 2, type: :string, json_name: "baseModel", deprecated: false)
end

defmodule Google.Ai.Generativelanguage.V1alpha.TuningTask do
  @moduledoc false

  use Protobuf, protoc_gen_elixir_version: "0.15.0", syntax: :proto3

  field(:start_time, 1,
    type: Google.Protobuf.Timestamp,
    json_name: "startTime",
    deprecated: false
  )

  field(:complete_time, 2,
    type: Google.Protobuf.Timestamp,
    json_name: "completeTime",
    deprecated: false
  )

  field(:snapshots, 3,
    repeated: true,
    type: Google.Ai.Generativelanguage.V1alpha.TuningSnapshot,
    deprecated: false
  )

  field(:training_data, 4,
    type: Google.Ai.Generativelanguage.V1alpha.Dataset,
    json_name: "trainingData",
    deprecated: false
  )

  field(:hyperparameters, 5,
    type: Google.Ai.Generativelanguage.V1alpha.Hyperparameters,
    deprecated: false
  )
end

defmodule Google.Ai.Generativelanguage.V1alpha.Hyperparameters do
  @moduledoc false

  use Protobuf, protoc_gen_elixir_version: "0.15.0", syntax: :proto3

  oneof(:learning_rate_option, 0)

  field(:learning_rate, 16, type: :float, json_name: "learningRate", oneof: 0, deprecated: false)

  field(:learning_rate_multiplier, 17,
    type: :float,
    json_name: "learningRateMultiplier",
    oneof: 0,
    deprecated: false
  )

  field(:epoch_count, 14,
    proto3_optional: true,
    type: :int32,
    json_name: "epochCount",
    deprecated: false
  )

  field(:batch_size, 15,
    proto3_optional: true,
    type: :int32,
    json_name: "batchSize",
    deprecated: false
  )
end

defmodule Google.Ai.Generativelanguage.V1alpha.Dataset do
  @moduledoc false

  use Protobuf, protoc_gen_elixir_version: "0.15.0", syntax: :proto3

  oneof(:dataset, 0)

  field(:examples, 1,
    type: Google.Ai.Generativelanguage.V1alpha.TuningExamples,
    oneof: 0,
    deprecated: false
  )
end

defmodule Google.Ai.Generativelanguage.V1alpha.TuningExamples do
  @moduledoc false

  use Protobuf, protoc_gen_elixir_version: "0.15.0", syntax: :proto3

  field(:examples, 1, repeated: true, type: Google.Ai.Generativelanguage.V1alpha.TuningExample)

  field(:multiturn_examples, 2,
    repeated: true,
    type: Google.Ai.Generativelanguage.V1alpha.TuningMultiturnExample,
    json_name: "multiturnExamples"
  )
end

defmodule Google.Ai.Generativelanguage.V1alpha.TuningPart do
  @moduledoc false

  use Protobuf, protoc_gen_elixir_version: "0.15.0", syntax: :proto3

  oneof(:data, 0)

  field(:text, 2, type: :string, oneof: 0)
end

defmodule Google.Ai.Generativelanguage.V1alpha.TuningContent do
  @moduledoc false

  use Protobuf, protoc_gen_elixir_version: "0.15.0", syntax: :proto3

  field(:parts, 1, repeated: true, type: Google.Ai.Generativelanguage.V1alpha.TuningPart)
  field(:role, 2, type: :string, deprecated: false)
end

defmodule Google.Ai.Generativelanguage.V1alpha.TuningMultiturnExample do
  @moduledoc false

  use Protobuf, protoc_gen_elixir_version: "0.15.0", syntax: :proto3

  field(:system_instruction, 8,
    proto3_optional: true,
    type: Google.Ai.Generativelanguage.V1alpha.TuningContent,
    json_name: "systemInstruction",
    deprecated: false
  )

  field(:contents, 1, repeated: true, type: Google.Ai.Generativelanguage.V1alpha.TuningContent)
end

defmodule Google.Ai.Generativelanguage.V1alpha.TuningExample do
  @moduledoc false

  use Protobuf, protoc_gen_elixir_version: "0.15.0", syntax: :proto3

  oneof(:model_input, 0)

  field(:text_input, 1, type: :string, json_name: "textInput", oneof: 0, deprecated: false)
  field(:output, 3, type: :string, deprecated: false)
end

defmodule Google.Ai.Generativelanguage.V1alpha.TuningSnapshot do
  @moduledoc false

  use Protobuf, protoc_gen_elixir_version: "0.15.0", syntax: :proto3

  field(:step, 1, type: :int32, deprecated: false)
  field(:epoch, 2, type: :int32, deprecated: false)
  field(:mean_loss, 3, type: :float, json_name: "meanLoss", deprecated: false)

  field(:compute_time, 4,
    type: Google.Protobuf.Timestamp,
    json_name: "computeTime",
    deprecated: false
  )
end