yandex.cloud.api.ai.llm.v1alpha.TextGenerationServiceGrpc Maven / Gradle / Ivy

Go to download
package yandex.cloud.api.ai.llm.v1alpha;

import static io.grpc.MethodDescriptor.generateFullMethodName;

/**
 */
@javax.annotation.Generated(
    value = "by gRPC proto compiler (version 1.43.2)",
    comments = "Source: yandex/cloud/ai/llm/v1alpha/llm_service.proto")
@io.grpc.stub.annotations.GrpcGenerated
public final class TextGenerationServiceGrpc {

  private TextGenerationServiceGrpc() {}

  public static final String SERVICE_NAME = "yandex.cloud.ai.llm.v1alpha.TextGenerationService";

  // Static method descriptors that strictly reflect the proto.
  private static volatile io.grpc.MethodDescriptor getInstructMethod;

  @io.grpc.stub.annotations.RpcMethod(
      fullMethodName = SERVICE_NAME + '/' + "Instruct",
      requestType = yandex.cloud.api.ai.llm.v1alpha.LlmService.InstructRequest.class,
      responseType = yandex.cloud.api.ai.llm.v1alpha.LlmService.InstructResponse.class,
      methodType = io.grpc.MethodDescriptor.MethodType.SERVER_STREAMING)
  public static io.grpc.MethodDescriptor getInstructMethod() {
    io.grpc.MethodDescriptor getInstructMethod;
    if ((getInstructMethod = TextGenerationServiceGrpc.getInstructMethod) == null) {
      synchronized (TextGenerationServiceGrpc.class) {
        if ((getInstructMethod = TextGenerationServiceGrpc.getInstructMethod) == null) {
          TextGenerationServiceGrpc.getInstructMethod = getInstructMethod =
              io.grpc.MethodDescriptor.newBuilder()
              .setType(io.grpc.MethodDescriptor.MethodType.SERVER_STREAMING)
              .setFullMethodName(generateFullMethodName(SERVICE_NAME, "Instruct"))
              .setSampledToLocalTracing(true)
              .setRequestMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
                  yandex.cloud.api.ai.llm.v1alpha.LlmService.InstructRequest.getDefaultInstance()))
              .setResponseMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
                  yandex.cloud.api.ai.llm.v1alpha.LlmService.InstructResponse.getDefaultInstance()))
              .setSchemaDescriptor(new TextGenerationServiceMethodDescriptorSupplier("Instruct"))
              .build();
        }
      }
    }
    return getInstructMethod;
  }

  private static volatile io.grpc.MethodDescriptor getChatMethod;

  @io.grpc.stub.annotations.RpcMethod(
      fullMethodName = SERVICE_NAME + '/' + "Chat",
      requestType = yandex.cloud.api.ai.llm.v1alpha.LlmService.ChatRequest.class,
      responseType = yandex.cloud.api.ai.llm.v1alpha.LlmService.ChatResponse.class,
      methodType = io.grpc.MethodDescriptor.MethodType.SERVER_STREAMING)
  public static io.grpc.MethodDescriptor getChatMethod() {
    io.grpc.MethodDescriptor getChatMethod;
    if ((getChatMethod = TextGenerationServiceGrpc.getChatMethod) == null) {
      synchronized (TextGenerationServiceGrpc.class) {
        if ((getChatMethod = TextGenerationServiceGrpc.getChatMethod) == null) {
          TextGenerationServiceGrpc.getChatMethod = getChatMethod =
              io.grpc.MethodDescriptor.newBuilder()
              .setType(io.grpc.MethodDescriptor.MethodType.SERVER_STREAMING)
              .setFullMethodName(generateFullMethodName(SERVICE_NAME, "Chat"))
              .setSampledToLocalTracing(true)
              .setRequestMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
                  yandex.cloud.api.ai.llm.v1alpha.LlmService.ChatRequest.getDefaultInstance()))
              .setResponseMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
                  yandex.cloud.api.ai.llm.v1alpha.LlmService.ChatResponse.getDefaultInstance()))
              .setSchemaDescriptor(new TextGenerationServiceMethodDescriptorSupplier("Chat"))
              .build();
        }
      }
    }
    return getChatMethod;
  }

  /**
   * Creates a new async stub that supports all call types for the service
   */
  public static TextGenerationServiceStub newStub(io.grpc.Channel channel) {
    io.grpc.stub.AbstractStub.StubFactory factory =
      new io.grpc.stub.AbstractStub.StubFactory() {
        @java.lang.Override
        public TextGenerationServiceStub newStub(io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
          return new TextGenerationServiceStub(channel, callOptions);
        }
      };
    return TextGenerationServiceStub.newStub(factory, channel);
  }

  /**
   * Creates a new blocking-style stub that supports unary and streaming output calls on the service
   */
  public static TextGenerationServiceBlockingStub newBlockingStub(
      io.grpc.Channel channel) {
    io.grpc.stub.AbstractStub.StubFactory factory =
      new io.grpc.stub.AbstractStub.StubFactory() {
        @java.lang.Override
        public TextGenerationServiceBlockingStub newStub(io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
          return new TextGenerationServiceBlockingStub(channel, callOptions);
        }
      };
    return TextGenerationServiceBlockingStub.newStub(factory, channel);
  }

  /**
   * Creates a new ListenableFuture-style stub that supports unary calls on the service
   */
  public static TextGenerationServiceFutureStub newFutureStub(
      io.grpc.Channel channel) {
    io.grpc.stub.AbstractStub.StubFactory factory =
      new io.grpc.stub.AbstractStub.StubFactory() {
        @java.lang.Override
        public TextGenerationServiceFutureStub newStub(io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
          return new TextGenerationServiceFutureStub(channel, callOptions);
        }
      };
    return TextGenerationServiceFutureStub.newStub(factory, channel);
  }

  /**
   */
  public static abstract class TextGenerationServiceImplBase implements io.grpc.BindableService {

    /**
     */
    public void instruct(yandex.cloud.api.ai.llm.v1alpha.LlmService.InstructRequest request,
        io.grpc.stub.StreamObserver responseObserver) {
      io.grpc.stub.ServerCalls.asyncUnimplementedUnaryCall(getInstructMethod(), responseObserver);
    }

    /**
     */
    public void chat(yandex.cloud.api.ai.llm.v1alpha.LlmService.ChatRequest request,
        io.grpc.stub.StreamObserver responseObserver) {
      io.grpc.stub.ServerCalls.asyncUnimplementedUnaryCall(getChatMethod(), responseObserver);
    }

    @java.lang.Override public final io.grpc.ServerServiceDefinition bindService() {
      return io.grpc.ServerServiceDefinition.builder(getServiceDescriptor())
          .addMethod(
            getInstructMethod(),
            io.grpc.stub.ServerCalls.asyncServerStreamingCall(
              new MethodHandlers<
                yandex.cloud.api.ai.llm.v1alpha.LlmService.InstructRequest,
                yandex.cloud.api.ai.llm.v1alpha.LlmService.InstructResponse>(
                  this, METHODID_INSTRUCT)))
          .addMethod(
            getChatMethod(),
            io.grpc.stub.ServerCalls.asyncServerStreamingCall(
              new MethodHandlers<
                yandex.cloud.api.ai.llm.v1alpha.LlmService.ChatRequest,
                yandex.cloud.api.ai.llm.v1alpha.LlmService.ChatResponse>(
                  this, METHODID_CHAT)))
          .build();
    }
  }

  /**
   */
  public static final class TextGenerationServiceStub extends io.grpc.stub.AbstractAsyncStub {
    private TextGenerationServiceStub(
        io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
      super(channel, callOptions);
    }

    @java.lang.Override
    protected TextGenerationServiceStub build(
        io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
      return new TextGenerationServiceStub(channel, callOptions);
    }

    /**
     */
    public void instruct(yandex.cloud.api.ai.llm.v1alpha.LlmService.InstructRequest request,
        io.grpc.stub.StreamObserver responseObserver) {
      io.grpc.stub.ClientCalls.asyncServerStreamingCall(
          getChannel().newCall(getInstructMethod(), getCallOptions()), request, responseObserver);
    }

    /**
     */
    public void chat(yandex.cloud.api.ai.llm.v1alpha.LlmService.ChatRequest request,
        io.grpc.stub.StreamObserver responseObserver) {
      io.grpc.stub.ClientCalls.asyncServerStreamingCall(
          getChannel().newCall(getChatMethod(), getCallOptions()), request, responseObserver);
    }
  }

  /**
   */
  public static final class TextGenerationServiceBlockingStub extends io.grpc.stub.AbstractBlockingStub {
    private TextGenerationServiceBlockingStub(
        io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
      super(channel, callOptions);
    }

    @java.lang.Override
    protected TextGenerationServiceBlockingStub build(
        io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
      return new TextGenerationServiceBlockingStub(channel, callOptions);
    }

    /**
     */
    public java.util.Iterator instruct(
        yandex.cloud.api.ai.llm.v1alpha.LlmService.InstructRequest request) {
      return io.grpc.stub.ClientCalls.blockingServerStreamingCall(
          getChannel(), getInstructMethod(), getCallOptions(), request);
    }

    /**
     */
    public java.util.Iterator chat(
        yandex.cloud.api.ai.llm.v1alpha.LlmService.ChatRequest request) {
      return io.grpc.stub.ClientCalls.blockingServerStreamingCall(
          getChannel(), getChatMethod(), getCallOptions(), request);
    }
  }

  /**
   */
  public static final class TextGenerationServiceFutureStub extends io.grpc.stub.AbstractFutureStub {
    private TextGenerationServiceFutureStub(
        io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
      super(channel, callOptions);
    }

    @java.lang.Override
    protected TextGenerationServiceFutureStub build(
        io.grpc.Channel channel, io.grpc.CallOptions callOptions) {
      return new TextGenerationServiceFutureStub(channel, callOptions);
    }
  }

  private static final int METHODID_INSTRUCT = 0;
  private static final int METHODID_CHAT = 1;

  private static final class MethodHandlers implements
      io.grpc.stub.ServerCalls.UnaryMethod,
      io.grpc.stub.ServerCalls.ServerStreamingMethod,
      io.grpc.stub.ServerCalls.ClientStreamingMethod,
      io.grpc.stub.ServerCalls.BidiStreamingMethod {
    private final TextGenerationServiceImplBase serviceImpl;
    private final int methodId;

    MethodHandlers(TextGenerationServiceImplBase serviceImpl, int methodId) {
      this.serviceImpl = serviceImpl;
      this.methodId = methodId;
    }

    @java.lang.Override
    @java.lang.SuppressWarnings("unchecked")
    public void invoke(Req request, io.grpc.stub.StreamObserver responseObserver) {
      switch (methodId) {
        case METHODID_INSTRUCT:
          serviceImpl.instruct((yandex.cloud.api.ai.llm.v1alpha.LlmService.InstructRequest) request,
              (io.grpc.stub.StreamObserver) responseObserver);
          break;
        case METHODID_CHAT:
          serviceImpl.chat((yandex.cloud.api.ai.llm.v1alpha.LlmService.ChatRequest) request,
              (io.grpc.stub.StreamObserver) responseObserver);
          break;
        default:
          throw new AssertionError();
      }
    }

    @java.lang.Override
    @java.lang.SuppressWarnings("unchecked")
    public io.grpc.stub.StreamObserver invoke(
        io.grpc.stub.StreamObserver responseObserver) {
      switch (methodId) {
        default:
          throw new AssertionError();
      }
    }
  }

  private static abstract class TextGenerationServiceBaseDescriptorSupplier
      implements io.grpc.protobuf.ProtoFileDescriptorSupplier, io.grpc.protobuf.ProtoServiceDescriptorSupplier {
    TextGenerationServiceBaseDescriptorSupplier() {}

    @java.lang.Override
    public com.google.protobuf.Descriptors.FileDescriptor getFileDescriptor() {
      return yandex.cloud.api.ai.llm.v1alpha.LlmService.getDescriptor();
    }

    @java.lang.Override
    public com.google.protobuf.Descriptors.ServiceDescriptor getServiceDescriptor() {
      return getFileDescriptor().findServiceByName("TextGenerationService");
    }
  }

  private static final class TextGenerationServiceFileDescriptorSupplier
      extends TextGenerationServiceBaseDescriptorSupplier {
    TextGenerationServiceFileDescriptorSupplier() {}
  }

  private static final class TextGenerationServiceMethodDescriptorSupplier
      extends TextGenerationServiceBaseDescriptorSupplier
      implements io.grpc.protobuf.ProtoMethodDescriptorSupplier {
    private final String methodName;

    TextGenerationServiceMethodDescriptorSupplier(String methodName) {
      this.methodName = methodName;
    }

    @java.lang.Override
    public com.google.protobuf.Descriptors.MethodDescriptor getMethodDescriptor() {
      return getServiceDescriptor().findMethodByName(methodName);
    }
  }

  private static volatile io.grpc.ServiceDescriptor serviceDescriptor;

  public static io.grpc.ServiceDescriptor getServiceDescriptor() {
    io.grpc.ServiceDescriptor result = serviceDescriptor;
    if (result == null) {
      synchronized (TextGenerationServiceGrpc.class) {
        result = serviceDescriptor;
        if (result == null) {
          serviceDescriptor = result = io.grpc.ServiceDescriptor.newBuilder(SERVICE_NAME)
              .setSchemaDescriptor(new TextGenerationServiceFileDescriptorSupplier())
              .addMethod(getInstructMethod())
              .addMethod(getChatMethod())
              .build();
        }
      }
    }
    return result;
  }
}