Merge pull request #147 from yandex-cloud/feature/endpoints-ai

fix: add missing endpoints for AI services
yandex-cloud · Oct 20, 2023 · ecb796e · ecb796e
2 parents 79ac104 + 37dd72c
commit ecb796e
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 13 deletions.
diff --git a/.github/actions/checkout-and-install-node/action.yml b/.github/actions/checkout-and-install-node/action.yml
@@ -12,6 +12,7 @@ runs:
       with:
         fetch-depth: 0
         persist-credentials: ${{ inputs.persist-credentials }}
+        submodules: ${{ inputs.submodules }}
     - uses: actions/setup-node@v2
       with:
         node-version-file: '.nvmrc'

diff --git a/.github/workflows/pr-checks.yml b/.github/workflows/pr-checks.yml
@@ -9,20 +9,25 @@ jobs:
   tests:
     runs-on: ubuntu-20.04
     steps:
-      - uses: yandex-cloud/nodejs-sdk/.github/actions/checkout-and-install-node@f69248b52b7991214847e889f28ba0883ed0ca2c
+      - uses: yandex-cloud/nodejs-sdk/.github/actions/checkout-and-install-node@9fc72bb2948886090f84c9236580fbca480bdea7
       - run: npm run test
-#  check-endpoints:
-#    runs-on: ubuntu-20.04
-#    steps:
-#      - uses: yandex-cloud/nodejs-sdk/.github/actions/checkout-and-install-node@f69248b52b7991214847e889f28ba0883ed0ca2c
-#      - run: npm run check-endpoints
+  check-endpoints:
+    runs-on: ubuntu-20.04
+    # Currently, the check-endpoints script fails on the master branch.
+    # So we need to continue on error.
+    continue-on-error: true
+    steps:
+      - uses: yandex-cloud/nodejs-sdk/.github/actions/checkout-and-install-node@9fc72bb2948886090f84c9236580fbca480bdea7
+        with:
+          submodules: recursive
+      - run: npm run check-endpoints
   lint:
     runs-on: ubuntu-20.04
     steps:
-      - uses: yandex-cloud/nodejs-sdk/.github/actions/checkout-and-install-node@f69248b52b7991214847e889f28ba0883ed0ca2c
+      - uses: yandex-cloud/nodejs-sdk/.github/actions/checkout-and-install-node@9fc72bb2948886090f84c9236580fbca480bdea7
       - run: npm run lint
   build:
     runs-on: ubuntu-20.04
     steps:
-      - uses: yandex-cloud/nodejs-sdk/.github/actions/checkout-and-install-node@f69248b52b7991214847e889f28ba0883ed0ca2c
+      - uses: yandex-cloud/nodejs-sdk/.github/actions/checkout-and-install-node@9fc72bb2948886090f84c9236580fbca480bdea7
       - run: npm run build
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -11,7 +11,7 @@ jobs:
     name: Release
     runs-on: ubuntu-20.04
     steps:
-      - uses: yandex-cloud/nodejs-sdk/.github/actions/checkout-and-install-node@f69248b52b7991214847e889f28ba0883ed0ca2c
+      - uses: yandex-cloud/nodejs-sdk/.github/actions/checkout-and-install-node@9fc72bb2948886090f84c9236580fbca480bdea7
         with:
           persist-credentials: false
       - env:

diff --git 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 => {