weave-typescript 0.51.3 → 0.53.0

package/dist/weavesql/weavedb/inference_model_route_sql.d.ts

@@ -0,0 +1,167 @@
+import { QueryArrayConfig, QueryArrayResult } from "pg";
+interface Client {
+    query: (config: QueryArrayConfig) => Promise<QueryArrayResult>;
+}
+export declare const upsertInferenceModelQuery = "-- name: UpsertInferenceModel :one\nWITH inserted AS (\n    INSERT INTO weave.inference_model (\n        organization_id,\n        model_id\n    ) VALUES (\n        $1,\n        $2\n    )\n    ON CONFLICT (organization_id, model_id) DO NOTHING\n    RETURNING\n        id,\n        organization_id,\n        model_id,\n        created_at,\n        updated_at,\n        true::boolean AS created\n)\nSELECT\n    id,\n    organization_id,\n    model_id,\n    created_at,\n    updated_at,\n    created\nFROM inserted\nUNION ALL\nSELECT\n    im.id,\n    im.organization_id,\n    im.model_id,\n    im.created_at,\n    im.updated_at,\n    false::boolean AS created\nFROM weave.inference_model im\nWHERE im.organization_id = $1\n  AND im.model_id = $2\n  AND NOT EXISTS (SELECT 1 FROM inserted)\nLIMIT 1";
+export interface UpsertInferenceModelArgs {
+    organizationId: string;
+    modelId: string;
+}
+export interface UpsertInferenceModelRow {
+    id: string;
+    organizationId: string;
+    modelId: string;
+    createdAt: Date;
+    updatedAt: Date;
+    created: boolean;
+}
+export declare function upsertInferenceModel(client: Client, args: UpsertInferenceModelArgs): Promise<UpsertInferenceModelRow | null>;
+export declare const getInferenceModelByModelIDQuery = "-- name: GetInferenceModelByModelID :one\nSELECT id, organization_id, model_id, created_at, updated_at\nFROM weave.inference_model\nWHERE organization_id = $1\n  AND model_id = $2";
+export interface GetInferenceModelByModelIDArgs {
+    organizationId: string;
+    modelId: string;
+}
+export interface GetInferenceModelByModelIDRow {
+    id: string;
+    organizationId: string;
+    modelId: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function getInferenceModelByModelID(client: Client, args: GetInferenceModelByModelIDArgs): Promise<GetInferenceModelByModelIDRow | null>;
+export declare const createInferenceModelRouteQuery = "-- name: CreateInferenceModelRoute :one\nWITH created AS (\n    INSERT INTO weave.inference_model_route (\n        organization_id,\n        inference_model_id,\n        inference_provider_configuration_id,\n        provider_model_id,\n        priority\n    ) VALUES (\n        $1,\n        $2,\n        $3,\n        $4,\n        COALESCE($5::integer, 100)\n    )\n    RETURNING id, organization_id, inference_model_id, inference_provider_configuration_id, provider_model_id, priority, created_at, updated_at\n)\nSELECT\n    r.id,\n    r.organization_id,\n    r.inference_model_id,\n    im.model_id,\n    r.inference_provider_configuration_id,\n    r.provider_model_id,\n    r.priority,\n    r.created_at,\n    r.updated_at\nFROM created r\n         JOIN weave.inference_model im\n              ON im.organization_id = r.organization_id\n                  AND im.id = r.inference_model_id";
+export interface CreateInferenceModelRouteArgs {
+    organizationId: string;
+    inferenceModelId: string;
+    inferenceProviderConfigurationId: string;
+    providerModelId: string;
+    priority: number | null;
+}
+export interface CreateInferenceModelRouteRow {
+    id: string;
+    organizationId: string;
+    inferenceModelId: string;
+    modelId: string;
+    inferenceProviderConfigurationId: string;
+    providerModelId: string;
+    priority: number;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function createInferenceModelRoute(client: Client, args: CreateInferenceModelRouteArgs): Promise<CreateInferenceModelRouteRow | null>;
+export declare const getInferenceModelRouteQuery = "-- name: GetInferenceModelRoute :one\nSELECT\n    r.id,\n    r.organization_id,\n    r.inference_model_id,\n    im.model_id,\n    r.inference_provider_configuration_id,\n    r.provider_model_id,\n    r.priority,\n    r.created_at,\n    r.updated_at\nFROM weave.inference_model_route r\n         JOIN weave.inference_model im\n              ON im.organization_id = r.organization_id\n                  AND im.id = r.inference_model_id\nWHERE r.organization_id = $1\n  AND r.id = $2";
+export interface GetInferenceModelRouteArgs {
+    organizationId: string;
+    inferenceModelRouteId: string;
+}
+export interface GetInferenceModelRouteRow {
+    id: string;
+    organizationId: string;
+    inferenceModelId: string;
+    modelId: string;
+    inferenceProviderConfigurationId: string;
+    providerModelId: string;
+    priority: number;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function getInferenceModelRoute(client: Client, args: GetInferenceModelRouteArgs): Promise<GetInferenceModelRouteRow | null>;
+export declare const updateInferenceModelRouteQuery = "-- name: UpdateInferenceModelRoute :one\nWITH updated AS (\n    UPDATE weave.inference_model_route\n    SET inference_model_id = $1,\n        inference_provider_configuration_id = $2,\n        provider_model_id = $3,\n        priority = $4\n    WHERE weave.inference_model_route.organization_id = $5\n      AND weave.inference_model_route.id = $6\n    RETURNING id, organization_id, inference_model_id, inference_provider_configuration_id, provider_model_id, priority, created_at, updated_at\n)\nSELECT\n    r.id,\n    r.organization_id,\n    r.inference_model_id,\n    im.model_id,\n    r.inference_provider_configuration_id,\n    r.provider_model_id,\n    r.priority,\n    r.created_at,\n    r.updated_at\nFROM updated r\n         JOIN weave.inference_model im\n              ON im.organization_id = r.organization_id\n                  AND im.id = r.inference_model_id";
+export interface UpdateInferenceModelRouteArgs {
+    inferenceModelId: string;
+    inferenceProviderConfigurationId: string;
+    providerModelId: string;
+    priority: number;
+    organizationId: string;
+    inferenceModelRouteId: string;
+}
+export interface UpdateInferenceModelRouteRow {
+    id: string;
+    organizationId: string;
+    inferenceModelId: string;
+    modelId: string;
+    inferenceProviderConfigurationId: string;
+    providerModelId: string;
+    priority: number;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function updateInferenceModelRoute(client: Client, args: UpdateInferenceModelRouteArgs): Promise<UpdateInferenceModelRouteRow | null>;
+export declare const deleteInferenceModelRouteQuery = "-- name: DeleteInferenceModelRoute :exec\nDELETE FROM weave.inference_model_route\nWHERE organization_id = $1\n  AND id = $2";
+export interface DeleteInferenceModelRouteArgs {
+    organizationId: string;
+    inferenceModelRouteId: string;
+}
+export declare function deleteInferenceModelRoute(client: Client, args: DeleteInferenceModelRouteArgs): Promise<void>;
+export declare const listInferenceModelRoutesQuery = "-- name: ListInferenceModelRoutes :many\nSELECT\n    r.id,\n    r.organization_id,\n    r.inference_model_id,\n    im.model_id,\n    r.inference_provider_configuration_id,\n    r.provider_model_id,\n    r.priority,\n    r.created_at,\n    r.updated_at\nFROM weave.inference_model_route r\n         JOIN weave.inference_model im\n              ON im.organization_id = r.organization_id\n                  AND im.id = r.inference_model_id\nWHERE r.organization_id = $1\n  AND ($2::text IS NULL OR im.model_id = $2::text)\nORDER BY im.model_id ASC, r.priority ASC, r.id ASC\nLIMIT COALESCE($4::integer, 100)\nOFFSET COALESCE($3::integer, 0)";
+export interface ListInferenceModelRoutesArgs {
+    organizationId: string;
+    modelId: string | null;
+    pageOffset: number | null;
+    pageSize: number | null;
+}
+export interface ListInferenceModelRoutesRow {
+    id: string;
+    organizationId: string;
+    inferenceModelId: string;
+    modelId: string;
+    inferenceProviderConfigurationId: string;
+    providerModelId: string;
+    priority: number;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function listInferenceModelRoutes(client: Client, args: ListInferenceModelRoutesArgs): Promise<ListInferenceModelRoutesRow[]>;
+export declare const getOrganizationChatSettingsQuery = "-- name: GetOrganizationChatSettings :one\nSELECT\n    o.id AS organization_id,\n    chat_im.model_id AS default_chat_model_id,\n    chat_r.inference_provider_configuration_id AS default_provider_configuration_id,\n    fast_im.model_id AS default_fast_model_id,\n    fast_r.inference_provider_configuration_id AS default_fast_provider_configuration_id\nFROM weave.organization o\n         LEFT JOIN weave.inference_model_route chat_r\n                   ON chat_r.organization_id = o.id\n                       AND chat_r.id = o.default_chat_model_route_id\n         LEFT JOIN weave.inference_model chat_im\n                   ON chat_im.organization_id = chat_r.organization_id\n                       AND chat_im.id = chat_r.inference_model_id\n         LEFT JOIN weave.inference_model_route fast_r\n                   ON fast_r.organization_id = o.id\n                       AND fast_r.id = o.default_fast_model_route_id\n         LEFT JOIN weave.inference_model fast_im\n                   ON fast_im.organization_id = fast_r.organization_id\n                       AND fast_im.id = fast_r.inference_model_id\nWHERE o.id = $1";
+export interface GetOrganizationChatSettingsArgs {
+    organizationId: string;
+}
+export interface GetOrganizationChatSettingsRow {
+    organizationId: string;
+    defaultChatModelId: string | null;
+    defaultProviderConfigurationId: string | null;
+    defaultFastModelId: string | null;
+    defaultFastProviderConfigurationId: string | null;
+}
+export declare function getOrganizationChatSettings(client: Client, args: GetOrganizationChatSettingsArgs): Promise<GetOrganizationChatSettingsRow | null>;
+export declare const setOrganizationChatSettingsQuery = "-- name: SetOrganizationChatSettings :one\nWITH requested_defaults AS (\n    SELECT\n        $1::uuid AS organization_id,\n        $2::text AS default_chat_model_id,\n        $3::uuid AS default_provider_configuration_id,\n        $4::text AS default_fast_model_id,\n        $5::uuid AS default_fast_provider_configuration_id\n),\nselected_chat_route AS (\n    SELECT\n        r.id,\n        requested_defaults.organization_id,\n        requested_defaults.default_chat_model_id,\n        requested_defaults.default_provider_configuration_id,\n        requested_defaults.default_fast_model_id,\n        requested_defaults.default_fast_provider_configuration_id\n    FROM requested_defaults\n             JOIN weave.inference_model_route r\n                  ON r.organization_id = requested_defaults.organization_id\n             JOIN weave.inference_model im\n                  ON im.organization_id = r.organization_id\n                      AND im.id = r.inference_model_id\n    WHERE im.model_id = requested_defaults.default_chat_model_id\n      AND r.inference_provider_configuration_id =\n          requested_defaults.default_provider_configuration_id\n),\nselected_fast_route AS (\n    SELECT r.id\n    FROM requested_defaults\n             JOIN weave.inference_model_route r\n                  ON r.organization_id = requested_defaults.organization_id\n             JOIN weave.inference_model im\n                  ON im.organization_id = r.organization_id\n                      AND im.id = r.inference_model_id\n    WHERE requested_defaults.default_fast_model_id IS NOT NULL\n      AND requested_defaults.default_fast_provider_configuration_id IS NOT NULL\n      AND im.model_id = requested_defaults.default_fast_model_id\n      AND r.inference_provider_configuration_id =\n          requested_defaults.default_fast_provider_configuration_id\n),\nupdated_organization AS (\n    UPDATE weave.organization o\n    SET default_chat_model_route_id = selected_chat_route.id,\n        default_fast_model_route_id = COALESCE(\n            selected_fast_route.id,\n            o.default_fast_model_route_id\n        )\n    FROM selected_chat_route\n             LEFT JOIN selected_fast_route ON TRUE\n    WHERE o.id = selected_chat_route.organization_id\n      AND (\n        (\n            selected_chat_route.default_fast_model_id IS NULL\n                AND selected_chat_route.default_fast_provider_configuration_id IS NULL\n            )\n            OR selected_fast_route.id IS NOT NULL\n        )\n    RETURNING\n        o.id AS organization_id,\n        o.default_chat_model_route_id,\n        o.default_fast_model_route_id\n)\nSELECT\n    updated_organization.organization_id,\n    chat_im.model_id AS default_chat_model_id,\n    chat_r.inference_provider_configuration_id AS default_provider_configuration_id,\n    fast_im.model_id AS default_fast_model_id,\n    fast_r.inference_provider_configuration_id AS default_fast_provider_configuration_id\nFROM updated_organization\n         JOIN weave.inference_model_route chat_r\n              ON chat_r.organization_id = updated_organization.organization_id\n                  AND chat_r.id = updated_organization.default_chat_model_route_id\n         JOIN weave.inference_model chat_im\n              ON chat_im.organization_id = chat_r.organization_id\n                  AND chat_im.id = chat_r.inference_model_id\n         LEFT JOIN weave.inference_model_route fast_r\n                   ON fast_r.organization_id = updated_organization.organization_id\n                       AND fast_r.id = updated_organization.default_fast_model_route_id\n         LEFT JOIN weave.inference_model fast_im\n                   ON fast_im.organization_id = fast_r.organization_id\n                       AND fast_im.id = fast_r.inference_model_id";
+export interface SetOrganizationChatSettingsArgs {
+    organizationId: string;
+    defaultChatModelId: string;
+    defaultProviderConfigurationId: string;
+    defaultFastModelId: string | null;
+    defaultFastProviderConfigurationId: string | null;
+}
+export interface SetOrganizationChatSettingsRow {
+    organizationId: string;
+    defaultChatModelId: string;
+    defaultProviderConfigurationId: string;
+    defaultFastModelId: string | null;
+    defaultFastProviderConfigurationId: string | null;
+}
+export declare function setOrganizationChatSettings(client: Client, args: SetOrganizationChatSettingsArgs): Promise<SetOrganizationChatSettingsRow | null>;
+export declare const getModelRuntimeConfigurationQuery = "-- name: GetModelRuntimeConfiguration :one\nSELECT\n    imr.id AS inference_model_route_id,\n    imr.organization_id,\n    im.id AS inference_model_id,\n    im.model_id,\n    imr.provider_model_id,\n    imr.priority,\n    p.id AS inference_provider_configuration_id,\n    p.kind AS provider_kind,\n    p.name AS provider_name,\n    p.slug AS provider_slug,\n    p.base_url AS provider_base_url,\n    c.credential_kind,\n    c.encrypted_credentials,\n    c.credential_hint,\n    c.key_id,\n    c.last_rotated_at\nFROM weave.inference_model_route imr\n         JOIN weave.inference_model im\n              ON im.organization_id = imr.organization_id\n                  AND im.id = imr.inference_model_id\n         JOIN weave.inference_provider_configuration p\n              ON p.organization_id = imr.organization_id\n                  AND p.id = imr.inference_provider_configuration_id\n         JOIN weave.inference_provider_credential c\n              ON c.inference_provider_configuration_id = p.id\nWHERE imr.organization_id = $1\n  AND im.model_id = $2\nORDER BY imr.priority ASC\nLIMIT 1";
+export interface GetModelRuntimeConfigurationArgs {
+    organizationId: string;
+    modelId: string;
+}
+export interface GetModelRuntimeConfigurationRow {
+    inferenceModelRouteId: string;
+    organizationId: string;
+    inferenceModelId: string;
+    modelId: string;
+    providerModelId: string;
+    priority: number;
+    inferenceProviderConfigurationId: string;
+    providerKind: string;
+    providerName: string;
+    providerSlug: string;
+    providerBaseUrl: string | null;
+    credentialKind: string;
+    encryptedCredentials: Buffer;
+    credentialHint: string;
+    keyId: string;
+    lastRotatedAt: Date;
+}
+export declare function getModelRuntimeConfiguration(client: Client, args: GetModelRuntimeConfigurationArgs): Promise<GetModelRuntimeConfigurationRow | null>;
+export {};

package/dist/weavesql/weavedb/inference_model_route_sql.js

@@ -0,0 +1,483 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getModelRuntimeConfigurationQuery = exports.setOrganizationChatSettingsQuery = exports.getOrganizationChatSettingsQuery = exports.listInferenceModelRoutesQuery = exports.deleteInferenceModelRouteQuery = exports.updateInferenceModelRouteQuery = exports.getInferenceModelRouteQuery = exports.createInferenceModelRouteQuery = exports.getInferenceModelByModelIDQuery = exports.upsertInferenceModelQuery = void 0;
+exports.upsertInferenceModel = upsertInferenceModel;
+exports.getInferenceModelByModelID = getInferenceModelByModelID;
+exports.createInferenceModelRoute = createInferenceModelRoute;
+exports.getInferenceModelRoute = getInferenceModelRoute;
+exports.updateInferenceModelRoute = updateInferenceModelRoute;
+exports.deleteInferenceModelRoute = deleteInferenceModelRoute;
+exports.listInferenceModelRoutes = listInferenceModelRoutes;
+exports.getOrganizationChatSettings = getOrganizationChatSettings;
+exports.setOrganizationChatSettings = setOrganizationChatSettings;
+exports.getModelRuntimeConfiguration = getModelRuntimeConfiguration;
+exports.upsertInferenceModelQuery = `-- name: UpsertInferenceModel :one
+WITH inserted AS (
+    INSERT INTO weave.inference_model (
+        organization_id,
+        model_id
+    ) VALUES (
+        $1,
+        $2
+    )
+    ON CONFLICT (organization_id, model_id) DO NOTHING
+    RETURNING
+        id,
+        organization_id,
+        model_id,
+        created_at,
+        updated_at,
+        true::boolean AS created
+)
+SELECT
+    id,
+    organization_id,
+    model_id,
+    created_at,
+    updated_at,
+    created
+FROM inserted
+UNION ALL
+SELECT
+    im.id,
+    im.organization_id,
+    im.model_id,
+    im.created_at,
+    im.updated_at,
+    false::boolean AS created
+FROM weave.inference_model im
+WHERE im.organization_id = $1
+  AND im.model_id = $2
+  AND NOT EXISTS (SELECT 1 FROM inserted)
+LIMIT 1`;
+async function upsertInferenceModel(client, args) {
+    const result = await client.query({
+        text: exports.upsertInferenceModelQuery,
+        values: [args.organizationId, args.modelId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        modelId: row[2],
+        createdAt: row[3],
+        updatedAt: row[4],
+        created: row[5]
+    };
+}
+exports.getInferenceModelByModelIDQuery = `-- name: GetInferenceModelByModelID :one
+SELECT id, organization_id, model_id, created_at, updated_at
+FROM weave.inference_model
+WHERE organization_id = $1
+  AND model_id = $2`;
+async function getInferenceModelByModelID(client, args) {
+    const result = await client.query({
+        text: exports.getInferenceModelByModelIDQuery,
+        values: [args.organizationId, args.modelId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        modelId: row[2],
+        createdAt: row[3],
+        updatedAt: row[4]
+    };
+}
+exports.createInferenceModelRouteQuery = `-- name: CreateInferenceModelRoute :one
+WITH created AS (
+    INSERT INTO weave.inference_model_route (
+        organization_id,
+        inference_model_id,
+        inference_provider_configuration_id,
+        provider_model_id,
+        priority
+    ) VALUES (
+        $1,
+        $2,
+        $3,
+        $4,
+        COALESCE($5::integer, 100)
+    )
+    RETURNING id, organization_id, inference_model_id, inference_provider_configuration_id, provider_model_id, priority, created_at, updated_at
+)
+SELECT
+    r.id,
+    r.organization_id,
+    r.inference_model_id,
+    im.model_id,
+    r.inference_provider_configuration_id,
+    r.provider_model_id,
+    r.priority,
+    r.created_at,
+    r.updated_at
+FROM created r
+         JOIN weave.inference_model im
+              ON im.organization_id = r.organization_id
+                  AND im.id = r.inference_model_id`;
+async function createInferenceModelRoute(client, args) {
+    const result = await client.query({
+        text: exports.createInferenceModelRouteQuery,
+        values: [args.organizationId, args.inferenceModelId, args.inferenceProviderConfigurationId, args.providerModelId, args.priority],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        inferenceModelId: row[2],
+        modelId: row[3],
+        inferenceProviderConfigurationId: row[4],
+        providerModelId: row[5],
+        priority: row[6],
+        createdAt: row[7],
+        updatedAt: row[8]
+    };
+}
+exports.getInferenceModelRouteQuery = `-- name: GetInferenceModelRoute :one
+SELECT
+    r.id,
+    r.organization_id,
+    r.inference_model_id,
+    im.model_id,
+    r.inference_provider_configuration_id,
+    r.provider_model_id,
+    r.priority,
+    r.created_at,
+    r.updated_at
+FROM weave.inference_model_route r
+         JOIN weave.inference_model im
+              ON im.organization_id = r.organization_id
+                  AND im.id = r.inference_model_id
+WHERE r.organization_id = $1
+  AND r.id = $2`;
+async function getInferenceModelRoute(client, args) {
+    const result = await client.query({
+        text: exports.getInferenceModelRouteQuery,
+        values: [args.organizationId, args.inferenceModelRouteId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        inferenceModelId: row[2],
+        modelId: row[3],
+        inferenceProviderConfigurationId: row[4],
+        providerModelId: row[5],
+        priority: row[6],
+        createdAt: row[7],
+        updatedAt: row[8]
+    };
+}
+exports.updateInferenceModelRouteQuery = `-- name: UpdateInferenceModelRoute :one
+WITH updated AS (
+    UPDATE weave.inference_model_route
+    SET inference_model_id = $1,
+        inference_provider_configuration_id = $2,
+        provider_model_id = $3,
+        priority = $4
+    WHERE weave.inference_model_route.organization_id = $5
+      AND weave.inference_model_route.id = $6
+    RETURNING id, organization_id, inference_model_id, inference_provider_configuration_id, provider_model_id, priority, created_at, updated_at
+)
+SELECT
+    r.id,
+    r.organization_id,
+    r.inference_model_id,
+    im.model_id,
+    r.inference_provider_configuration_id,
+    r.provider_model_id,
+    r.priority,
+    r.created_at,
+    r.updated_at
+FROM updated r
+         JOIN weave.inference_model im
+              ON im.organization_id = r.organization_id
+                  AND im.id = r.inference_model_id`;
+async function updateInferenceModelRoute(client, args) {
+    const result = await client.query({
+        text: exports.updateInferenceModelRouteQuery,
+        values: [args.inferenceModelId, args.inferenceProviderConfigurationId, args.providerModelId, args.priority, args.organizationId, args.inferenceModelRouteId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        inferenceModelId: row[2],
+        modelId: row[3],
+        inferenceProviderConfigurationId: row[4],
+        providerModelId: row[5],
+        priority: row[6],
+        createdAt: row[7],
+        updatedAt: row[8]
+    };
+}
+exports.deleteInferenceModelRouteQuery = `-- name: DeleteInferenceModelRoute :exec
+DELETE FROM weave.inference_model_route
+WHERE organization_id = $1
+  AND id = $2`;
+async function deleteInferenceModelRoute(client, args) {
+    await client.query({
+        text: exports.deleteInferenceModelRouteQuery,
+        values: [args.organizationId, args.inferenceModelRouteId],
+        rowMode: "array"
+    });
+}
+exports.listInferenceModelRoutesQuery = `-- name: ListInferenceModelRoutes :many
+SELECT
+    r.id,
+    r.organization_id,
+    r.inference_model_id,
+    im.model_id,
+    r.inference_provider_configuration_id,
+    r.provider_model_id,
+    r.priority,
+    r.created_at,
+    r.updated_at
+FROM weave.inference_model_route r
+         JOIN weave.inference_model im
+              ON im.organization_id = r.organization_id
+                  AND im.id = r.inference_model_id
+WHERE r.organization_id = $1
+  AND ($2::text IS NULL OR im.model_id = $2::text)
+ORDER BY im.model_id ASC, r.priority ASC, r.id ASC
+LIMIT COALESCE($4::integer, 100)
+OFFSET COALESCE($3::integer, 0)`;
+async function listInferenceModelRoutes(client, args) {
+    const result = await client.query({
+        text: exports.listInferenceModelRoutesQuery,
+        values: [args.organizationId, args.modelId, args.pageOffset, args.pageSize],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            organizationId: row[1],
+            inferenceModelId: row[2],
+            modelId: row[3],
+            inferenceProviderConfigurationId: row[4],
+            providerModelId: row[5],
+            priority: row[6],
+            createdAt: row[7],
+            updatedAt: row[8]
+        };
+    });
+}
+exports.getOrganizationChatSettingsQuery = `-- name: GetOrganizationChatSettings :one
+SELECT
+    o.id AS organization_id,
+    chat_im.model_id AS default_chat_model_id,
+    chat_r.inference_provider_configuration_id AS default_provider_configuration_id,
+    fast_im.model_id AS default_fast_model_id,
+    fast_r.inference_provider_configuration_id AS default_fast_provider_configuration_id
+FROM weave.organization o
+         LEFT JOIN weave.inference_model_route chat_r
+                   ON chat_r.organization_id = o.id
+                       AND chat_r.id = o.default_chat_model_route_id
+         LEFT JOIN weave.inference_model chat_im
+                   ON chat_im.organization_id = chat_r.organization_id
+                       AND chat_im.id = chat_r.inference_model_id
+         LEFT JOIN weave.inference_model_route fast_r
+                   ON fast_r.organization_id = o.id
+                       AND fast_r.id = o.default_fast_model_route_id
+         LEFT JOIN weave.inference_model fast_im
+                   ON fast_im.organization_id = fast_r.organization_id
+                       AND fast_im.id = fast_r.inference_model_id
+WHERE o.id = $1`;
+async function getOrganizationChatSettings(client, args) {
+    const result = await client.query({
+        text: exports.getOrganizationChatSettingsQuery,
+        values: [args.organizationId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        organizationId: row[0],
+        defaultChatModelId: row[1],
+        defaultProviderConfigurationId: row[2],
+        defaultFastModelId: row[3],
+        defaultFastProviderConfigurationId: row[4]
+    };
+}
+exports.setOrganizationChatSettingsQuery = `-- name: SetOrganizationChatSettings :one
+WITH requested_defaults AS (
+    SELECT
+        $1::uuid AS organization_id,
+        $2::text AS default_chat_model_id,
+        $3::uuid AS default_provider_configuration_id,
+        $4::text AS default_fast_model_id,
+        $5::uuid AS default_fast_provider_configuration_id
+),
+selected_chat_route AS (
+    SELECT
+        r.id,
+        requested_defaults.organization_id,
+        requested_defaults.default_chat_model_id,
+        requested_defaults.default_provider_configuration_id,
+        requested_defaults.default_fast_model_id,
+        requested_defaults.default_fast_provider_configuration_id
+    FROM requested_defaults
+             JOIN weave.inference_model_route r
+                  ON r.organization_id = requested_defaults.organization_id
+             JOIN weave.inference_model im
+                  ON im.organization_id = r.organization_id
+                      AND im.id = r.inference_model_id
+    WHERE im.model_id = requested_defaults.default_chat_model_id
+      AND r.inference_provider_configuration_id =
+          requested_defaults.default_provider_configuration_id
+),
+selected_fast_route AS (
+    SELECT r.id
+    FROM requested_defaults
+             JOIN weave.inference_model_route r
+                  ON r.organization_id = requested_defaults.organization_id
+             JOIN weave.inference_model im
+                  ON im.organization_id = r.organization_id
+                      AND im.id = r.inference_model_id
+    WHERE requested_defaults.default_fast_model_id IS NOT NULL
+      AND requested_defaults.default_fast_provider_configuration_id IS NOT NULL
+      AND im.model_id = requested_defaults.default_fast_model_id
+      AND r.inference_provider_configuration_id =
+          requested_defaults.default_fast_provider_configuration_id
+),
+updated_organization AS (
+    UPDATE weave.organization o
+    SET default_chat_model_route_id = selected_chat_route.id,
+        default_fast_model_route_id = COALESCE(
+            selected_fast_route.id,
+            o.default_fast_model_route_id
+        )
+    FROM selected_chat_route
+             LEFT JOIN selected_fast_route ON TRUE
+    WHERE o.id = selected_chat_route.organization_id
+      AND (
+        (
+            selected_chat_route.default_fast_model_id IS NULL
+                AND selected_chat_route.default_fast_provider_configuration_id IS NULL
+            )
+            OR selected_fast_route.id IS NOT NULL
+        )
+    RETURNING
+        o.id AS organization_id,
+        o.default_chat_model_route_id,
+        o.default_fast_model_route_id
+)
+SELECT
+    updated_organization.organization_id,
+    chat_im.model_id AS default_chat_model_id,
+    chat_r.inference_provider_configuration_id AS default_provider_configuration_id,
+    fast_im.model_id AS default_fast_model_id,
+    fast_r.inference_provider_configuration_id AS default_fast_provider_configuration_id
+FROM updated_organization
+         JOIN weave.inference_model_route chat_r
+              ON chat_r.organization_id = updated_organization.organization_id
+                  AND chat_r.id = updated_organization.default_chat_model_route_id
+         JOIN weave.inference_model chat_im
+              ON chat_im.organization_id = chat_r.organization_id
+                  AND chat_im.id = chat_r.inference_model_id
+         LEFT JOIN weave.inference_model_route fast_r
+                   ON fast_r.organization_id = updated_organization.organization_id
+                       AND fast_r.id = updated_organization.default_fast_model_route_id
+         LEFT JOIN weave.inference_model fast_im
+                   ON fast_im.organization_id = fast_r.organization_id
+                       AND fast_im.id = fast_r.inference_model_id`;
+async function setOrganizationChatSettings(client, args) {
+    const result = await client.query({
+        text: exports.setOrganizationChatSettingsQuery,
+        values: [args.organizationId, args.defaultChatModelId, args.defaultProviderConfigurationId, args.defaultFastModelId, args.defaultFastProviderConfigurationId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        organizationId: row[0],
+        defaultChatModelId: row[1],
+        defaultProviderConfigurationId: row[2],
+        defaultFastModelId: row[3],
+        defaultFastProviderConfigurationId: row[4]
+    };
+}
+exports.getModelRuntimeConfigurationQuery = `-- name: GetModelRuntimeConfiguration :one
+SELECT
+    imr.id AS inference_model_route_id,
+    imr.organization_id,
+    im.id AS inference_model_id,
+    im.model_id,
+    imr.provider_model_id,
+    imr.priority,
+    p.id AS inference_provider_configuration_id,
+    p.kind AS provider_kind,
+    p.name AS provider_name,
+    p.slug AS provider_slug,
+    p.base_url AS provider_base_url,
+    c.credential_kind,
+    c.encrypted_credentials,
+    c.credential_hint,
+    c.key_id,
+    c.last_rotated_at
+FROM weave.inference_model_route imr
+         JOIN weave.inference_model im
+              ON im.organization_id = imr.organization_id
+                  AND im.id = imr.inference_model_id
+         JOIN weave.inference_provider_configuration p
+              ON p.organization_id = imr.organization_id
+                  AND p.id = imr.inference_provider_configuration_id
+         JOIN weave.inference_provider_credential c
+              ON c.inference_provider_configuration_id = p.id
+WHERE imr.organization_id = $1
+  AND im.model_id = $2
+ORDER BY imr.priority ASC
+LIMIT 1`;
+async function getModelRuntimeConfiguration(client, args) {
+    const result = await client.query({
+        text: exports.getModelRuntimeConfigurationQuery,
+        values: [args.organizationId, args.modelId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        inferenceModelRouteId: row[0],
+        organizationId: row[1],
+        inferenceModelId: row[2],
+        modelId: row[3],
+        providerModelId: row[4],
+        priority: row[5],
+        inferenceProviderConfigurationId: row[6],
+        providerKind: row[7],
+        providerName: row[8],
+        providerSlug: row[9],
+        providerBaseUrl: row[10],
+        credentialKind: row[11],
+        encryptedCredentials: row[12],
+        credentialHint: row[13],
+        keyId: row[14],
+        lastRotatedAt: row[15]
+    };
+}