Skip to content

Commit

Permalink
fix: Merge request config headers and params (#124)
Browse files Browse the repository at this point in the history
* fix error

* fix: Changes from lint

* modify an existing test

* fix: Changes from lint

* update test

* fix test

---------

Co-authored-by: cloud-sdk-js <[email protected]>
Co-authored-by: Shibesh Duwadi <[email protected]>
  • Loading branch information
3 people authored Sep 10, 2024
1 parent abd1423 commit 2e54be6
Show file tree
Hide file tree
Showing 3 changed files with 19 additions and 12 deletions.
11 changes: 8 additions & 3 deletions packages/ai-api/src/tests/deployment-api.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ describe('deployment', () => {
expect(result).toEqual(expectedResponse);
});

it('parses a successful response for post request', async () => {
it('parses a successful response for post request with required headers', async () => {
const expectedResponse: AiDeploymentCreationResponse = {
deploymentUrl: '',
id: '4e5f6g7h',
Expand All @@ -86,7 +86,10 @@ describe('deployment', () => {

nock(aiCoreDestination.url, {
reqheaders: {
'AI-Resource-Group': 'default'
'AI-Resource-Group': 'default',
'some-test-header': 'test-header-value',
'content-type': 'application/json',
'ai-client-type': 'AI SDK JavaScript'
}
})
.post('/v2/lm/deployments')
Expand All @@ -101,7 +104,9 @@ describe('deployment', () => {
const result: AiDeploymentCreationResponse =
await DeploymentApi.deploymentCreate(deploymentPostData, {
'AI-Resource-Group': 'default'
}).execute();
})
.addCustomHeaders({ 'some-test-header': 'test-header-value' })
.execute();

expect(result).toEqual(expectedResponse);
});
Expand Down
10 changes: 2 additions & 8 deletions packages/core/src/http-client.ts
Original file line number Diff line number Diff line change
Expand Up @@ -102,13 +102,7 @@ function mergeWithDefaultRequestConfig(
return {
...defaultConfig,
...requestConfig,
headers: mergeIgnoreCase({
...defaultConfig.headers,
...requestConfig?.headers
}),
params: mergeIgnoreCase({
...defaultConfig.params,
...requestConfig?.params
})
headers: mergeIgnoreCase(defaultConfig.headers, requestConfig?.headers),
params: mergeIgnoreCase(defaultConfig.params, requestConfig?.params)
};
}
10 changes: 9 additions & 1 deletion packages/core/src/openapi-request-builder.ts
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,15 @@ export class OpenApiRequestBuilder<
const { url, data, ...rest } = await this.requestConfig();
// TODO: Remove explicit url! once we updated the type in the Cloud SDK, since url is always defined.
return executeRequest({ url: url! }, data, {
...rest
...rest,
headers: {
...rest.headers?.requestConfig,
...rest.headers?.custom
},
params: {
...rest.params?.requestConfig,
...rest.params?.custom
}
});
}

Expand Down

0 comments on commit 2e54be6

Please sign in to comment.