chore: import upstream snapshot with attribution
CI / Migrate Dev DB (push) Has been skipped
CI / Detect Version (push) Has been cancelled
CI / Migrate DB (push) Has been cancelled
CI / Build Dev ECR (./docker/app.Dockerfile, ECR_APP) (push) Has been cancelled
CI / Build Dev ECR (./docker/db.Dockerfile, ECR_MIGRATIONS) (push) Has been cancelled
CI / Build Dev ECR (./docker/pii.Dockerfile, ECR_PII) (push) Has been cancelled
CI / Build Dev ECR (./docker/realtime.Dockerfile, ECR_REALTIME) (push) Has been cancelled
CI / Deploy Trigger.dev (Dev) (push) Has been cancelled
CI / Build AMD64 (./docker/app.Dockerfile, ECR_APP, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build AMD64 (./docker/db.Dockerfile, ECR_MIGRATIONS, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build AMD64 (./docker/pii.Dockerfile, ECR_PII, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build AMD64 (./docker/realtime.Dockerfile, ECR_REALTIME, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/app.Dockerfile, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/db.Dockerfile, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/pii.Dockerfile, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/realtime.Dockerfile, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Check Docs Changes (push) Has been cancelled
CI / Process Docs (push) Has been cancelled
CI / Create GitHub Release (push) Has been cancelled
CI / Test and Build (push) Has been cancelled
Publish CLI Package / publish-npm (push) Has been cancelled
Publish Python SDK / publish-pypi (push) Has been cancelled
Publish TypeScript SDK / publish-npm (push) Has been cancelled
CI / Migrate Dev DB (push) Has been skipped
CI / Detect Version (push) Has been cancelled
CI / Migrate DB (push) Has been cancelled
CI / Build Dev ECR (./docker/app.Dockerfile, ECR_APP) (push) Has been cancelled
CI / Build Dev ECR (./docker/db.Dockerfile, ECR_MIGRATIONS) (push) Has been cancelled
CI / Build Dev ECR (./docker/pii.Dockerfile, ECR_PII) (push) Has been cancelled
CI / Build Dev ECR (./docker/realtime.Dockerfile, ECR_REALTIME) (push) Has been cancelled
CI / Deploy Trigger.dev (Dev) (push) Has been cancelled
CI / Build AMD64 (./docker/app.Dockerfile, ECR_APP, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build AMD64 (./docker/db.Dockerfile, ECR_MIGRATIONS, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build AMD64 (./docker/pii.Dockerfile, ECR_PII, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build AMD64 (./docker/realtime.Dockerfile, ECR_REALTIME, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/app.Dockerfile, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/db.Dockerfile, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/pii.Dockerfile, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/realtime.Dockerfile, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Check Docs Changes (push) Has been cancelled
CI / Process Docs (push) Has been cancelled
CI / Create GitHub Release (push) Has been cancelled
CI / Test and Build (push) Has been cancelled
Publish CLI Package / publish-npm (push) Has been cancelled
Publish Python SDK / publish-pypi (push) Has been cancelled
Publish TypeScript SDK / publish-npm (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,206 @@
|
||||
/**
|
||||
* @vitest-environment node
|
||||
*/
|
||||
import { describe, expect, it } from 'vitest'
|
||||
import { extractDeletedResourcesFromToolResult, extractResourcesFromToolResult } from './extraction'
|
||||
|
||||
describe('extractResourcesFromToolResult', () => {
|
||||
it('extracts file resources from create_file results', () => {
|
||||
const resources = extractResourcesFromToolResult(
|
||||
'create_file',
|
||||
{
|
||||
fileName: 'notes.md',
|
||||
},
|
||||
{
|
||||
success: true,
|
||||
message: 'File "notes.md" created successfully',
|
||||
data: {
|
||||
id: 'file_123',
|
||||
name: 'notes.md',
|
||||
contentType: 'text/markdown',
|
||||
},
|
||||
}
|
||||
)
|
||||
|
||||
expect(resources).toEqual([
|
||||
{
|
||||
type: 'file',
|
||||
id: 'file_123',
|
||||
title: 'notes.md',
|
||||
},
|
||||
])
|
||||
})
|
||||
|
||||
it('uses the knowledge base id for knowledge_base tag mutations', () => {
|
||||
const resources = extractResourcesFromToolResult(
|
||||
'knowledge_base',
|
||||
{
|
||||
operation: 'update_tag',
|
||||
args: {
|
||||
knowledgeBaseId: 'kb_123',
|
||||
tagDefinitionId: 'tag_456',
|
||||
},
|
||||
},
|
||||
{
|
||||
success: true,
|
||||
message: 'Tag updated successfully',
|
||||
data: {
|
||||
id: 'tag_456',
|
||||
displayName: 'Priority',
|
||||
fieldType: 'text',
|
||||
},
|
||||
}
|
||||
)
|
||||
|
||||
expect(resources).toEqual([
|
||||
{
|
||||
type: 'knowledgebase',
|
||||
id: 'kb_123',
|
||||
title: 'Knowledge Base',
|
||||
},
|
||||
])
|
||||
})
|
||||
|
||||
it('uses knowledgeBaseId from the tool result when update_tag args omit it', () => {
|
||||
const resources = extractResourcesFromToolResult(
|
||||
'knowledge_base',
|
||||
{
|
||||
operation: 'update_tag',
|
||||
args: {
|
||||
tagDefinitionId: 'tag_456',
|
||||
},
|
||||
},
|
||||
{
|
||||
success: true,
|
||||
message: 'Tag updated successfully',
|
||||
data: {
|
||||
id: 'tag_456',
|
||||
knowledgeBaseId: 'kb_123',
|
||||
displayName: 'Priority',
|
||||
fieldType: 'text',
|
||||
},
|
||||
}
|
||||
)
|
||||
|
||||
expect(resources).toEqual([
|
||||
{
|
||||
type: 'knowledgebase',
|
||||
id: 'kb_123',
|
||||
title: 'Knowledge Base',
|
||||
},
|
||||
])
|
||||
})
|
||||
|
||||
it('does not create resources for read-only knowledge base tag operations', () => {
|
||||
const resources = extractResourcesFromToolResult(
|
||||
'knowledge_base',
|
||||
{
|
||||
operation: 'list_tags',
|
||||
args: {
|
||||
knowledgeBaseId: 'kb_123',
|
||||
},
|
||||
},
|
||||
{
|
||||
success: true,
|
||||
data: [],
|
||||
}
|
||||
)
|
||||
|
||||
expect(resources).toEqual([])
|
||||
})
|
||||
|
||||
it.each([
|
||||
['generate_video', 'ad-clip.mp4'],
|
||||
['generate_audio', 'voiceover.mp3'],
|
||||
['ffmpeg', 'final-ad.mp4'],
|
||||
])('auto-opens the generated file from %s results', (toolName, fileName) => {
|
||||
const resources = extractResourcesFromToolResult(
|
||||
toolName,
|
||||
{},
|
||||
{
|
||||
success: true,
|
||||
message: `Saved at "files/${fileName}"`,
|
||||
fileId: 'file_media_123',
|
||||
fileName,
|
||||
}
|
||||
)
|
||||
|
||||
expect(resources).toEqual([{ type: 'file', id: 'file_media_123', title: fileName }])
|
||||
})
|
||||
|
||||
it('does not create a resource for ffmpeg probe (no file written)', () => {
|
||||
const resources = extractResourcesFromToolResult(
|
||||
'ffmpeg',
|
||||
{ operation: 'probe' },
|
||||
{
|
||||
success: true,
|
||||
message: 'Probed media',
|
||||
probe: { durationSeconds: 12.5, width: 1080, height: 1920 },
|
||||
}
|
||||
)
|
||||
|
||||
expect(resources).toEqual([])
|
||||
})
|
||||
|
||||
it('auto-opens a scheduledtask resource from manage_scheduled_task create results', () => {
|
||||
const resources = extractResourcesFromToolResult(
|
||||
'manage_scheduled_task',
|
||||
{ operation: 'create', args: { title: 'Daily Report' } },
|
||||
{ jobId: 'sched_123', title: 'Daily Report', message: 'Job created successfully.' }
|
||||
)
|
||||
|
||||
expect(resources).toEqual([{ type: 'scheduledtask', id: 'sched_123', title: 'Daily Report' }])
|
||||
})
|
||||
|
||||
it('auto-opens a scheduledtask resource on update, falling back to the args title', () => {
|
||||
const resources = extractResourcesFromToolResult(
|
||||
'manage_scheduled_task',
|
||||
{ operation: 'update', args: { jobId: 'sched_123', title: 'Renamed Task' } },
|
||||
{ jobId: 'sched_123', updated: ['title'], message: 'Job updated successfully' }
|
||||
)
|
||||
|
||||
expect(resources).toEqual([{ type: 'scheduledtask', id: 'sched_123', title: 'Renamed Task' }])
|
||||
})
|
||||
|
||||
it('does not auto-open for read-only manage_scheduled_task operations', () => {
|
||||
expect(
|
||||
extractResourcesFromToolResult(
|
||||
'manage_scheduled_task',
|
||||
{ operation: 'list' },
|
||||
{ jobs: [], count: 0 }
|
||||
)
|
||||
).toEqual([])
|
||||
expect(
|
||||
extractResourcesFromToolResult(
|
||||
'manage_scheduled_task',
|
||||
{ operation: 'get', args: { jobId: 'sched_123' } },
|
||||
{ id: 'sched_123', title: 'Daily Report' }
|
||||
)
|
||||
).toEqual([])
|
||||
})
|
||||
})
|
||||
|
||||
describe('extractDeletedResourcesFromToolResult', () => {
|
||||
it('removes scheduledtask resources on manage_scheduled_task delete', () => {
|
||||
const resources = extractDeletedResourcesFromToolResult(
|
||||
'manage_scheduled_task',
|
||||
{ operation: 'delete', args: { jobIds: ['sched_1', 'sched_2'] } },
|
||||
{ deleted: ['sched_1', 'sched_2'], notFound: [] }
|
||||
)
|
||||
|
||||
expect(resources).toEqual([
|
||||
{ type: 'scheduledtask', id: 'sched_1', title: 'Scheduled Task' },
|
||||
{ type: 'scheduledtask', id: 'sched_2', title: 'Scheduled Task' },
|
||||
])
|
||||
})
|
||||
|
||||
it('does not remove anything for non-delete manage_scheduled_task ops', () => {
|
||||
expect(
|
||||
extractDeletedResourcesFromToolResult(
|
||||
'manage_scheduled_task',
|
||||
{ operation: 'update', args: { jobId: 'sched_1' } },
|
||||
{ jobId: 'sched_1', updated: ['title'] }
|
||||
)
|
||||
).toEqual([])
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user