From 891f93dfc7f67fdcc0bb78482fad29f6c292e8cf Mon Sep 17 00:00:00 2001
From: Nikolay Matrosov <nikolay.matrosov@gmail.com>
Date: Thu, 19 Oct 2023 18:10:12 +0200
Subject: [PATCH] feat: add missing endpoints for AI services

---
 src/service-endpoints.ts | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/src/service-endpoints.ts b/src/service-endpoints.ts
index a85b05c8..d3e65e61 100644
--- a/src/service-endpoints.ts
+++ b/src/service-endpoints.ts
@@ -1,7 +1,4 @@
-import {
-    ServiceClientConstructor,
-    ServiceDefinition,
-} from '@grpc/grpc-js';
+import { ServiceClientConstructor, ServiceDefinition } from '@grpc/grpc-js';
 import { GeneratedServiceClientCtor } from './types';
 
 interface ServiceEndpoint {
@@ -17,6 +14,15 @@ export const SERVICE_ENDPOINTS_LIST: ServiceEndpointsList = [
         serviceIds: ['yandex.cloud.operation.OperationService'],
         endpoint: 'operation.api.cloud.yandex.net:443',
     },
+    {
+        serviceIds: [
+            'yandex.cloud.ai.llm.v1alpha.EmbeddingsService',
+            'yandex.cloud.ai.llm.v1alpha.TextGenerationAsyncService',
+            'yandex.cloud.ai.llm.v1alpha.TextGenerationService',
+            'yandex.cloud.ai.llm.v1alpha.TokenizerService',
+        ],
+        endpoint: 'llm.api.cloud.yandex.net:443',
+    },
     {
         serviceIds: [
             'yandex.cloud.backup.v1.BackupService',
@@ -287,6 +293,7 @@ export const SERVICE_ENDPOINTS_LIST: ServiceEndpointsList = [
     {
         serviceIds: [
             'yandex.cloud.ai.ocr.v1.TextRecognitionService',
+            'yandex.cloud.ai.ocr.v1.TextRecognitionAsyncService',
         ],
         endpoint: 'ocr.api.cloud.yandex.net:443',
     },
@@ -412,6 +419,13 @@ export const SERVICE_ENDPOINTS_LIST: ServiceEndpointsList = [
         ],
         endpoint: 'loadtesting.api.cloud.yandex.net:443',
     },
+    {
+        serviceIds: [
+            'speechkit.stt.v3.AsyncRecognizer',
+            'speechkit.stt.v3.Recognizer',
+        ],
+        endpoint: 'stt.api.cloud.yandex.net:443',
+    },
 ];
 
 export const getServiceClientEndpoint = <T extends ServiceDefinition>(generatedClientCtor: GeneratedServiceClientCtor<T>): string => {