@autobe/agent
Version:
AI backend server code generator
79 lines • 3.86 kB
JavaScript
;
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.writeDocumentUntilReviewPassed = writeDocumentUntilReviewPassed;
const orchestrateAnalyzeReviewer_1 = require("./orchestrateAnalyzeReviewer");
const orchestrateAnalyzeWrite_1 = require("./orchestrateAnalyzeWrite");
function writeDocumentUntilReviewPassed(ctx, props) {
return __awaiter(this, void 0, void 0, function* () {
var _a, _b, _c, _d, _e, _f, _g;
const retry = (_a = props.retry) !== null && _a !== void 0 ? _a : 3;
const pointer = {
value: { files: {} },
};
/**
* `retry` means the number of times to retry the review. so if `retry` is -1,
* it means not execute this logic.
*/
if (retry === -1) {
return pointer;
}
let isToolCalled = false;
const writer = (0, orchestrateAnalyzeWrite_1.orchestrateAnalyzeWrite)(ctx, {
totalFiles: props.totalFiles,
roles: props.roles,
targetFile: props.filename,
review: (_b = props.prevReview) !== null && _b !== void 0 ? _b : "",
setDocument: (v) => {
var _a;
isToolCalled = true;
pointer.value = { files: Object.assign(Object.assign({}, (_a = pointer.value) === null || _a === void 0 ? void 0 : _a.files), v) };
},
});
yield writer.conversate("Write Document.").finally(() => {
const tokenUsage = writer.getTokenUsage();
ctx.usage().record(tokenUsage, ["analyze"]);
});
if (isToolCalled === false) {
throw new Error("Failed to write document by unknown reason.");
}
ctx.dispatch({
type: "analyzeWrite",
files: Object.assign({}, (_c = pointer.value) === null || _c === void 0 ? void 0 : _c.files),
total: props.progress.total,
completed: ++props.progress.completed,
step: (_e = (_d = ctx.state().analyze) === null || _d === void 0 ? void 0 : _d.step) !== null && _e !== void 0 ? _e : 0,
created_at: new Date().toISOString(),
});
const reviewResult = yield (0, orchestrateAnalyzeReviewer_1.orchestrateAnalyzeReviewer)(ctx, pointer.value);
if (reviewResult.type === "accept") {
return pointer;
}
ctx.dispatch({
type: "analyzeReview",
files: Object.assign({}, pointer.value.files),
review: reviewResult.value,
total: props.progress.total,
completed: props.progress.completed,
step: (_g = (_f = ctx.state().analyze) === null || _f === void 0 ? void 0 : _f.step) !== null && _g !== void 0 ? _g : 0,
created_at: new Date().toISOString(),
});
return yield writeDocumentUntilReviewPassed(ctx, {
totalFiles: props.totalFiles,
filename: props.filename,
roles: props.roles,
progress: props.progress,
retry: retry - 1,
prevReview: reviewResult.value,
});
});
}
//# sourceMappingURL=writeDocumentUntilReviewPassed.js.map