fixes for imports

This commit is contained in:
dal 2025-09-19 10:31:46 -06:00
parent 101d4902bf
commit 84314ebc13
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
6 changed files with 21 additions and 18 deletions

View File

@ -1,4 +1,4 @@
import type { User } from '@buster/database';
import type { User } from '@buster/database/queries';
import { Hono } from 'hono';
import { sample } from './sample';

View File

@ -1,4 +1,4 @@
import type { User } from '@buster/database';
import type { User } from '@buster/database/queries';
import { GetDatasetSampleParamsSchema } from '@buster/server-shared';
import type { GetDatasetSampleResponse } from '@buster/server-shared';
import { zValidator } from '@hono/zod-validator';

View File

@ -1,8 +1,8 @@
import { AccessControlError, ensureDatasetAccess } from '@buster/access-controls';
import { executeSampleQuery } from '@buster/data-source';
import type { Credentials } from '@buster/data-source';
import type { User } from '@buster/database';
import { getDataSourceById, getDataSourceCredentials } from '@buster/database';
import type { User } from '@buster/database/queries';
import { getDataSourceById, getDataSourceCredentials } from '@buster/database/queries';
import { ModelSchema } from '@buster/server-shared';
import type { GetDatasetSampleResponse } from '@buster/server-shared';
import { HTTPException } from 'hono/http-exception';

View File

@ -1,4 +1,4 @@
import type { User } from '@buster/database';
import type { User } from '@buster/database/queries';
import { Hono } from 'hono';
import { requireAuth } from '../../../../../middleware/auth';
import { GET } from './GET';

View File

@ -46,4 +46,4 @@
"devDependencies": {
"@trigger.dev/build": "4.0.4"
}
}
}

View File

@ -201,11 +201,12 @@ describe('Done Tool Streaming Tests', () => {
// No file response messages should be created for report-only case
const fileResponseCallWithFiles = (
queries.updateMessageEntries as unknown as { mock: { calls: [Record<string, any>][] } }
).mock.calls.find((c) =>
Array.isArray((c[0] as { responseMessages?: unknown[] }).responseMessages) &&
((c[0] as { responseMessages?: { type?: string }[] }).responseMessages || []).some(
(m) => m?.type === 'file'
)
).mock.calls.find(
(c) =>
Array.isArray((c[0] as { responseMessages?: unknown[] }).responseMessages) &&
((c[0] as { responseMessages?: { type?: string }[] }).responseMessages || []).some(
(m) => m?.type === 'file'
)
);
expect(fileResponseCallWithFiles).toBeUndefined();
});
@ -297,16 +298,18 @@ describe('Done Tool Streaming Tests', () => {
// Response messages should include the metric file
const fileResponseCall = (
queries.updateMessageEntries as unknown as { mock: { calls: [Record<string, any>][] } }
).mock.calls.find((c) =>
Array.isArray((c[0] as { responseMessages?: unknown[] }).responseMessages) &&
((c[0] as { responseMessages?: { type?: string }[] }).responseMessages || []).some(
(m) => m?.type === 'file'
)
).mock.calls.find(
(c) =>
Array.isArray((c[0] as { responseMessages?: unknown[] }).responseMessages) &&
((c[0] as { responseMessages?: { type?: string }[] }).responseMessages || []).some(
(m) => m?.type === 'file'
)
);
expect(fileResponseCall).toBeDefined();
const responseMessages = (fileResponseCall?.[0] as { responseMessages?: Record<string, any>[] })
?.responseMessages as Record<string, any>[];
const responseMessages = (
fileResponseCall?.[0] as { responseMessages?: Record<string, any>[] }
)?.responseMessages as Record<string, any>[];
const metricResponse = responseMessages?.find((m) => m.id === metricId);
expect(metricResponse).toBeDefined();
expect(metricResponse?.file_type).toBe('metric_file');