@flashbacktech/flashbackclient 0.2.47 → 0.2.49

package/dist/api/client.d.ts

@@ -17,7 +17,7 @@ import { CreateOrgUserRequest, CreateOrgUserResponse, DeleteOrgUserResponse, Get
 import { SystemEventQueryRequest, SystemEventQueryResponse } from './types/platform/systemevent';
 import { PreVerifyEmailResponse, UserUpdateRequest, UserUpdateResponse } from './types/platform/user';
 import { CreateRepoAiApiKeyRequest, CreateRepoAiApiKeyResponse, DeleteRepoAiApiKeyResponse, GetRepoAiApiKeysResponse, UpdateRepoAiApiKeyRequest, UpdateRepoAiApiKeyResponse } from './types/ai/aiapikey';
-import { AiLlmStatsResponse, CreateAiLlmRequest, CreateAiLlmResponse, DeleteAiLlmResponse, GetAiLlmsResponse, UpdateAiLlmRequest, UpdateAiLlmResponse, ValidateAiLlmResponse } from './types/ai/aillm';
+import { AiLlmStatsResponse, CreateAiLlmRequest, CreateAiLlmResponse, DeleteAiLlmResponse, GetAiLlmsResponse, UpdateAiLlmRequest, UpdateAiLlmResponse, ValidateAiLlmResponse, GetAiModelsRequest, GetAiModelsResponse } from './types/ai/aillm';
 import { CreatePolicyRequest, GetPoliciesQuery, GetPolicyViolationsQuery, GetPolicyViolationsResponse, GetPolicyAlertsQuery, GetPolicyAlertsResponse, PolicyDTO, UpdatePolicyRequest, PolicyValidationRequest, PolicyValidationResponse, PolicyRecommendationRequest, PolicyRecommendationResponse } from './types/ai/policy';
 import { CreateConversationRequest, CreateConversationResponse, SendPromptRequest, SendPromptResponse, GetConversationsRequest, GetConversationsResponse, GetConversationMessagesResponse, GetConversationMessagesRequest, DeleteConversationRequest, DeleteConversationResponse } from './types/ai/conversation';
 import { GetLinksRequest, GetLinksResponse, CreateLinkRequest, CreateLinkResponse, UpdateLinkRequest, UpdateLinkResponse, DeleteLinkResponse, GetLinkByTokenResponse } from './types/platform/links';
@@ -273,5 +273,6 @@ export declare class ApiClient implements IApiClient {
     getPolicyAlertsByPolicyId: (policyId: string, query: Omit<GetPolicyAlertsQuery, "policyId">) => Promise<GetPolicyAlertsResponse>;
     validatePolicies: (data: PolicyValidationRequest) => Promise<PolicyValidationResponse>;
     recommendPolicies: (data: PolicyRecommendationRequest) => Promise<PolicyRecommendationResponse>;
+    getAiModels: (data: GetAiModelsRequest) => Promise<GetAiModelsResponse>;
 }
 export {};

package/dist/api/client.js

@@ -698,6 +698,21 @@ class ApiClient {
             if (query.skip !== undefined) {
                 queryParams.append('skip', query.skip.toString());
             }
+            if (query.llmType && query.llmType.length > 0) {
+                queryParams.append('llmType', query.llmType.join(','));
+            }
+            if (query.llmModel && query.llmModel.length > 0) {
+                queryParams.append('llmModel', query.llmModel.join(','));
+            }
+            if (query.host && query.host.length > 0) {
+                queryParams.append('host', query.host.join(','));
+            }
+            if (query.severity && query.severity.length > 0) {
+                queryParams.append('severity', query.severity.join(','));
+            }
+            if (query.block !== undefined) {
+                queryParams.append('block', query.block.toString());
+            }
             return this.makeRequest(`policy/violations?${queryParams.toString()}`, 'GET', null);
         };
         this.getPolicyViolationsByPolicyId = async (policyId, query) => {
@@ -720,6 +735,21 @@ class ApiClient {
             if (query.skip !== undefined) {
                 queryParams.append('skip', query.skip.toString());
             }
+            if (query.llmType && query.llmType.length > 0) {
+                queryParams.append('llmType', query.llmType.join(','));
+            }
+            if (query.llmModel && query.llmModel.length > 0) {
+                queryParams.append('llmModel', query.llmModel.join(','));
+            }
+            if (query.host && query.host.length > 0) {
+                queryParams.append('host', query.host.join(','));
+            }
+            if (query.severity && query.severity.length > 0) {
+                queryParams.append('severity', query.severity.join(','));
+            }
+            if (query.block !== undefined) {
+                queryParams.append('block', query.block.toString());
+            }
             return this.makeRequest(`policy/${policyId}/violations${queryParams.toString() ? `?${queryParams.toString()}` : ''}`, 'GET', null);
         };
         this.getPolicyAlerts = async (query) => {
@@ -745,6 +775,15 @@ class ApiClient {
             if (query.skip !== undefined) {
                 queryParams.append('skip', query.skip.toString());
             }
+            if (query.llmType && query.llmType.length > 0) {
+                queryParams.append('llmType', query.llmType.join(','));
+            }
+            if (query.llmModel && query.llmModel.length > 0) {
+                queryParams.append('llmModel', query.llmModel.join(','));
+            }
+            if (query.host && query.host.length > 0) {
+                queryParams.append('host', query.host.join(','));
+            }
             return this.makeRequest(`policy/alerts?${queryParams.toString()}`, 'GET', null);
         };
         this.getPolicyAlertsByPolicyId = async (policyId, query) => {
@@ -767,6 +806,15 @@ class ApiClient {
             if (query.skip !== undefined) {
                 queryParams.append('skip', query.skip.toString());
             }
+            if (query.llmType && query.llmType.length > 0) {
+                queryParams.append('llmType', query.llmType.join(','));
+            }
+            if (query.llmModel && query.llmModel.length > 0) {
+                queryParams.append('llmModel', query.llmModel.join(','));
+            }
+            if (query.host && query.host.length > 0) {
+                queryParams.append('host', query.host.join(','));
+            }
             return this.makeRequest(`policy/${policyId}/alerts${queryParams.toString() ? `?${queryParams.toString()}` : ''}`, 'GET', null);
         };
         this.validatePolicies = async (data) => {
@@ -775,6 +823,9 @@ class ApiClient {
         this.recommendPolicies = async (data) => {
             return this.makeRequest('policy/recommend', 'POST', data);
         };
+        this.getAiModels = async (data) => {
+            return this.makeRequest('ai/models', 'POST', data);
+        };
         this.baseURL = baseURL;
         this.headers = {};
         this.debug = false;
@@ -866,6 +917,12 @@ class ApiClient {
         if (params.hosts && params.hosts.length > 0) {
             queryParams.append('hosts', params.hosts.join(','));
         }
+        if (params.llmType && params.llmType.length > 0) {
+            queryParams.append('llmType', params.llmType.join(','));
+        }
+        if (params.llmModel && params.llmModel.length > 0) {
+            queryParams.append('llmModel', params.llmModel.join(','));
+        }
         return this.makeRequest(`aistats/daily?${queryParams.toString()}`, 'GET', null);
     }
     async getAiStatsMinute(params) {
@@ -888,6 +945,12 @@ class ApiClient {
         if (params.hosts && params.hosts.length > 0) {
             queryParams.append('hosts', params.hosts.join(','));
         }
+        if (params.llmType && params.llmType.length > 0) {
+            queryParams.append('llmType', params.llmType.join(','));
+        }
+        if (params.llmModel && params.llmModel.length > 0) {
+            queryParams.append('llmModel', params.llmModel.join(','));
+        }
         return this.makeRequest(`aistats/minute?${queryParams.toString()}`, 'GET', null);
     }
     async getNodeStatsMinute(params) {

package/dist/api/types/ai/aillm.d.ts

@@ -77,3 +77,20 @@ export interface RepoAiLlmInfo {
     aiLlmId?: string;
     aiLlm?: AiLlmDTO;
 }
+export interface GetAiModelsRequest {
+    repoId: string;
+    aiType?: AiType;
+    llmType?: string;
+}
+export interface AiModelDTO {
+    id: string;
+    name: string;
+    created?: number;
+    aiType: AiType;
+    llmType: string;
+    llmProvider: string;
+}
+export interface GetAiModelsResponse {
+    success: boolean;
+    models: AiModelDTO[];
+}

package/dist/api/types/ai/policy.d.ts

@@ -63,6 +63,11 @@ export interface GetPolicyViolationsQuery {
     to?: string;
     take?: number;
     skip?: number;
+    llmType?: string[];
+    llmModel?: string[];
+    host?: string[];
+    severity?: RiskType[];
+    block?: boolean;
 }
 export interface PolicyViolationDTO {
     id: string;
@@ -77,6 +82,11 @@ export interface PolicyViolationDTO {
     userName: string;
     repoAiApiKeyId: string;
     repoAiApiKeyName: string;
+    severity: RiskType | null;
+    block: boolean;
+    llmType: string;
+    llmModel: string;
+    host: string;
 }
 export interface GetPolicyViolationsResponse {
     success: boolean;
@@ -93,6 +103,9 @@ export interface GetPolicyAlertsQuery {
     to?: string;
     take?: number;
     skip?: number;
+    llmType?: string[];
+    llmModel?: string[];
+    host?: string[];
 }
 export interface PolicyAlertDTO {
     id: string;
@@ -107,6 +120,9 @@ export interface PolicyAlertDTO {
     userName: string;
     repoAiApiKeyId: string;
     repoAiApiKeyName: string;
+    llmType: string;
+    llmModel: string;
+    host: string;
 }
 export interface GetPolicyAlertsResponse {
     success: boolean;

package/dist/api/types/ai/stats.d.ts

@@ -5,6 +5,8 @@ export interface AiStatsQueryParams {
     aiLlmId?: string[];
     repoAiApiKeyId?: string[];
     hosts?: string[];
+    llmType?: string[];
+    llmModel?: string[];
 }
 export interface AiStatsResponse {
     success: boolean;
@@ -25,6 +27,9 @@ export interface AiStatsData {
     policyViolations: number;
     numAlerts: number;
     numBlocks: number;
+    numSeverityLow: number;
+    numSeverityMedium: number;
+    numSeverityHigh: number;
     latency_ms: number;
     llmType: string;
     llmModel: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@flashbacktech/flashbackclient",
-  "version": "0.2.47",
+  "version": "0.2.49",
   "type": "commonjs",
   "publishConfig": {
     "access": "public"