@daocloud-proto/hydra 0.18.0-dev-10 → 0.18.0-dev-12

package/admin-api/guardrails/v1alpha1/guardrails.pb.ts

@@ -6,7 +6,6 @@
 
 import * as HydraCommonCommon from "../../../common/common.pb"
 import * as fm from "../../../fetch.pb"
-import * as GoogleApiHttpbody from "../../../google/api/httpbody.pb"
 import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
 
 type Absent<T, K extends keyof T> = { [k in Exclude<keyof T, K>]?: undefined };
@@ -281,7 +280,7 @@ export class AIGuardrailsService {
   static QueryAIAuditLogStatistics(req: QueryAIAuditLogStatisticsRequest, initReq?: fm.InitReq): Promise<QueryAIAuditLogStatisticsResponse> {
     return fm.fetchReq<QueryAIAuditLogStatisticsRequest, QueryAIAuditLogStatisticsResponse>(`/apis/admin.hydra.io/v1alpha1/ai-guardrails-logs/statistics?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
   }
-  static ExportAIAuditLog(req: ExportAIAuditLogRequest, initReq?: fm.InitReq): Promise<GoogleApiHttpbody.HttpBody> {
-    return fm.fetchReq<ExportAIAuditLogRequest, GoogleApiHttpbody.HttpBody>(`/apis/admin.hydra.io/v1alpha1/ai-guardrails-logs/export?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  static ExportAIAuditLog(req: ExportAIAuditLogRequest, initReq?: fm.InitReq): Promise<Response> {
+    return fm.fetchRawReq(`/apis/admin.hydra.io/v1alpha1/ai-guardrails-logs/export?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
   }
 }

package/admin-api/maas/v1alpha1/maas.pb.ts

@@ -259,15 +259,20 @@ export type ModelRateLimitPolicy = {
   generation?: string
 }
 
+export type RateLimitValue = {
+  name?: string
+  value?: string
+}
+
 export type RateLimitTarget = {
   selectors?: RateLimitSelector[]
   dimension?: RateLimitDimension
-  values?: string[]
+  values?: RateLimitValue[]
 }
 
 export type RateLimitSelector = {
   dimension?: RateLimitDimension
-  values?: string[]
+  values?: RateLimitValue[]
 }
 
 export type RequestRateLimit = {
@@ -308,6 +313,7 @@ export type RateLimitExemption = {
 }
 
 export type ModelRateLimitPolicyRule = {
+  name?: string
   id?: string
   type?: RateLimitType
   target?: RateLimitTarget

package/fetch.pb.ts

@@ -338,4 +338,21 @@ export function renderURLSearchParams<T extends RequestPayload>(
   );
 
   return new URLSearchParams(urlSearchParams).toString();
-}
+}
+
+export async function fetchRawReq(path: string, init?: InitReq): Promise<Response> {
+  const {pathPrefix, ...req} = init || {}
+  const url = pathPrefix ? `${pathPrefix}${path}` : path
+  const response = await fetch(url, req)
+  if (!response.ok) {
+    const text = await response.text()
+    let body: unknown = text
+    try {
+      body = JSON.parse(text)
+    } catch {
+      // Preserve a non-JSON error body when an intermediary does not use the gateway error schema.
+    }
+    throw body
+  }
+  return response
+}

package/management-api/model-usage/v1alpha1/model_usage.pb.ts

@@ -0,0 +1,66 @@
+/* eslint-disable */
+// @ts-nocheck
+/*
+* This file is a generated Typescript file for GRPC Gateway, DO NOT MODIFY
+*/
+
+import * as HydraCommonCommon from "../../../common/common.pb"
+import * as fm from "../../../fetch.pb"
+import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
+
+export enum ModelUsageExportFormat {
+  MODEL_USAGE_EXPORT_FORMAT_UNSPECIFIED = "MODEL_USAGE_EXPORT_FORMAT_UNSPECIFIED",
+  MODEL_USAGE_EXPORT_FORMAT_CSV = "MODEL_USAGE_EXPORT_FORMAT_CSV",
+  MODEL_USAGE_EXPORT_FORMAT_EXCEL = "MODEL_USAGE_EXPORT_FORMAT_EXCEL",
+}
+
+export type ExportModelUsageRequest = {
+  startTime?: GoogleProtobufTimestamp.Timestamp
+  endTime?: GoogleProtobufTimestamp.Timestamp
+  apiKeyIds?: string[]
+  models?: string[]
+  userId?: string
+  responseCodes?: string[]
+  format?: ModelUsageExportFormat
+}
+
+export type ListModelUsageRequest = {
+  startTime?: GoogleProtobufTimestamp.Timestamp
+  endTime?: GoogleProtobufTimestamp.Timestamp
+  apiKeyIds?: string[]
+  models?: string[]
+  userId?: string
+  page?: HydraCommonCommon.Pagination
+  responseCodes?: string[]
+}
+
+export type ModelUsageItem = {
+  requestTime?: GoogleProtobufTimestamp.Timestamp
+  requestId?: string
+  model?: string
+  apiKeyNameOrId?: string
+  statusCode?: string
+  totalTokens?: string
+  inputTokens?: string
+  outputTokens?: string
+}
+
+export type ModelUsageColumn = {
+  field?: string
+  title?: string
+}
+
+export type ListModelUsageResponse = {
+  items?: ModelUsageItem[]
+  page?: HydraCommonCommon.Pagination
+  columns?: ModelUsageColumn[]
+}
+
+export class ModelUsageService {
+  static ListModelUsage(req: ListModelUsageRequest, initReq?: fm.InitReq): Promise<ListModelUsageResponse> {
+    return fm.fetchReq<ListModelUsageRequest, ListModelUsageResponse>(`/apis/hydra.io/v1alpha1/model-usage?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  }
+  static ExportModelUsage(req: ExportModelUsageRequest, initReq?: fm.InitReq): Promise<Response> {
+    return fm.fetchRawReq(`/apis/hydra.io/v1alpha1/model-usage/export?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@daocloud-proto/hydra",
-  "version": "v0.18.0-dev-10",
+  "version": "v0.18.0-dev-12",
   "description": "",
   "author": "",
   "license": "ISC"