feat: list all ai models (#562)

Author: NazarLysyi

src/ai/index.ts

@@ -355,6 +355,17 @@ export class Ai extends CrowdinApi {
         return this.getList(url, options?.limit, options?.offset);
     }
 
+    /**
+     * @param options optional parameters for the request
+     * @see https://support.crowdin.com/developer/enterprise/api/v2/#tag/AI/operation/api.ai.providers.models.enterprise.getMany
+     */
+    listAiOrganizationAllProviderModels(
+        options?: PaginationOptions,
+    ): Promise<ResponseList<AiModel.AiProviderModelResponse>> {
+        const url = `${this.url}/ai/providers/models`;
+        return this.getList(url, options?.limit, options?.offset);
+    }
+
     /**
      * @param aiProviderId ai Provider identifier
      * @param request request body
@@ -824,6 +835,20 @@ export class Ai extends CrowdinApi {
         return this.getList(url);
     }
 
+    /**
+     * @param userId user identifier
+     * @param options optional parameters for the request
+     * @see https://support.crowdin.com/developer/api/v2/#tag/AI/operation/api.ai.providers.models.crowdin.getMany
+     *
+     */
+    listAiUserAllProviderModels(
+        userId: number,
+        options?: PaginationOptions,
+    ): Promise<ResponseList<AiModel.AiProviderModelResponse>> {
+        const url = `${this.url}/users/${userId}/ai/providers/models`;
+        return this.getList(url, options?.limit, options?.offset);
+    }
+
     /**
      * @param userId user Identifier
      * @param aiProviderId ai Provider identifier

tests/ai/api.test.ts

@@ -450,6 +450,24 @@ describe('AI API', () => {
                     limit: limit,
                 },
             })
+            .get('/ai/providers/models', undefined, {
+                reqheaders: {
+                    Authorization: `Bearer ${api.token}`,
+                },
+            })
+            .reply(200, {
+                data: [
+                    {
+                        data: {
+                            id: aiModelId,
+                        },
+                    },
+                ],
+                pagination: {
+                    offset: 0,
+                    limit: limit,
+                },
+            })
             .post(`/ai/providers/${aiProviderId}/chat/completions`, field, {
                 reqheaders: {
                     Authorization: `Bearer ${api.token}`,
@@ -902,6 +920,24 @@ describe('AI API', () => {
                     limit: limit,
                 },
             })
+            .get(`/users/${userId}/ai/providers/models`, undefined, {
+                reqheaders: {
+                    Authorization: `Bearer ${api.token}`,
+                },
+            })
+            .reply(200, {
+                data: [
+                    {
+                        data: {
+                            id: aiModelId,
+                        },
+                    },
+                ],
+                pagination: {
+                    offset: 0,
+                    limit: limit,
+                },
+            })
             .post(`/users/${userId}/ai/providers/${aiProviderId}/chat/completions`, field, {
                 reqheaders: {
                     Authorization: `Bearer ${api.token}`,
@@ -1158,6 +1194,13 @@ describe('AI API', () => {
         expect(providers.pagination.limit).toBe(limit);
     });
 
+    it('List AI Organization All Provider Models', async () => {
+        const providers = await api.listAiOrganizationAllProviderModels();
+        expect(providers.data.length).toBe(1);
+        expect(providers.data[0].data.id).toBe(aiModelId);
+        expect(providers.pagination.limit).toBe(limit);
+    });
+
     it('Create AI Organization Proxy Chat Completion', async () => {
         const proxy = await api.createAiOrganizationProxyChatCompletion(aiProviderId, field);
         expect(proxy.data).toStrictEqual(field);
@@ -1376,6 +1419,13 @@ describe('AI API', () => {
         expect(providers.pagination.limit).toBe(limit);
     });
 
+    it('List AI User All Provider Models', async () => {
+        const providers = await api.listAiUserAllProviderModels(userId);
+        expect(providers.data.length).toBe(1);
+        expect(providers.data[0].data.id).toBe(aiModelId);
+        expect(providers.pagination.limit).toBe(limit);
+    });
+
     it('Create AI User Proxy Chat Completion', async () => {
         const proxy = await api.createAiUserProxyChatCompletion(userId, aiProviderId, field);
         expect(proxy.data).toStrictEqual(field);