mirror of
https://github.com/fleetdm/fleet
synced 2026-05-17 05:58:40 +00:00
850e429b80
+ Use the same "Failed to generate human interpretation using generative AI." message if you send it something weird like "asdf" (so you don't get a vague 500 error). + Simplify the code a bit --------- Co-authored-by: Eric <[email protected]>
106 lines
3.1 KiB
JavaScript
Vendored
106 lines
3.1 KiB
JavaScript
Vendored
module.exports = {
|
|
|
|
|
|
friendlyName: 'Get human interpretation from osquery sql',
|
|
|
|
|
|
description: 'Infer policy information from osquery SQL.',
|
|
|
|
|
|
inputs: {
|
|
|
|
sql: {
|
|
type: 'string',
|
|
required: true
|
|
},
|
|
|
|
},
|
|
|
|
|
|
exits: {
|
|
|
|
success: {
|
|
outputFriendlyName: 'Humanesque interpretation',
|
|
outputDescription: 'If the call to the LLM fails, then a success response is sent with an explanation about the failure (e.g. "under heavy load", etc)',
|
|
outputExample: {
|
|
risks: 'Using an outdated macOS version risks exposure to security vulnerabilities and potential system instability.',
|
|
whatWillProbablyHappenDuringMaintenance: 'We will update your macOS to version 14.4.1 to enhance security and stability.'
|
|
}
|
|
},
|
|
|
|
},
|
|
|
|
|
|
fn: async function ({sql}) {
|
|
|
|
if (!sails.config.custom.openAiSecret) {
|
|
throw new Error('sails.config.custom.openAiSecret not set.');
|
|
}//•
|
|
|
|
// Build our prompt
|
|
let prompt = `Given this osquery policy: aka a query which either passes (≥1 row) or fails (0 rows) for a given laptop, what risks might we anticipate from that laptop having failed the policy?
|
|
|
|
Here is the query:
|
|
\`\`\`
|
|
${sql}
|
|
\`\`\`
|
|
|
|
Remember to minimize the number of words used!
|
|
|
|
Please give me all of the above in JSON, with this data shape:
|
|
|
|
{
|
|
risks: 'TODO',
|
|
whatWillProbablyHappenDuringMaintenance: 'TODO'
|
|
}
|
|
|
|
Please do not add any text outside of the JSON report or wrap it in a code fence.`;
|
|
// Fallback message in case LLM API request fails.
|
|
let failureMessage = 'Failed to generate human interpretation using generative AI.';
|
|
|
|
let BASE_MODEL = 'gpt-4';// The base model to use. https://platform.openai.com/docs/models/gpt-4
|
|
// (Max tokens for gpt-3.5 ≈≈ 4000) (Max tokens for gpt-4 ≈≈ 8000)
|
|
// [?] API: https://platform.openai.com/docs/api-reference/chat/create
|
|
let openAiResponse = await sails.helpers.http.post('https://api.openai.com/v1/chat/completions', {
|
|
model: BASE_MODEL,
|
|
messages: [// https://platform.openai.com/docs/guides/chat/introduction
|
|
{
|
|
role: 'user',
|
|
content: prompt
|
|
}
|
|
],
|
|
temperature: 0.7,
|
|
max_tokens: 256//eslint-disable-line camelcase
|
|
}, {
|
|
Authorization: `Bearer ${sails.config.custom.openAiSecret}`
|
|
})
|
|
.tolerate((err)=>{
|
|
sails.log.warn(failureMessage+' Error details from LLM: '+err.stack);
|
|
return {
|
|
choices: [
|
|
{
|
|
message: {
|
|
content: `{ "risks": "${failureMessage}", "whatWillProbablyHappenDuringMaintenance": "${failureMessage}" }`
|
|
}
|
|
}
|
|
]
|
|
};
|
|
});
|
|
|
|
let report;
|
|
try {
|
|
report = JSON.parse(openAiResponse.choices[0].message.content);
|
|
} catch (err) {
|
|
sails.log.warn('When trying to parse a JSON report returned from the Open AI API, an error occurred. Error details from JSON.parse: '+err.stack+'\n Report returned from Open AI:'+openAiResponse.choices[0].message.content);
|
|
report = {
|
|
risks: failureMessage,
|
|
whatWillProbablyHappenDuringMaintenance: failureMessage
|
|
};
|
|
}
|
|
|
|
return report;
|
|
|
|
}
|
|
|
|
|
|
};
|