Ling0925
/
buster
mirror of https://github.com/buster-so/buster.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/server/src/api/v2/datasets/[id]/index.ts
View File

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

2
apps/server/src/api/v2/datasets/[id]/sample/GET.ts
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 { GetDatasetSampleParamsSchema } from '@buster/server-shared';
import type { GetDatasetSampleResponse } from '@buster/server-shared'; import type { GetDatasetSampleResponse } from '@buster/server-shared';
import { zValidator } from '@hono/zod-validator'; import { zValidator } from '@hono/zod-validator';

4
apps/server/src/api/v2/datasets/[id]/sample/get-dataset-sample.ts
View File

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

2
apps/server/src/api/v2/datasets/[id]/sample/index.ts
View File

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

27
packages/ai/src/tools/communication-tools/done-tool/done-tool-streaming.test.ts
View File

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