Skip to content

chore(atlas-service): add logging for query and aggregation generation request and response COMPASS-7228 #4871

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Sep 19, 2023
Merged

chore(atlas-service): add logging for query and aggregation generation request and response COMPASS-7228 #4871

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
ad12f2e
add logging for atlas-service ai request and response
Anemy Sep 19, 2023
7945baf
update log ids
Anemy Sep 19, 2023
18cfc14
merge main
Anemy Sep 19, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
87 changes: 56 additions & 31 deletions packages/atlas-service/src/main.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -337,12 +337,12 @@ export class AtlasService {
log.info(mongoLogId(1_001_000_218), 'AtlasService', 'Starting sign in');

try {
const userInfo = await this.getUserInfo({ signal });
log.info(
mongoLogId(1_001_000_219),
'AtlasService',
'Signed in successfully'
);
const userInfo = await this.getUserInfo({ signal });
track('Atlas Sign In Success', getTrackingUserInfo(userInfo));
return userInfo;
} catch (err) {
Expand Down Expand Up @@ -544,11 +544,17 @@ export class AtlasService {
throwIfNetworkTrafficDisabled();

const userId = (await this.getActiveCompassUser()).id;
const url = `${this.config.atlasApiUnauthBaseUrl}/ai/api/v1/hello/${userId}`;

const res = await this.fetch(
`${this.config.atlasApiUnauthBaseUrl}/ai/api/v1/hello/${userId}`
log.info(
mongoLogId(1_001_000_227),
'AtlasService',
'Fetching if the AI feature is enabled via hello endpoint',
{ userId, url }
);

const res = await this.fetch(url);

await throwIfNotOk(res);

const body = await res.json();
Expand All @@ -559,12 +565,6 @@ export class AtlasService {
}

static async setupAIAccess(): Promise<void> {
log.info(
mongoLogId(1_001_000_227),
'AtlasService',
'Fetching if the AI feature is enabled'
);

try {
throwIfNetworkTrafficDisabled();

Expand All @@ -579,6 +579,7 @@ export class AtlasService {
'Fetched if the AI feature is enabled',
{
enabled: isAIFeatureEnabled,
featureResponse,
}
);

Expand Down Expand Up @@ -644,18 +645,30 @@ export class AtlasService {
}

const token = await this.maybeGetToken({ signal });
const url = `${this.config.atlasApiBaseUrl}/ai/api/v1/mql-aggregation`;

const res = await this.fetch(
`${this.config.atlasApiBaseUrl}/ai/api/v1/mql-aggregation`,
{
signal: signal as NodeFetchAbortSignal | undefined,
method: 'POST',
headers: {
Authorization: `Bearer ${token ?? ''}`,
'Content-Type': 'application/json',
},
body: msgBody,
}
log.info(
mongoLogId(1_001_000_247),
'AtlasService',
'Running aggregation generation request',
{ url, body: msgBody }
);

const res = await this.fetch(url, {
signal: signal as NodeFetchAbortSignal | undefined,
method: 'POST',
headers: {
Authorization: `Bearer ${token ?? ''}`,
'Content-Type': 'application/json',
},
body: msgBody,
});

log.info(
mongoLogId(1_001_000_248),
'AtlasService',
'Received aggregation generation response',
{ status: res.status, statusText: res.statusText }
);

await throwIfNotOk(res);
Expand Down Expand Up @@ -713,18 +726,30 @@ export class AtlasService {
}

const token = await this.maybeGetToken({ signal });
const url = `${this.config.atlasApiBaseUrl}/ai/api/v1/mql-query`;

const res = await this.fetch(
`${this.config.atlasApiBaseUrl}/ai/api/v1/mql-query`,
{
signal: signal as NodeFetchAbortSignal | undefined,
method: 'POST',
headers: {
Authorization: `Bearer ${token ?? ''}`,
'Content-Type': 'application/json',
},
body: msgBody,
}
log.info(
mongoLogId(1_001_000_249),
'AtlasService',
'Running query generation request',
{ url, body: msgBody }
);

const res = await this.fetch(url, {
signal: signal as NodeFetchAbortSignal | undefined,
method: 'POST',
headers: {
Authorization: `Bearer ${token ?? ''}`,
'Content-Type': 'application/json',
},
body: msgBody,
});

log.info(
mongoLogId(1_001_000_250),
'AtlasService',
'Received query generation response',
{ status: res.status, statusText: res.statusText }
);

await throwIfNotOk(res);
Expand Down