fix api config auth and image model compatibility
Some checks failed
Build & Push Docker Image / build-and-push (push) Has been cancelled
Some checks failed
Build & Push Docker Image / build-and-push (push) Has been cancelled
This commit is contained in:
@@ -11,7 +11,7 @@ services:
|
|||||||
ports:
|
ports:
|
||||||
- "13306:3306"
|
- "13306:3306"
|
||||||
volumes:
|
volumes:
|
||||||
- mysql_data:/var/lib/mysql
|
- /docker/waoowaoo/backup/mysql_data:/var/lib/mysql
|
||||||
command:
|
command:
|
||||||
- "--default-authentication-plugin=mysql_native_password"
|
- "--default-authentication-plugin=mysql_native_password"
|
||||||
- "--sql_mode=STRICT_TRANS_TABLES,NO_ENGINE_SUBSTITUTION"
|
- "--sql_mode=STRICT_TRANS_TABLES,NO_ENGINE_SUBSTITUTION"
|
||||||
@@ -30,7 +30,7 @@ services:
|
|||||||
ports:
|
ports:
|
||||||
- "16379:6379"
|
- "16379:6379"
|
||||||
volumes:
|
volumes:
|
||||||
- redis_data:/data
|
- /docker/waoowaoo/backup/redis_data:/data
|
||||||
command: ["redis-server", "--appendonly", "yes"]
|
command: ["redis-server", "--appendonly", "yes"]
|
||||||
healthcheck:
|
healthcheck:
|
||||||
test: ["CMD", "redis-cli", "ping"]
|
test: ["CMD", "redis-cli", "ping"]
|
||||||
@@ -52,7 +52,7 @@ services:
|
|||||||
- "19000:9000"
|
- "19000:9000"
|
||||||
- "19001:9001"
|
- "19001:9001"
|
||||||
volumes:
|
volumes:
|
||||||
- minio_data:/data
|
- /docker/waoowaoo/backup/minio:/data
|
||||||
healthcheck:
|
healthcheck:
|
||||||
test: ["CMD", "curl", "-f", "http://127.0.0.1:9000/minio/health/live"]
|
test: ["CMD", "curl", "-f", "http://127.0.0.1:9000/minio/health/live"]
|
||||||
interval: 5s
|
interval: 5s
|
||||||
@@ -62,7 +62,10 @@ services:
|
|||||||
|
|
||||||
# ==================== App (Next.js + Workers) ====================
|
# ==================== App (Next.js + Workers) ====================
|
||||||
app:
|
app:
|
||||||
image: ghcr.io/saturndec/waoowaoo:latest
|
build:
|
||||||
|
context: .
|
||||||
|
dockerfile: Dockerfile
|
||||||
|
image: waoowaoo:local
|
||||||
container_name: waoowaoo-app
|
container_name: waoowaoo-app
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
environment:
|
environment:
|
||||||
@@ -120,8 +123,8 @@ services:
|
|||||||
- "13000:3000"
|
- "13000:3000"
|
||||||
- "13010:3010"
|
- "13010:3010"
|
||||||
volumes:
|
volumes:
|
||||||
- ./data:/app/data
|
- /docker/waoowaoo/backup/data:/app/data
|
||||||
- ./docker-logs:/app/logs
|
- /docker/waoowaoo/backup/docker-logs:/app/logs
|
||||||
depends_on:
|
depends_on:
|
||||||
mysql:
|
mysql:
|
||||||
condition: service_healthy
|
condition: service_healthy
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ import { prisma } from '@/lib/prisma'
|
|||||||
import { requireUserAuth, isErrorResponse } from '@/lib/api-auth'
|
import { requireUserAuth, isErrorResponse } from '@/lib/api-auth'
|
||||||
import { ApiError, apiHandler } from '@/lib/api-errors'
|
import { ApiError, apiHandler } from '@/lib/api-errors'
|
||||||
import { isArtStyleValue } from '@/lib/constants'
|
import { isArtStyleValue } from '@/lib/constants'
|
||||||
|
import { assertPersistedUser } from '@/lib/persisted-user'
|
||||||
|
|
||||||
function validateArtStyleField(value: unknown): string {
|
function validateArtStyleField(value: unknown): string {
|
||||||
if (typeof value !== 'string') {
|
if (typeof value !== 'string') {
|
||||||
@@ -29,6 +30,7 @@ export const GET = apiHandler(async () => {
|
|||||||
const authResult = await requireUserAuth()
|
const authResult = await requireUserAuth()
|
||||||
if (isErrorResponse(authResult)) return authResult
|
if (isErrorResponse(authResult)) return authResult
|
||||||
const { session } = authResult
|
const { session } = authResult
|
||||||
|
await assertPersistedUser(session.user.id)
|
||||||
|
|
||||||
// 获取或创建用户偏好
|
// 获取或创建用户偏好
|
||||||
const preference = await prisma.userPreference.upsert({
|
const preference = await prisma.userPreference.upsert({
|
||||||
@@ -46,6 +48,7 @@ export const PATCH = apiHandler(async (request: NextRequest) => {
|
|||||||
const authResult = await requireUserAuth()
|
const authResult = await requireUserAuth()
|
||||||
if (isErrorResponse(authResult)) return authResult
|
if (isErrorResponse(authResult)) return authResult
|
||||||
const { session } = authResult
|
const { session } = authResult
|
||||||
|
await assertPersistedUser(session.user.id)
|
||||||
|
|
||||||
const body = await request.json()
|
const body = await request.json()
|
||||||
|
|
||||||
|
|||||||
@@ -10,6 +10,7 @@ import { prisma } from '@/lib/prisma'
|
|||||||
import { encryptApiKey, decryptApiKey } from '@/lib/crypto-utils'
|
import { encryptApiKey, decryptApiKey } from '@/lib/crypto-utils'
|
||||||
import { requireUserAuth, isErrorResponse } from '@/lib/api-auth'
|
import { requireUserAuth, isErrorResponse } from '@/lib/api-auth'
|
||||||
import { apiHandler, ApiError } from '@/lib/api-errors'
|
import { apiHandler, ApiError } from '@/lib/api-errors'
|
||||||
|
import { assertPersistedUser } from '@/lib/persisted-user'
|
||||||
import {
|
import {
|
||||||
composeModelKey,
|
composeModelKey,
|
||||||
parseModelKeyStrict,
|
parseModelKeyStrict,
|
||||||
@@ -1654,6 +1655,7 @@ export const GET = apiHandler(async () => {
|
|||||||
if (isErrorResponse(authResult)) return authResult
|
if (isErrorResponse(authResult)) return authResult
|
||||||
const { session } = authResult
|
const { session } = authResult
|
||||||
const userId = session.user.id
|
const userId = session.user.id
|
||||||
|
await assertPersistedUser(userId)
|
||||||
|
|
||||||
const pref = await prisma.userPreference.findUnique({
|
const pref = await prisma.userPreference.findUnique({
|
||||||
where: { userId },
|
where: { userId },
|
||||||
@@ -1766,6 +1768,7 @@ export const PUT = apiHandler(async (request: NextRequest) => {
|
|||||||
if (isErrorResponse(authResult)) return authResult
|
if (isErrorResponse(authResult)) return authResult
|
||||||
const { session } = authResult
|
const { session } = authResult
|
||||||
const userId = session.user.id
|
const userId = session.user.id
|
||||||
|
await assertPersistedUser(userId)
|
||||||
|
|
||||||
let body: ApiConfigPutBody
|
let body: ApiConfigPutBody
|
||||||
try {
|
try {
|
||||||
|
|||||||
@@ -23,6 +23,18 @@ import { generateSiliconFlowAudio, generateSiliconFlowImage, generateSiliconFlow
|
|||||||
|
|
||||||
const OFFICIAL_ONLY_PROVIDER_KEYS = new Set(['bailian', 'siliconflow'])
|
const OFFICIAL_ONLY_PROVIDER_KEYS = new Set(['bailian', 'siliconflow'])
|
||||||
|
|
||||||
|
const OPENAI_COMPAT_IMAGE_MODEL_ID_ALIASES: Record<string, string> = {
|
||||||
|
'gemini-3-pro-image': 'gemini-3-pro-image-preview',
|
||||||
|
'gemini-3.1-flash-image': 'gemini-3.1-flash-image-preview',
|
||||||
|
'gemini-2.5-flash-image': 'gemini-2.5-flash-image-preview',
|
||||||
|
}
|
||||||
|
|
||||||
|
function normalizeOpenAICompatImageModelId(modelId: string): string {
|
||||||
|
const normalized = modelId.trim()
|
||||||
|
if (!normalized) return normalized
|
||||||
|
return OPENAI_COMPAT_IMAGE_MODEL_ID_ALIASES[normalized] || normalized
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 将 aspectRatio 映射为 OpenAI 兼容的 size
|
* 将 aspectRatio 映射为 OpenAI 兼容的 size
|
||||||
*/
|
*/
|
||||||
@@ -105,6 +117,7 @@ export async function generateImage(
|
|||||||
const { referenceImages, ...generatorOptions } = options || {}
|
const { referenceImages, ...generatorOptions } = options || {}
|
||||||
if (gatewayRoute === 'openai-compat') {
|
if (gatewayRoute === 'openai-compat') {
|
||||||
const compatTemplate = selection.compatMediaTemplate
|
const compatTemplate = selection.compatMediaTemplate
|
||||||
|
const normalizedCompatModelId = normalizeOpenAICompatImageModelId(selection.modelId)
|
||||||
if (providerKey === 'openai-compatible' && !compatTemplate) {
|
if (providerKey === 'openai-compatible' && !compatTemplate) {
|
||||||
throw new Error(`MODEL_COMPAT_MEDIA_TEMPLATE_REQUIRED: ${selection.modelKey}`)
|
throw new Error(`MODEL_COMPAT_MEDIA_TEMPLATE_REQUIRED: ${selection.modelKey}`)
|
||||||
}
|
}
|
||||||
@@ -112,7 +125,7 @@ export async function generateImage(
|
|||||||
return await generateImageViaOpenAICompatTemplate({
|
return await generateImageViaOpenAICompatTemplate({
|
||||||
userId,
|
userId,
|
||||||
providerId: selection.provider,
|
providerId: selection.provider,
|
||||||
modelId: selection.modelId,
|
modelId: normalizedCompatModelId,
|
||||||
modelKey: selection.modelKey,
|
modelKey: selection.modelKey,
|
||||||
prompt,
|
prompt,
|
||||||
referenceImages,
|
referenceImages,
|
||||||
@@ -141,7 +154,7 @@ export async function generateImage(
|
|||||||
return await generateImageViaOpenAICompat({
|
return await generateImageViaOpenAICompat({
|
||||||
userId,
|
userId,
|
||||||
providerId: selection.provider,
|
providerId: selection.provider,
|
||||||
modelId: selection.modelId,
|
modelId: normalizedCompatModelId,
|
||||||
prompt,
|
prompt,
|
||||||
referenceImages,
|
referenceImages,
|
||||||
options: {
|
options: {
|
||||||
|
|||||||
31
src/lib/persisted-user.ts
Normal file
31
src/lib/persisted-user.ts
Normal file
@@ -0,0 +1,31 @@
|
|||||||
|
import { ApiError } from '@/lib/api-errors'
|
||||||
|
import { prisma } from '@/lib/prisma'
|
||||||
|
import { withPrismaRetry } from '@/lib/prisma-retry'
|
||||||
|
|
||||||
|
const SESSION_USER_MISSING_MESSAGE = '登录状态已失效,请重新登录'
|
||||||
|
|
||||||
|
export async function assertPersistedUser(userId: string): Promise<void> {
|
||||||
|
const normalizedUserId = typeof userId === 'string' ? userId.trim() : ''
|
||||||
|
if (!normalizedUserId) {
|
||||||
|
throw new ApiError('UNAUTHORIZED', {
|
||||||
|
code: 'SESSION_USER_MISSING',
|
||||||
|
field: 'userId',
|
||||||
|
message: SESSION_USER_MISSING_MESSAGE,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
const user = await withPrismaRetry(() =>
|
||||||
|
prisma.user.findUnique({
|
||||||
|
where: { id: normalizedUserId },
|
||||||
|
select: { id: true },
|
||||||
|
}),
|
||||||
|
)
|
||||||
|
|
||||||
|
if (user) return
|
||||||
|
|
||||||
|
throw new ApiError('UNAUTHORIZED', {
|
||||||
|
code: 'SESSION_USER_MISSING',
|
||||||
|
field: 'userId',
|
||||||
|
message: SESSION_USER_MISSING_MESSAGE,
|
||||||
|
})
|
||||||
|
}
|
||||||
@@ -34,6 +34,9 @@ type SavedProvider = {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const prismaMock = vi.hoisted(() => ({
|
const prismaMock = vi.hoisted(() => ({
|
||||||
|
user: {
|
||||||
|
findUnique: vi.fn<(...args: unknown[]) => Promise<{ id: string } | null>>(),
|
||||||
|
},
|
||||||
userPreference: {
|
userPreference: {
|
||||||
findUnique: vi.fn<(...args: unknown[]) => Promise<UserPreferenceSnapshot | null>>(),
|
findUnique: vi.fn<(...args: unknown[]) => Promise<UserPreferenceSnapshot | null>>(),
|
||||||
upsert: vi.fn<(...args: unknown[]) => Promise<unknown>>(),
|
upsert: vi.fn<(...args: unknown[]) => Promise<unknown>>(),
|
||||||
@@ -103,6 +106,7 @@ describe('api specific - user api-config PUT provider uniqueness', () => {
|
|||||||
vi.clearAllMocks()
|
vi.clearAllMocks()
|
||||||
resetAuthMockState()
|
resetAuthMockState()
|
||||||
|
|
||||||
|
prismaMock.user.findUnique.mockResolvedValue({ id: 'user-1' })
|
||||||
prismaMock.userPreference.findUnique.mockResolvedValue({
|
prismaMock.userPreference.findUnique.mockResolvedValue({
|
||||||
customProviders: null,
|
customProviders: null,
|
||||||
customModels: null,
|
customModels: null,
|
||||||
@@ -111,6 +115,37 @@ describe('api specific - user api-config PUT provider uniqueness', () => {
|
|||||||
getBillingModeMock.mockResolvedValue('OFF')
|
getBillingModeMock.mockResolvedValue('OFF')
|
||||||
})
|
})
|
||||||
|
|
||||||
|
it('returns unauthorized when session user no longer exists before saving providers', async () => {
|
||||||
|
installAuthMocks()
|
||||||
|
mockAuthenticated('user-missing')
|
||||||
|
prismaMock.user.findUnique.mockResolvedValueOnce(null)
|
||||||
|
const route = await import('@/app/api/user/api-config/route')
|
||||||
|
|
||||||
|
const req = buildMockRequest({
|
||||||
|
path: '/api/user/api-config',
|
||||||
|
method: 'PUT',
|
||||||
|
body: {
|
||||||
|
providers: [
|
||||||
|
{ id: 'openai-compatible:oa-1', name: 'OpenAI A', baseUrl: 'https://oa-a.test', apiKey: 'oa-key-a' },
|
||||||
|
],
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
const res = await route.PUT(req, routeContext)
|
||||||
|
const body = await res.json() as {
|
||||||
|
error?: {
|
||||||
|
code?: string
|
||||||
|
message?: string
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
expect(res.status).toBe(401)
|
||||||
|
expect(body.error?.code).toBe('UNAUTHORIZED')
|
||||||
|
expect(body.error?.message).toBe('登录状态已失效,请重新登录')
|
||||||
|
expect(prismaMock.userPreference.findUnique).not.toHaveBeenCalled()
|
||||||
|
expect(prismaMock.userPreference.upsert).not.toHaveBeenCalled()
|
||||||
|
})
|
||||||
|
|
||||||
it('allows multiple providers with the same api type when provider ids differ', async () => {
|
it('allows multiple providers with the same api type when provider ids differ', async () => {
|
||||||
installAuthMocks()
|
installAuthMocks()
|
||||||
mockAuthenticated('user-1')
|
mockAuthenticated('user-1')
|
||||||
|
|||||||
@@ -9,6 +9,9 @@ const authMock = vi.hoisted(() => ({
|
|||||||
}))
|
}))
|
||||||
|
|
||||||
const prismaMock = vi.hoisted(() => ({
|
const prismaMock = vi.hoisted(() => ({
|
||||||
|
user: {
|
||||||
|
findUnique: vi.fn<(...args: unknown[]) => Promise<{ id: string } | null>>(),
|
||||||
|
},
|
||||||
userPreference: {
|
userPreference: {
|
||||||
upsert: vi.fn(async () => ({
|
upsert: vi.fn(async () => ({
|
||||||
userId: 'user-1',
|
userId: 'user-1',
|
||||||
@@ -25,6 +28,7 @@ describe('api specific - user preference art style validation', () => {
|
|||||||
|
|
||||||
beforeEach(() => {
|
beforeEach(() => {
|
||||||
vi.clearAllMocks()
|
vi.clearAllMocks()
|
||||||
|
prismaMock.user.findUnique.mockResolvedValue({ id: 'user-1' })
|
||||||
})
|
})
|
||||||
|
|
||||||
it('accepts valid artStyle and persists normalized value', async () => {
|
it('accepts valid artStyle and persists normalized value', async () => {
|
||||||
@@ -58,4 +62,27 @@ describe('api specific - user preference art style validation', () => {
|
|||||||
expect(body.error.code).toBe('INVALID_PARAMS')
|
expect(body.error.code).toBe('INVALID_PARAMS')
|
||||||
expect(prismaMock.userPreference.upsert).not.toHaveBeenCalled()
|
expect(prismaMock.userPreference.upsert).not.toHaveBeenCalled()
|
||||||
})
|
})
|
||||||
|
|
||||||
|
it('returns unauthorized when session user record is missing', async () => {
|
||||||
|
prismaMock.user.findUnique.mockResolvedValueOnce(null)
|
||||||
|
const mod = await import('@/app/api/user-preference/route')
|
||||||
|
const req = buildMockRequest({
|
||||||
|
path: '/api/user-preference',
|
||||||
|
method: 'PATCH',
|
||||||
|
body: { artStyle: 'realistic' },
|
||||||
|
})
|
||||||
|
|
||||||
|
const res = await mod.PATCH(req, routeContext)
|
||||||
|
const body = await res.json() as {
|
||||||
|
error?: {
|
||||||
|
code?: string
|
||||||
|
message?: string
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
expect(res.status).toBe(401)
|
||||||
|
expect(body.error?.code).toBe('UNAUTHORIZED')
|
||||||
|
expect(body.error?.message).toBe('登录状态已失效,请重新登录')
|
||||||
|
expect(prismaMock.userPreference.upsert).not.toHaveBeenCalled()
|
||||||
|
})
|
||||||
})
|
})
|
||||||
|
|||||||
98
tests/regression/user-session-persistence-guard.test.ts
Normal file
98
tests/regression/user-session-persistence-guard.test.ts
Normal file
@@ -0,0 +1,98 @@
|
|||||||
|
import { beforeEach, describe, expect, it, vi } from 'vitest'
|
||||||
|
import { buildMockRequest } from '../helpers/request'
|
||||||
|
import {
|
||||||
|
installAuthMocks,
|
||||||
|
mockAuthenticated,
|
||||||
|
resetAuthMockState,
|
||||||
|
} from '../helpers/auth'
|
||||||
|
|
||||||
|
const prismaMock = vi.hoisted(() => ({
|
||||||
|
user: {
|
||||||
|
findUnique: vi.fn<(...args: unknown[]) => Promise<{ id: string } | null>>(),
|
||||||
|
},
|
||||||
|
userPreference: {
|
||||||
|
findUnique: vi.fn<(...args: unknown[]) => Promise<{ customProviders: null; customModels: null } | null>>(),
|
||||||
|
upsert: vi.fn<(...args: unknown[]) => Promise<unknown>>(),
|
||||||
|
},
|
||||||
|
}))
|
||||||
|
|
||||||
|
const getBillingModeMock = vi.hoisted(() => vi.fn(async () => 'OFF'))
|
||||||
|
|
||||||
|
vi.mock('@/lib/prisma', () => ({
|
||||||
|
prisma: prismaMock,
|
||||||
|
}))
|
||||||
|
|
||||||
|
vi.mock('@/lib/billing/mode', () => ({
|
||||||
|
getBillingMode: getBillingModeMock,
|
||||||
|
}))
|
||||||
|
|
||||||
|
describe('regression - persisted session user guard', () => {
|
||||||
|
const routeContext = { params: Promise.resolve({}) }
|
||||||
|
|
||||||
|
beforeEach(() => {
|
||||||
|
vi.resetModules()
|
||||||
|
vi.clearAllMocks()
|
||||||
|
resetAuthMockState()
|
||||||
|
installAuthMocks()
|
||||||
|
mockAuthenticated('user-missing')
|
||||||
|
prismaMock.user.findUnique.mockResolvedValue(null)
|
||||||
|
prismaMock.userPreference.findUnique.mockResolvedValue({
|
||||||
|
customProviders: null,
|
||||||
|
customModels: null,
|
||||||
|
})
|
||||||
|
prismaMock.userPreference.upsert.mockResolvedValue({ id: 'pref-1' })
|
||||||
|
getBillingModeMock.mockResolvedValue('OFF')
|
||||||
|
})
|
||||||
|
|
||||||
|
it('api-config PUT returns unauthorized before touching userPreference when session user row is missing', async () => {
|
||||||
|
const route = await import('@/app/api/user/api-config/route')
|
||||||
|
const req = buildMockRequest({
|
||||||
|
path: '/api/user/api-config',
|
||||||
|
method: 'PUT',
|
||||||
|
body: {
|
||||||
|
providers: [
|
||||||
|
{
|
||||||
|
id: 'openai-compatible:oa-1',
|
||||||
|
name: 'OpenAI A',
|
||||||
|
baseUrl: 'https://oa-a.test',
|
||||||
|
apiKey: 'oa-key-a',
|
||||||
|
},
|
||||||
|
],
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
const res = await route.PUT(req, routeContext)
|
||||||
|
const body = await res.json() as { error?: { code?: string; message?: string } }
|
||||||
|
|
||||||
|
expect(res.status).toBe(401)
|
||||||
|
expect(body.error?.code).toBe('UNAUTHORIZED')
|
||||||
|
expect(body.error?.message).toBe('登录状态已失效,请重新登录')
|
||||||
|
expect(prismaMock.user.findUnique).toHaveBeenCalledWith({
|
||||||
|
where: { id: 'user-missing' },
|
||||||
|
select: { id: true },
|
||||||
|
})
|
||||||
|
expect(prismaMock.userPreference.findUnique).not.toHaveBeenCalled()
|
||||||
|
expect(prismaMock.userPreference.upsert).not.toHaveBeenCalled()
|
||||||
|
})
|
||||||
|
|
||||||
|
it('user-preference PATCH returns unauthorized before upsert when session user row is missing', async () => {
|
||||||
|
const route = await import('@/app/api/user-preference/route')
|
||||||
|
const req = buildMockRequest({
|
||||||
|
path: '/api/user-preference',
|
||||||
|
method: 'PATCH',
|
||||||
|
body: { artStyle: 'realistic' },
|
||||||
|
})
|
||||||
|
|
||||||
|
const res = await route.PATCH(req, routeContext)
|
||||||
|
const body = await res.json() as { error?: { code?: string; message?: string } }
|
||||||
|
|
||||||
|
expect(res.status).toBe(401)
|
||||||
|
expect(body.error?.code).toBe('UNAUTHORIZED')
|
||||||
|
expect(body.error?.message).toBe('登录状态已失效,请重新登录')
|
||||||
|
expect(prismaMock.user.findUnique).toHaveBeenCalledWith({
|
||||||
|
where: { id: 'user-missing' },
|
||||||
|
select: { id: true },
|
||||||
|
})
|
||||||
|
expect(prismaMock.userPreference.upsert).not.toHaveBeenCalled()
|
||||||
|
})
|
||||||
|
})
|
||||||
@@ -118,6 +118,29 @@ describe('generator-api gateway routing', () => {
|
|||||||
expect(result).toEqual({ success: true, imageUrl: 'compat-template-image' })
|
expect(result).toEqual({ success: true, imageUrl: 'compat-template-image' })
|
||||||
})
|
})
|
||||||
|
|
||||||
|
it('normalizes legacy openai-compatible gemini image model ids before template gateway call', async () => {
|
||||||
|
resolveModelSelectionMock.mockResolvedValueOnce({
|
||||||
|
provider: 'openai-compatible:oa-1',
|
||||||
|
modelId: 'gemini-3.1-flash-image',
|
||||||
|
modelKey: 'openai-compatible:oa-1::gemini-3.1-flash-image',
|
||||||
|
mediaType: 'image',
|
||||||
|
compatMediaTemplate: {
|
||||||
|
version: 1,
|
||||||
|
mediaType: 'image',
|
||||||
|
mode: 'sync',
|
||||||
|
create: { method: 'POST', path: '/v1/images/generations' },
|
||||||
|
response: { outputUrlPath: '$.data[0].url' },
|
||||||
|
},
|
||||||
|
})
|
||||||
|
resolveModelGatewayRouteMock.mockReturnValueOnce('openai-compat')
|
||||||
|
|
||||||
|
await generateImage('user-1', 'openai-compatible:oa-1::gemini-3.1-flash-image', 'draw hero')
|
||||||
|
|
||||||
|
expect(generateImageViaOpenAICompatTemplateMock).toHaveBeenCalledWith(expect.objectContaining({
|
||||||
|
modelId: 'gemini-3.1-flash-image-preview',
|
||||||
|
}))
|
||||||
|
})
|
||||||
|
|
||||||
it('routes official image requests to provider generator', async () => {
|
it('routes official image requests to provider generator', async () => {
|
||||||
resolveModelSelectionMock.mockResolvedValueOnce({
|
resolveModelSelectionMock.mockResolvedValueOnce({
|
||||||
provider: 'google',
|
provider: 'google',
|
||||||
|
|||||||
Reference in New Issue
Block a user