weave-typescript 0.51.3 → 0.52.0

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

@@ -0,0 +1,161 @@
+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    im.model_id AS default_chat_model_id,\n    r.inference_provider_configuration_id AS default_provider_configuration_id\nFROM weave.organization o\n         LEFT JOIN weave.inference_model_route r\n                   ON r.organization_id = o.id\n                       AND r.id = o.default_chat_model_route_id\n         LEFT JOIN weave.inference_model im\n                   ON im.organization_id = r.organization_id\n                       AND im.id = r.inference_model_id\nWHERE o.id = $1";
+export interface GetOrganizationChatSettingsArgs {
+    organizationId: string;
+}
+export interface GetOrganizationChatSettingsRow {
+    organizationId: string;
+    defaultChatModelId: string | null;
+    defaultProviderConfigurationId: string | null;
+}
+export declare function getOrganizationChatSettings(client: Client, args: GetOrganizationChatSettingsArgs): Promise<GetOrganizationChatSettingsRow | null>;
+export declare const setOrganizationChatSettingsQuery = "-- name: SetOrganizationChatSettings :one\nWITH selected_route AS (\n    SELECT r.id\n    FROM 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\n    WHERE r.organization_id = $1\n      AND im.model_id = $2\n      AND r.inference_provider_configuration_id = $3\n)\nUPDATE weave.organization o\nSET default_chat_model_route_id = selected_route.id\nFROM selected_route\nWHERE o.id = $1\nRETURNING\n    o.id AS organization_id,\n    $2::text AS default_chat_model_id,\n    $3::uuid AS default_provider_configuration_id";
+export interface SetOrganizationChatSettingsArgs {
+    organizationId: string;
+    defaultChatModelId: string;
+    defaultProviderConfigurationId: string;
+}
+export interface SetOrganizationChatSettingsRow {
+    organizationId: string;
+    defaultChatModelId: string;
+    defaultProviderConfigurationId: string;
+}
+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,408 @@
+"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,
+    im.model_id AS default_chat_model_id,
+    r.inference_provider_configuration_id AS default_provider_configuration_id
+FROM weave.organization o
+         LEFT JOIN weave.inference_model_route r
+                   ON r.organization_id = o.id
+                       AND r.id = o.default_chat_model_route_id
+         LEFT JOIN weave.inference_model im
+                   ON im.organization_id = r.organization_id
+                       AND im.id = 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]
+    };
+}
+exports.setOrganizationChatSettingsQuery = `-- name: SetOrganizationChatSettings :one
+WITH selected_route AS (
+    SELECT r.id
+    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 im.model_id = $2
+      AND r.inference_provider_configuration_id = $3
+)
+UPDATE weave.organization o
+SET default_chat_model_route_id = selected_route.id
+FROM selected_route
+WHERE o.id = $1
+RETURNING
+    o.id AS organization_id,
+    $2::text AS default_chat_model_id,
+    $3::uuid AS default_provider_configuration_id`;
+async function setOrganizationChatSettings(client, args) {
+    const result = await client.query({
+        text: exports.setOrganizationChatSettingsQuery,
+        values: [args.organizationId, args.defaultChatModelId, args.defaultProviderConfigurationId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        organizationId: row[0],
+        defaultChatModelId: row[1],
+        defaultProviderConfigurationId: row[2]
+    };
+}
+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]
+    };
+}