Skip to content

Commit

Permalink
Adjust tests
Browse files Browse the repository at this point in the history
  • Loading branch information
MatKuhr committed Feb 10, 2025
1 parent ae338e7 commit d6e0c81
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 15 deletions.
2 changes: 0 additions & 2 deletions packages/orchestration/src/orchestration-client.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -341,9 +341,7 @@ describe('orchestration service client', () => {
config: {
filters: [
{
id: 'filter1',
data_repositories: ['*'],
search_config: {},
data_repository_type: 'vector'
}
],
Expand Down
13 changes: 3 additions & 10 deletions packages/orchestration/src/util/grounding.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4,11 +4,7 @@ import type { DocumentGroundingServiceConfig } from '../orchestration-types.js';
describe('document grounding util', () => {
it('builds grounding configuration with minimal required properties', () => {
const groundingConfig: DocumentGroundingServiceConfig = {
filters: [
{
id: 'filter-id'
}
],
filters: [{}],
input_params: ['input'],
output_param: 'output'
};
Expand All @@ -17,7 +13,6 @@ describe('document grounding util', () => {
config: {
filters: [
{
id: 'filter-id',
data_repository_type: 'vector'
}
],
Expand All @@ -31,9 +26,8 @@ describe('document grounding util', () => {
const groundingConfig: DocumentGroundingServiceConfig = {
filters: [
{
id: 'filter-id',
data_repositories: ['repo1', 'repo2'],
data_repository_type: 'custom-type'
data_repository_type: 'help.sap.com'
}
],
input_params: ['input'],
Expand All @@ -44,9 +38,8 @@ describe('document grounding util', () => {
config: {
filters: [
{
id: 'filter-id',
data_repositories: ['repo1', 'repo2'],
data_repository_type: 'custom-type'
data_repository_type: 'help.sap.com'
}
],
input_params: ['input'],
Expand Down
4 changes: 1 addition & 3 deletions sample-code/src/orchestration.ts
Original file line number Diff line number Diff line change
Expand Up @@ -313,7 +313,6 @@ export async function orchestrationMaskGroundingInput(): Promise<OrchestrationRe
output_param: 'groundingOutput',
filters: [
{
id: 'filter1',
data_repository_type: 'help.sap.com'
}
]
Expand Down Expand Up @@ -396,7 +395,7 @@ export async function orchestrationGroundingVector(): Promise<OrchestrationRespo
grounding: buildDocumentGroundingConfig({
input_params: ['groundingRequest'],
output_param: 'groundingOutput',
filters: [{ id: 'filter1' }]
filters: [{ data_repository_type: 'vector' }]
})
},
{ resourceGroup: 'ai-sdk-js-e2e' }
Expand Down Expand Up @@ -431,7 +430,6 @@ export async function orchestrationGroundingHelpSapCom(): Promise<OrchestrationR
output_param: 'groundingOutput',
filters: [
{
id: 'filter1',
data_repository_type: 'help.sap.com'
}
]
Expand Down

0 comments on commit d6e0c81

Please sign in to comment.