Foundation Models Text Generation API, gRPC: TextGenerationAsyncService.Completion

Статья создана

Yandex Cloud

Обновлена 26 ноября 2024 г.

gRPC request
CompletionRequest
CompletionOptions
Message
ToolCallList
ToolCall
FunctionCall
ToolResultList
ToolResult
FunctionResult
Tool
FunctionTool
operation.Operation
CompletionResponse
Alternative
Message
ToolCallList
ToolCall
FunctionCall
ToolResultList
ToolResult
FunctionResult
ContentUsage

A method for generating text completions in asynchronous mode.

gRPC request

rpc Completion (CompletionRequest) returns (operation.Operation)

CompletionRequest

{
  "model_uri": "string",
  "completion_options": {
    "stream": "bool",
    "temperature": "google.protobuf.DoubleValue",
    "max_tokens": "google.protobuf.Int64Value"
  },
  "messages": [
    {
      "role": "string",
      // Includes only one of the fields `text`, `tool_call_list`, `tool_result_list`
      "text": "string",
      "tool_call_list": {
        "tool_calls": [
          {
            // Includes only one of the fields `function_call`
            "function_call": {
              "name": "string",
              "arguments": "google.protobuf.Struct"
            }
            // end of the list of possible fields
          }
        ]
      },
      "tool_result_list": {
        "tool_results": [
          {
            // Includes only one of the fields `function_result`
            "function_result": {
              "name": "string",
              // Includes only one of the fields `content`
              "content": "string"
              // end of the list of possible fields
            }
            // end of the list of possible fields
          }
        ]
      }
      // end of the list of possible fields
    }
  ],
  "tools": [
    {
      // Includes only one of the fields `function`
      "function": {
        "name": "string",
        "description": "string",
        "parameters": "google.protobuf.Struct"
      }
      // end of the list of possible fields
    }
  ]
}

Request for the service to generate text completion.

Field	Description
model_uri	string The ID of the model to be used for completion generation.
completion_options	CompletionOptions Configuration options for completion generation.
messages[]	Message A list of messages representing the context for the completion model.
tools[]	Tool List of tools that are available for the model to invoke during the completion generation. Note: This parameter is not yet supported and will be ignored if provided.

CompletionOptions

Defines the options for completion generation.

Field	Description
stream	bool Enables streaming of partially generated text.
temperature	google.protobuf.DoubleValue Affects creativity and randomness of responses. Should be a double number between 0 (inclusive) and 1 (inclusive). Lower values produce more straightforward responses while higher values lead to increased creativity and randomness. Default temperature: 0.3
max_tokens	google.protobuf.Int64Value The limit on the number of tokens used for single completion generation. Must be greater than zero. This maximum allowed parameter value may depend on the model being used.

Message

A message object representing a wrapper over the inputs and outputs of the completion model.

Field	Description
role	string The ID of the message sender. Supported roles: `system`: Special role used to define the behaviour of the completion model. `assistant`: A role used by the model to generate responses. `user`: A role used by the user to describe requests to the model.
text	string Textual content of the message. Includes only one of the fields `text`, `tool_call_list`, `tool_result_list`. Message content.
tool_call_list	ToolCallList List of tool calls made by the model as part of the response generation. Includes only one of the fields `text`, `tool_call_list`, `tool_result_list`. Message content.
tool_result_list	ToolResultList List of tool results returned from external tools that were invoked by the model. Includes only one of the fields `text`, `tool_call_list`, `tool_result_list`. Message content.

ToolCallList

Represents a list of tool calls.

Field

Description

tool_calls[]

ToolCall

A list of tool calls to be executed.

ToolCall

Represents a call to a tool.

Field

Description

function_call

FunctionCall

Represents a call to a function.

Includes only one of the fields function_call.

FunctionCall

Represents the invocation of a function with specific arguments.

Field

Description

name

string

The name of the function being called.

arguments

google.protobuf.Struct

The structured arguments passed to the function.
These arguments must adhere to the JSON Schema defined in the corresponding function's parameters.

ToolResultList

Represents a list of tool results.

Field

Description

tool_results[]

ToolResult

A list of tool results.

ToolResult

Represents the result of a tool call.

Field

Description

function_result

FunctionResult

Represents the result of a function call.

Includes only one of the fields function_result.

FunctionResult

Represents the result of a function call.

Field

Description

name

string

The name of the function that was executed.

content

string

The result of the function call, represented as a string.
This field can be used to store the output of the function.

Includes only one of the fields content.

Tool

Represents a tool that can be invoked during completion generation.

Field

Description

function

FunctionTool

Represents a function that can be called.

Includes only one of the fields function.

FunctionTool

Represents a function tool that can be invoked during completion generation.

Field	Description
name	string The name of the function.
description	string A description of the function's purpose or behavior.
parameters	google.protobuf.Struct A JSON Schema that defines the expected parameters for the function. The schema should describe the required fields, their types, and any constraints or default values.

operation.Operation

{
  "id": "string",
  "description": "string",
  "created_at": "google.protobuf.Timestamp",
  "created_by": "string",
  "modified_at": "google.protobuf.Timestamp",
  "done": "bool",
  "metadata": "google.protobuf.Any",
  // Includes only one of the fields `error`, `response`
  "error": "google.rpc.Status",
  "response": {
    "alternatives": [
      {
        "message": {
          "role": "string",
          // Includes only one of the fields `text`, `tool_call_list`, `tool_result_list`
          "text": "string",
          "tool_call_list": {
            "tool_calls": [
              {
                // Includes only one of the fields `function_call`
                "function_call": {
                  "name": "string",
                  "arguments": "google.protobuf.Struct"
                }
                // end of the list of possible fields
              }
            ]
          },
          "tool_result_list": {
            "tool_results": [
              {
                // Includes only one of the fields `function_result`
                "function_result": {
                  "name": "string",
                  // Includes only one of the fields `content`
                  "content": "string"
                  // end of the list of possible fields
                }
                // end of the list of possible fields
              }
            ]
          }
          // end of the list of possible fields
        },
        "status": "AlternativeStatus"
      }
    ],
    "usage": {
      "input_text_tokens": "int64",
      "completion_tokens": "int64",
      "total_tokens": "int64"
    },
    "model_version": "string"
  }
  // end of the list of possible fields
}

An Operation resource. For more information, see Operation.

Field	Description
id	string ID of the operation.
description	string Description of the operation. 0-256 characters long.
created_at	google.protobuf.Timestamp Creation timestamp.
created_by	string ID of the user or service account who initiated the operation.
modified_at	google.protobuf.Timestamp The time when the Operation resource was last modified.
done	bool If the value is `false`, it means the operation is still in progress. If `true`, the operation is completed, and either `error` or `response` is available.
metadata	google.protobuf.Any Service-specific metadata associated with the operation. It typically contains the ID of the target resource that the operation is performed on. Any method that returns a long-running operation should document the metadata type, if any.
error	google.rpc.Status The error result of the operation in case of failure or cancellation. Includes only one of the fields `error`, `response`. The operation result. If `done == false` and there was no failure detected, neither `error` nor `response` is set. If `done == false` and there was a failure detected, `error` is set. If `done == true`, exactly one of `error` or `response` is set.
response	CompletionResponse The normal response of the operation in case of success. If the original method returns no data on success, such as Delete, the response is google.protobuf.Empty. If the original method is the standard Create/Update, the response should be the target resource of the operation. Any method that returns a long-running operation should document the response type, if any. Includes only one of the fields `error`, `response`. The operation result. If `done == false` and there was no failure detected, neither `error` nor `response` is set. If `done == false` and there was a failure detected, `error` is set. If `done == true`, exactly one of `error` or `response` is set.

CompletionResponse

Response containing generated text completions.

Field	Description
alternatives[]	Alternative A list of generated completion alternatives.
usage	ContentUsage A set of statistics describing the number of content tokens used by the completion model.
model_version	string The model version changes with each new releases.

Alternative

Represents a generated completion alternative, including its content and generation status.

Field

Description

message

Message

A message with the content of the alternative.

status

enum AlternativeStatus

The generation status of the alternative.

ALTERNATIVE_STATUS_UNSPECIFIED: Unspecified generation status.
ALTERNATIVE_STATUS_PARTIAL: Partially generated alternative.
ALTERNATIVE_STATUS_TRUNCATED_FINAL: Incomplete final alternative resulting from reaching the maximum allowed number of tokens.
ALTERNATIVE_STATUS_FINAL: Final alternative generated without running into any limits.
ALTERNATIVE_STATUS_CONTENT_FILTER: Generation was stopped due to the discovery of potentially sensitive content in the prompt or generated response.
To fix, modify the prompt and restart generation.
ALTERNATIVE_STATUS_TOOL_CALLS: Tools were invoked during the completion generation.

Message

A message object representing a wrapper over the inputs and outputs of the completion model.

Field	Description
role	string The ID of the message sender. Supported roles: `system`: Special role used to define the behaviour of the completion model. `assistant`: A role used by the model to generate responses. `user`: A role used by the user to describe requests to the model.
text	string Textual content of the message. Includes only one of the fields `text`, `tool_call_list`, `tool_result_list`. Message content.
tool_call_list	ToolCallList List of tool calls made by the model as part of the response generation. Includes only one of the fields `text`, `tool_call_list`, `tool_result_list`. Message content.
tool_result_list	ToolResultList List of tool results returned from external tools that were invoked by the model. Includes only one of the fields `text`, `tool_call_list`, `tool_result_list`. Message content.

ToolCallList

Represents a list of tool calls.

Field

Description

tool_calls[]

ToolCall

A list of tool calls to be executed.

ToolCall

Represents a call to a tool.

Field

Description

function_call

FunctionCall

Represents a call to a function.

Includes only one of the fields function_call.

FunctionCall

Represents the invocation of a function with specific arguments.

Field

Description

name

string

The name of the function being called.

arguments

google.protobuf.Struct

The structured arguments passed to the function.
These arguments must adhere to the JSON Schema defined in the corresponding function's parameters.

ToolResultList

Represents a list of tool results.

Field

Description

tool_results[]

ToolResult

A list of tool results.

ToolResult

Represents the result of a tool call.

Field

Description

function_result

FunctionResult

Represents the result of a function call.

Includes only one of the fields function_result.

FunctionResult

Represents the result of a function call.

Field

Description

name

string

The name of the function that was executed.

content

string

The result of the function call, represented as a string.
This field can be used to store the output of the function.

Includes only one of the fields content.

ContentUsage

An object representing the number of content tokens used by the completion model.

Field	Description
input_text_tokens	int64 The number of tokens in the textual part of the model input.
completion_tokens	int64 The number of tokens in the generated completion.
total_tokens	int64 The total number of tokens, including all input tokens and all generated tokens.

Foundation Models Text Generation API, gRPC: TextGenerationAsyncService.Completion

gRPC requestgRPC request

CompletionRequestCompletionRequest

CompletionOptionsCompletionOptions

MessageMessage

ToolCallListToolCallList

ToolCallToolCall

FunctionCallFunctionCall

ToolResultListToolResultList

ToolResultToolResult

FunctionResultFunctionResult

ToolTool

FunctionToolFunctionTool

operation.Operationoperation.Operation

CompletionResponseCompletionResponse

AlternativeAlternative

MessageMessage

ToolCallListToolCallList

ToolCallToolCall

FunctionCallFunctionCall

ToolResultListToolResultList

ToolResultToolResult

FunctionResultFunctionResult

ContentUsageContentUsage

Была ли статья полезна?

gRPC request

CompletionRequest

CompletionOptions

Message

ToolCallList

ToolCall

FunctionCall

ToolResultList

ToolResult

FunctionResult

Tool

FunctionTool

operation.Operation

CompletionResponse

Alternative

Message

ToolCallList

ToolCall

FunctionCall

ToolResultList

ToolResult

FunctionResult

ContentUsage