Merge pull request #185 from yandex-cloud/services-update

feat: services update

Author: DavyJohnes

clients/ai-assistants-v1/generated/google/protobuf/struct.ts

@@ -5,6 +5,8 @@ import _m0 from 'protobufjs/minimal';
 import {
     PromptTruncationOptions,
     CompletionOptions,
+    Tool,
+    ToolCallList,
 } from '../../../../../../yandex/cloud/ai/assistants/v1/common';
 import { Timestamp } from '../../../../../../google/protobuf/timestamp';
 import { Error } from '../../../../../../yandex/cloud/ai/common/common';
@@ -41,6 +43,8 @@ export interface Run {
      * If specified, these options will override the assistant's completion settings for this run.
      */
     customCompletionOptions?: CompletionOptions;
+    /** List of tools that are available for the assistant to use in this run. */
+    tools: Tool[];
 }
 
 export interface Run_LabelsEntry {
@@ -58,6 +62,8 @@ export interface RunState {
     error?: Error | undefined;
     /** Final message generated by an assistant if a run has completed successfully. */
     completedMessage?: Message | undefined;
+    /** A list of tool calls requested by the assistant. */
+    toolCallList?: ToolCallList | undefined;
 }
 
 /** Enum representing the status of a run. */
@@ -72,6 +78,8 @@ export enum RunState_RunStatus {
     FAILED = 3,
     /** COMPLETED - Run has completed successfully. */
     COMPLETED = 4,
+    /** TOOL_CALLS - The run is waiting for tool calls to be executed and their results to be submitted. */
+    TOOL_CALLS = 5,
     UNRECOGNIZED = -1,
 }
 
@@ -92,6 +100,9 @@ export function runState_RunStatusFromJSON(object: any): RunState_RunStatus {
         case 4:
         case 'COMPLETED':
             return RunState_RunStatus.COMPLETED;
+        case 5:
+        case 'TOOL_CALLS':
+            return RunState_RunStatus.TOOL_CALLS;
         case -1:
         case 'UNRECOGNIZED':
         default:
@@ -111,6 +122,8 @@ export function runState_RunStatusToJSON(object: RunState_RunStatus): string {
             return 'FAILED';
         case RunState_RunStatus.COMPLETED:
             return 'COMPLETED';
+        case RunState_RunStatus.TOOL_CALLS:
+            return 'TOOL_CALLS';
         default:
             return 'UNKNOWN';
     }
@@ -182,6 +195,9 @@ export const Run = {
                 writer.uint32(82).fork(),
             ).ldelim();
         }
+        for (const v of message.tools) {
+            Tool.encode(v!, writer.uint32(90).fork()).ldelim();
+        }
         return writer;
     },
 
@@ -190,6 +206,7 @@ export const Run = {
         let end = length === undefined ? reader.len : reader.pos + length;
         const message = { ...baseRun } as Run;
         message.labels = {};
+        message.tools = [];
         while (reader.pos < end) {
             const tag = reader.uint32();
             switch (tag >>> 3) {
@@ -232,6 +249,9 @@ export const Run = {
                         reader.uint32(),
                     );
                     break;
+                case 11:
+                    message.tools.push(Tool.decode(reader, reader.uint32()));
+                    break;
                 default:
                     reader.skipType(tag & 7);
                     break;
@@ -283,6 +303,7 @@ export const Run = {
             object.customCompletionOptions !== undefined && object.customCompletionOptions !== null
                 ? CompletionOptions.fromJSON(object.customCompletionOptions)
                 : undefined;
+        message.tools = (object.tools ?? []).map((e: any) => Tool.fromJSON(e));
         return message;
     },
 
@@ -311,6 +332,11 @@ export const Run = {
             (obj.customCompletionOptions = message.customCompletionOptions
                 ? CompletionOptions.toJSON(message.customCompletionOptions)
                 : undefined);
+        if (message.tools) {
+            obj.tools = message.tools.map((e) => (e ? Tool.toJSON(e) : undefined));
+        } else {
+            obj.tools = [];
+        }
         return obj;
     },
 
@@ -347,6 +373,7 @@ export const Run = {
             object.customCompletionOptions !== undefined && object.customCompletionOptions !== null
                 ? CompletionOptions.fromPartial(object.customCompletionOptions)
                 : undefined;
+        message.tools = object.tools?.map((e) => Tool.fromPartial(e)) || [];
         return message;
     },
 };
@@ -433,6 +460,9 @@ export const RunState = {
         if (message.completedMessage !== undefined) {
             Message.encode(message.completedMessage, writer.uint32(26).fork()).ldelim();
         }
+        if (message.toolCallList !== undefined) {
+            ToolCallList.encode(message.toolCallList, writer.uint32(34).fork()).ldelim();
+        }
         return writer;
     },
 
@@ -452,6 +482,9 @@ export const RunState = {
                 case 3:
                     message.completedMessage = Message.decode(reader, reader.uint32());
                     break;
+                case 4:
+                    message.toolCallList = ToolCallList.decode(reader, reader.uint32());
+                    break;
                 default:
                     reader.skipType(tag & 7);
                     break;
@@ -474,6 +507,10 @@ export const RunState = {
             object.completedMessage !== undefined && object.completedMessage !== null
                 ? Message.fromJSON(object.completedMessage)
                 : undefined;
+        message.toolCallList =
+            object.toolCallList !== undefined && object.toolCallList !== null
+                ? ToolCallList.fromJSON(object.toolCallList)
+                : undefined;
         return message;
     },
 
@@ -486,6 +523,10 @@ export const RunState = {
             (obj.completedMessage = message.completedMessage
                 ? Message.toJSON(message.completedMessage)
                 : undefined);
+        message.toolCallList !== undefined &&
+            (obj.toolCallList = message.toolCallList
+                ? ToolCallList.toJSON(message.toolCallList)
+                : undefined);
         return obj;
     },
 
@@ -500,6 +541,10 @@ export const RunState = {
             object.completedMessage !== undefined && object.completedMessage !== null
                 ? Message.fromPartial(object.completedMessage)
                 : undefined;
+        message.toolCallList =
+            object.toolCallList !== undefined && object.toolCallList !== null
+                ? ToolCallList.fromPartial(object.toolCallList)
+                : undefined;
         return message;
     },
 };