ai-code-writer
Version:
An AI code writer application using OpenAI APIs for audio transcription and chat completion.
71 lines (70 loc) • 3.37 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());
});
};
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const ChatMessageEntity_1 = __importDefault(require("../../../ChatMessageEntity"));
class FileTask {
constructor(conversationLogger) {
this.conversationLogger = conversationLogger;
}
addFileListToInitialConversation(filesContents, conversationHistory) {
const messageItem = new ChatMessageEntity_1.default();
messageItem.role = 'system';
messageItem.content = 'Existing files in den Project:\n* ' +
filesContents.map(f => f.filePath).join('\n* ');
conversationHistory.push(messageItem);
}
addFileContentsToInitialConversation(filesContent, conversationHistory) {
return __awaiter(this, void 0, void 0, function* () {
for (const { filePath, content } of filesContent) {
const messageItem = new ChatMessageEntity_1.default();
messageItem.role = 'system';
messageItem.content = `File content of ${filePath}:
\`\`\`
${content}
\`\`\`
`;
messageItem.filePath = filePath;
console.log('File read:', filePath);
conversationHistory.push(messageItem);
yield this.conversationLogger.logConversation(messageItem);
}
});
}
removeOldFileFromHistory(conversationHistory, filePath) {
const existingMessageIndex = conversationHistory.findIndex(message => message.filePath === filePath);
if (existingMessageIndex == -1)
return;
const deletedMessage = conversationHistory.splice(existingMessageIndex, 1)[0];
this.removeCallFromResultMessages(deletedMessage, conversationHistory);
this.removeCallFromMessagesToolCalls(deletedMessage, conversationHistory);
}
removeCallFromResultMessages(deletedMessage, conversationHistory) {
for (const deletedCall of deletedMessage.toolCalls) {
const callId = deletedCall.id;
const index = conversationHistory.findIndex(c => c.callId == callId);
if (index == -1)
continue;
conversationHistory.splice(index, 1);
}
}
removeCallFromMessagesToolCalls(deletedMessage, conversationHistory) {
for (const message of conversationHistory) {
const index = message.toolCalls.findIndex(c => c.id == deletedMessage.callId);
if (index == -1)
continue;
message.toolCalls.splice(index, 1);
}
}
}
exports.default = FileTask;