chore: update api

Author: peppescg

src/api/generated/@tanstack/react-query.gen.ts

@@ -7,11 +7,11 @@ import {
   healthCheckHealthGet,
   v1ListProviderEndpoints,
   v1AddProviderEndpoint,
+  v1ListAllModelsForAllProviders,
+  v1ListModelsByProvider,
   v1GetProviderEndpoint,
   v1UpdateProviderEndpoint,
   v1DeleteProviderEndpoint,
-  v1ListModelsByProvider,
-  v1ListAllModelsForAllProviders,
   v1ListWorkspaces,
   v1CreateWorkspace,
   v1ListActiveWorkspaces,
@@ -36,14 +36,14 @@ import type {
   V1AddProviderEndpointData,
   V1AddProviderEndpointError,
   V1AddProviderEndpointResponse,
+  V1ListModelsByProviderData,
   V1GetProviderEndpointData,
   V1UpdateProviderEndpointData,
   V1UpdateProviderEndpointError,
   V1UpdateProviderEndpointResponse,
   V1DeleteProviderEndpointData,
   V1DeleteProviderEndpointError,
   V1DeleteProviderEndpointResponse,
-  V1ListModelsByProviderData,
   V1CreateWorkspaceData,
   V1CreateWorkspaceError,
   V1CreateWorkspaceResponse,
@@ -190,6 +190,48 @@ export const v1AddProviderEndpointMutation = (
   return mutationOptions;
 };
 
+export const v1ListAllModelsForAllProvidersQueryKey = (
+  options?: OptionsLegacyParser,
+) => [createQueryKey("v1ListAllModelsForAllProviders", options)];
+
+export const v1ListAllModelsForAllProvidersOptions = (
+  options?: OptionsLegacyParser,
+) => {
+  return queryOptions({
+    queryFn: async ({ queryKey, signal }) => {
+      const { data } = await v1ListAllModelsForAllProviders({
+        ...options,
+        ...queryKey[0],
+        signal,
+        throwOnError: true,
+      });
+      return data;
+    },
+    queryKey: v1ListAllModelsForAllProvidersQueryKey(options),
+  });
+};
+
+export const v1ListModelsByProviderQueryKey = (
+  options: OptionsLegacyParser<V1ListModelsByProviderData>,
+) => [createQueryKey("v1ListModelsByProvider", options)];
+
+export const v1ListModelsByProviderOptions = (
+  options: OptionsLegacyParser<V1ListModelsByProviderData>,
+) => {
+  return queryOptions({
+    queryFn: async ({ queryKey, signal }) => {
+      const { data } = await v1ListModelsByProvider({
+        ...options,
+        ...queryKey[0],
+        signal,
+        throwOnError: true,
+      });
+      return data;
+    },
+    queryKey: v1ListModelsByProviderQueryKey(options),
+  });
+};
+
 export const v1GetProviderEndpointQueryKey = (
   options: OptionsLegacyParser<V1GetProviderEndpointData>,
 ) => [createQueryKey("v1GetProviderEndpoint", options)];
@@ -251,48 +293,6 @@ export const v1DeleteProviderEndpointMutation = (
   return mutationOptions;
 };
 
-export const v1ListModelsByProviderQueryKey = (
-  options: OptionsLegacyParser<V1ListModelsByProviderData>,
-) => [createQueryKey("v1ListModelsByProvider", options)];
-
-export const v1ListModelsByProviderOptions = (
-  options: OptionsLegacyParser<V1ListModelsByProviderData>,
-) => {
-  return queryOptions({
-    queryFn: async ({ queryKey, signal }) => {
-      const { data } = await v1ListModelsByProvider({
-        ...options,
-        ...queryKey[0],
-        signal,
-        throwOnError: true,
-      });
-      return data;
-    },
-    queryKey: v1ListModelsByProviderQueryKey(options),
-  });
-};
-
-export const v1ListAllModelsForAllProvidersQueryKey = (
-  options?: OptionsLegacyParser,
-) => [createQueryKey("v1ListAllModelsForAllProviders", options)];
-
-export const v1ListAllModelsForAllProvidersOptions = (
-  options?: OptionsLegacyParser,
-) => {
-  return queryOptions({
-    queryFn: async ({ queryKey, signal }) => {
-      const { data } = await v1ListAllModelsForAllProviders({
-        ...options,
-        ...queryKey[0],
-        signal,
-        throwOnError: true,
-      });
-      return data;
-    },
-    queryKey: v1ListAllModelsForAllProvidersQueryKey(options),
-  });
-};
-
 export const v1ListWorkspacesQueryKey = (options?: OptionsLegacyParser) => [
   createQueryKey("v1ListWorkspaces", options),
 ];

src/api/generated/sdk.gen.ts

@@ -14,6 +14,11 @@ import type {
   V1AddProviderEndpointData,
   V1AddProviderEndpointError,
   V1AddProviderEndpointResponse,
+  V1ListAllModelsForAllProvidersError,
+  V1ListAllModelsForAllProvidersResponse,
+  V1ListModelsByProviderData,
+  V1ListModelsByProviderError,
+  V1ListModelsByProviderResponse,
   V1GetProviderEndpointData,
   V1GetProviderEndpointError,
   V1GetProviderEndpointResponse,
@@ -23,11 +28,6 @@ import type {
   V1DeleteProviderEndpointData,
   V1DeleteProviderEndpointError,
   V1DeleteProviderEndpointResponse,
-  V1ListModelsByProviderData,
-  V1ListModelsByProviderError,
-  V1ListModelsByProviderResponse,
-  V1ListAllModelsForAllProvidersError,
-  V1ListAllModelsForAllProvidersResponse,
   V1ListWorkspacesError,
   V1ListWorkspacesResponse,
   V1CreateWorkspaceData,
@@ -131,6 +131,42 @@ export const v1AddProviderEndpoint = <ThrowOnError extends boolean = false>(
   });
 };
 
+/**
+ * List All Models For All Providers
+ * List all models for all providers.
+ */
+export const v1ListAllModelsForAllProviders = <
+  ThrowOnError extends boolean = false,
+>(
+  options?: OptionsLegacyParser<unknown, ThrowOnError>,
+) => {
+  return (options?.client ?? client).get<
+    V1ListAllModelsForAllProvidersResponse,
+    V1ListAllModelsForAllProvidersError,
+    ThrowOnError
+  >({
+    ...options,
+    url: "/api/v1/provider-endpoints/models",
+  });
+};
+
+/**
+ * List Models By Provider
+ * List models by provider.
+ */
+export const v1ListModelsByProvider = <ThrowOnError extends boolean = false>(
+  options: OptionsLegacyParser<V1ListModelsByProviderData, ThrowOnError>,
+) => {
+  return (options?.client ?? client).get<
+    V1ListModelsByProviderResponse,
+    V1ListModelsByProviderError,
+    ThrowOnError
+  >({
+    ...options,
+    url: "/api/v1/provider-endpoints/{provider_id}/models",
+  });
+};
+
 /**
  * Get Provider Endpoint
  * Get a provider endpoint by ID.
@@ -182,42 +218,6 @@ export const v1DeleteProviderEndpoint = <ThrowOnError extends boolean = false>(
   });
 };
 
-/**
- * List Models By Provider
- * List models by provider.
- */
-export const v1ListModelsByProvider = <ThrowOnError extends boolean = false>(
-  options: OptionsLegacyParser<V1ListModelsByProviderData, ThrowOnError>,
-) => {
-  return (options?.client ?? client).get<
-    V1ListModelsByProviderResponse,
-    V1ListModelsByProviderError,
-    ThrowOnError
-  >({
-    ...options,
-    url: "/api/v1/provider-endpoints/{provider_name}/models",
-  });
-};
-
-/**
- * List All Models For All Providers
- * List all models for all providers.
- */
-export const v1ListAllModelsForAllProviders = <
-  ThrowOnError extends boolean = false,
->(
-  options?: OptionsLegacyParser<unknown, ThrowOnError>,
-) => {
-  return (options?.client ?? client).get<
-    V1ListAllModelsForAllProvidersResponse,
-    V1ListAllModelsForAllProvidersError,
-    ThrowOnError
-  >({
-    ...options,
-    url: "/api/v1/provider-endpoints/models",
-  });
-};
-
 /**
  * List Workspaces
  * List all workspaces.

src/api/generated/types.gen.ts

@@ -53,7 +53,7 @@ export type Conversation = {
   type: QuestionType;
   chat_id: string;
   conversation_timestamp: string;
-  token_usage: TokenUsage | null;
+  token_usage_agg: TokenUsageAggregate | null;
 };
 
 export type CreateOrRenameWorkspaceRequest = {
@@ -84,7 +84,8 @@ export type ListWorkspacesResponse = {
  */
 export type ModelByProvider = {
   name: string;
-  provider: string;
+  provider_id: string;
+  provider_name: string;
 };
 
 /**
@@ -120,7 +121,7 @@ export enum ProviderAuthType {
  * so we can use this for muxing messages.
  */
 export type ProviderEndpoint = {
-  id: number;
+  id?: string | null;
   name: string;
   description?: string;
   provider_type: ProviderType;
@@ -135,6 +136,9 @@ export enum ProviderType {
   OPENAI = "openai",
   ANTHROPIC = "anthropic",
   VLLM = "vllm",
+  OLLAMA = "ollama",
+  LM_STUDIO = "lm_studio",
+  LLAMACPP = "llamacpp",
 }
 
 /**
@@ -150,13 +154,26 @@ export enum QuestionType {
   FIM = "fim",
 }
 
+/**
+ * TokenUsage it's not a table, it's a model to represent the token usage.
+ * The data is stored in the outputs table.
+ */
+export type TokenUsage = {
+  input_tokens?: number;
+  output_tokens?: number;
+  input_cost?: number;
+  output_cost?: number;
+};
+
 /**
  * Represents the tokens used. Includes the information of the tokens used by model.
  * `used_tokens` are the total tokens used in the `tokens_by_model` list.
  */
-export type TokenUsage = {
-  tokens_by_model: Array<TokenUsageByModel>;
-  used_tokens: number;
+export type TokenUsageAggregate = {
+  tokens_by_model: {
+    [key: string]: TokenUsageByModel;
+  };
+  token_usage: TokenUsage;
 };
 
 /**
@@ -165,7 +182,7 @@ export type TokenUsage = {
 export type TokenUsageByModel = {
   provider_type: ProviderType;
   model: string;
-  used_tokens: number;
+  token_usage: TokenUsage;
 };
 
 export type ValidationError = {
@@ -201,9 +218,23 @@ export type V1AddProviderEndpointResponse = ProviderEndpoint;
 
 export type V1AddProviderEndpointError = HTTPValidationError;
 
+export type V1ListAllModelsForAllProvidersResponse = Array<ModelByProvider>;
+
+export type V1ListAllModelsForAllProvidersError = unknown;
+
+export type V1ListModelsByProviderData = {
+  path: {
+    provider_id: string;
+  };
+};
+
+export type V1ListModelsByProviderResponse = Array<ModelByProvider>;
+
+export type V1ListModelsByProviderError = HTTPValidationError;
+
 export type V1GetProviderEndpointData = {
   path: {
-    provider_id: number;
+    provider_id: string;
   };
 };
 
@@ -214,7 +245,7 @@ export type V1GetProviderEndpointError = HTTPValidationError;
 export type V1UpdateProviderEndpointData = {
   body: ProviderEndpoint;
   path: {
-    provider_id: number;
+    provider_id: string;
   };
 };
 
@@ -224,28 +255,14 @@ export type V1UpdateProviderEndpointError = HTTPValidationError;
 
 export type V1DeleteProviderEndpointData = {
   path: {
-    provider_id: number;
+    provider_id: string;
   };
 };
 
 export type V1DeleteProviderEndpointResponse = unknown;
 
 export type V1DeleteProviderEndpointError = HTTPValidationError;
 
-export type V1ListModelsByProviderData = {
-  path: {
-    provider_name: string;
-  };
-};
-
-export type V1ListModelsByProviderResponse = Array<ModelByProvider>;
-
-export type V1ListModelsByProviderError = HTTPValidationError;
-
-export type V1ListAllModelsForAllProvidersResponse = Array<ModelByProvider>;
-
-export type V1ListAllModelsForAllProvidersError = unknown;
-
 export type V1ListWorkspacesResponse = ListWorkspacesResponse;
 
 export type V1ListWorkspacesError = unknown;
@@ -393,6 +410,6 @@ export type V1GetWorkspaceTokenUsageData = {
   };
 };
 
-export type V1GetWorkspaceTokenUsageResponse = TokenUsage;
+export type V1GetWorkspaceTokenUsageResponse = TokenUsageAggregate;
 
 export type V1GetWorkspaceTokenUsageError = HTTPValidationError;