@proteinjs/conversation 1.0.7 → 1.0.9

package/dist/src/Conversation.js

@@ -1,499 +1,309 @@
-'use strict';
-var __awaiter =
-  (this && this.__awaiter) ||
-  function (thisArg, _arguments, P, generator) {
-    function adopt(value) {
-      return value instanceof P
-        ? value
-        : new P(function (resolve) {
-            resolve(value);
-          });
-    }
+"use strict";
+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());
+        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 __generator =
-  (this && this.__generator) ||
-  function (thisArg, body) {
-    var _ = {
-        label: 0,
-        sent: function () {
-          if (t[0] & 1) throw t[1];
-          return t[1];
-        },
-        trys: [],
-        ops: [],
-      },
-      f,
-      y,
-      t,
-      g;
-    return (
-      (g = { next: verb(0), throw: verb(1), return: verb(2) }),
-      typeof Symbol === 'function' &&
-        (g[Symbol.iterator] = function () {
-          return this;
-        }),
-      g
-    );
-    function verb(n) {
-      return function (v) {
-        return step([n, v]);
-      };
-    }
+};
+var __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
+    return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
     function step(op) {
-      if (f) throw new TypeError('Generator is already executing.');
-      while ((g && ((g = 0), op[0] && (_ = 0)), _))
-        try {
-          if (
-            ((f = 1),
-            y &&
-              (t = op[0] & 2 ? y['return'] : op[0] ? y['throw'] || ((t = y['return']) && t.call(y), 0) : y.next) &&
-              !(t = t.call(y, op[1])).done)
-          )
-            return t;
-          if (((y = 0), t)) op = [op[0] & 2, t.value];
-          switch (op[0]) {
-            case 0:
-            case 1:
-              t = op;
-              break;
-            case 4:
-              _.label++;
-              return { value: op[1], done: false };
-            case 5:
-              _.label++;
-              y = op[1];
-              op = [0];
-              continue;
-            case 7:
-              op = _.ops.pop();
-              _.trys.pop();
-              continue;
-            default:
-              if (!((t = _.trys), (t = t.length > 0 && t[t.length - 1])) && (op[0] === 6 || op[0] === 2)) {
-                _ = 0;
-                continue;
-              }
-              if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) {
-                _.label = op[1];
-                break;
-              }
-              if (op[0] === 6 && _.label < t[1]) {
-                _.label = t[1];
-                t = op;
-                break;
-              }
-              if (t && _.label < t[2]) {
-                _.label = t[2];
-                _.ops.push(op);
-                break;
-              }
-              if (t[2]) _.ops.pop();
-              _.trys.pop();
-              continue;
-          }
-          op = body.call(thisArg, _);
-        } catch (e) {
-          op = [6, e];
-          y = 0;
-        } finally {
-          f = t = 0;
-        }
-      if (op[0] & 5) throw op[1];
-      return { value: op[0] ? op[1] : void 0, done: true };
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
     }
-  };
-Object.defineProperty(exports, '__esModule', { value: true });
-exports.summarizeConversationHistoryFunction =
-  exports.summarizeConversationHistoryFunctionName =
-  exports.Conversation =
-    void 0;
-var OpenAi_1 = require('./OpenAi');
-var MessageHistory_1 = require('./history/MessageHistory');
-var util_1 = require('@proteinjs/util');
-var util_node_1 = require('@proteinjs/util-node');
-var tiktoken_1 = require('tiktoken');
-var PackageFunctions_1 = require('./fs/package/PackageFunctions');
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.summarizeConversationHistoryFunction = exports.summarizeConversationHistoryFunctionName = exports.Conversation = void 0;
+var OpenAi_1 = require("./OpenAi");
+var MessageHistory_1 = require("./history/MessageHistory");
+var util_1 = require("@proteinjs/util");
+var util_node_1 = require("@proteinjs/util-node");
+var tiktoken_1 = require("tiktoken");
+var PackageFunctions_1 = require("./fs/package/PackageFunctions");
 var Conversation = /** @class */ (function () {
-  function Conversation(params) {
-    var _a, _b, _c, _d;
-    this.tokenLimit = 3000;
-    this.systemMessages = [];
-    this.functions = [];
-    this.messageModerators = [];
-    this.generatedCode = false;
-    this.generatedList = false;
-    this.params = params;
-    this.history = new MessageHistory_1.MessageHistory({
-      maxMessages: (_a = params.limits) === null || _a === void 0 ? void 0 : _a.maxMessagesInHistory,
-      enforceMessageLimit: (_b = params.limits) === null || _b === void 0 ? void 0 : _b.enforceLimits,
-    });
-    this.logger = new util_1.Logger(params.name, params.logLevel);
-    if (params.modules) {
-      this.addModules(params.modules);
-    }
-    if (
-      typeof ((_c = params.limits) === null || _c === void 0 ? void 0 : _c.enforceLimits) === 'undefined' ||
-      params.limits.enforceLimits
-    ) {
-      this.addFunctions('Conversation', [(0, exports.summarizeConversationHistoryFunction)(this)]);
-    }
-    if ((_d = params.limits) === null || _d === void 0 ? void 0 : _d.tokenLimit) {
-      this.tokenLimit = params.limits.tokenLimit;
-    }
-  }
-  Conversation.prototype.addModules = function (modules) {
-    for (var _i = 0, modules_1 = modules; _i < modules_1.length; _i++) {
-      var module_1 = modules_1[_i];
-      if (module_1.getSystemMessages().length < 1) {
-        continue;
-      }
-      this.addSystemMessagesToHistory([
-        'The following are instructions from the '
-          .concat(module_1.getName(), ' module: ')
-          .concat(module_1.getSystemMessages().join('. ')),
-      ]);
-      this.addFunctions(module_1.getName(), module_1.getFunctions());
-      this.addMessageModerators(module_1.getMessageModerators());
-    }
-  };
-  Conversation.prototype.addFunctions = function (moduleName, functions) {
-    var _a;
-    (_a = this.functions).push.apply(_a, functions);
-    var functionInstructions = 'The following are instructions from functions in the '.concat(moduleName, ' module:');
-    var functionInstructionsAdded = false;
-    for (var _i = 0, functions_1 = functions; _i < functions_1.length; _i++) {
-      var f = functions_1[_i];
-      if (f.instructions) {
-        if (!f.instructions || f.instructions.length < 1) {
-          continue;
+    function Conversation(params) {
+        var _a, _b, _c, _d;
+        this.tokenLimit = 3000;
+        this.systemMessages = [];
+        this.functions = [];
+        this.messageModerators = [];
+        this.generatedCode = false;
+        this.generatedList = false;
+        this.params = params;
+        this.history = new MessageHistory_1.MessageHistory({
+            maxMessages: (_a = params.limits) === null || _a === void 0 ? void 0 : _a.maxMessagesInHistory,
+            enforceMessageLimit: (_b = params.limits) === null || _b === void 0 ? void 0 : _b.enforceLimits,
+        });
+        this.logger = new util_1.Logger(params.name, params.logLevel);
+        if (params.modules) {
+            this.addModules(params.modules);
+        }
+        if (typeof ((_c = params.limits) === null || _c === void 0 ? void 0 : _c.enforceLimits) === 'undefined' || params.limits.enforceLimits) {
+            this.addFunctions('Conversation', [(0, exports.summarizeConversationHistoryFunction)(this)]);
+        }
+        if ((_d = params.limits) === null || _d === void 0 ? void 0 : _d.tokenLimit) {
+            this.tokenLimit = params.limits.tokenLimit;
         }
-        functionInstructionsAdded = true;
-        var instructionsParagraph = f.instructions.join('. ');
-        functionInstructions += ' '.concat(f.definition.name, ': ').concat(instructionsParagraph, '.');
-      }
-    }
-    if (!functionInstructionsAdded) {
-      return;
     }
-    this.addSystemMessagesToHistory([functionInstructions]);
-  };
-  Conversation.prototype.addMessageModerators = function (messageModerators) {
-    var _a;
-    (_a = this.messageModerators).push.apply(_a, messageModerators);
-  };
-  Conversation.prototype.enforceTokenLimit = function (messages, model) {
-    var _a;
-    return __awaiter(this, void 0, void 0, function () {
-      var resolvedModel, encoder, conversation, encoded, summarizeConversationRequest, referenceSummaryRequest;
-      return __generator(this, function (_b) {
-        switch (_b.label) {
-          case 0:
-            if (((_a = this.params.limits) === null || _a === void 0 ? void 0 : _a.enforceLimits) === false) {
-              return [2 /*return*/];
+    Conversation.prototype.addModules = function (modules) {
+        for (var _i = 0, modules_1 = modules; _i < modules_1.length; _i++) {
+            var module_1 = modules_1[_i];
+            if (module_1.getSystemMessages().length < 1) {
+                continue;
             }
-            resolvedModel = model ? model : OpenAi_1.DEFAULT_MODEL;
-            encoder = (0, tiktoken_1.encoding_for_model)(resolvedModel);
-            conversation = this.history.toString() + messages.join('. ');
-            encoded = encoder.encode(conversation);
-            console.log('current tokens: '.concat(encoded.length));
-            if (encoded.length < this.tokenLimit) {
-              return [2 /*return*/];
+            this.addSystemMessagesToHistory([
+                "The following are instructions from the ".concat(module_1.getName(), " module: ").concat(module_1.getSystemMessages().join('. ')),
+            ]);
+            this.addFunctions(module_1.getName(), module_1.getFunctions());
+            this.addMessageModerators(module_1.getMessageModerators());
+        }
+    };
+    Conversation.prototype.addFunctions = function (moduleName, functions) {
+        var _a;
+        (_a = this.functions).push.apply(_a, functions);
+        var functionInstructions = "The following are instructions from functions in the ".concat(moduleName, " module:");
+        var functionInstructionsAdded = false;
+        for (var _i = 0, functions_1 = functions; _i < functions_1.length; _i++) {
+            var f = functions_1[_i];
+            if (f.instructions) {
+                if (!f.instructions || f.instructions.length < 1) {
+                    continue;
+                }
+                functionInstructionsAdded = true;
+                var instructionsParagraph = f.instructions.join('. ');
+                functionInstructions += " ".concat(f.definition.name, ": ").concat(instructionsParagraph, ".");
             }
-            summarizeConversationRequest = 'First, call the '.concat(
-              exports.summarizeConversationHistoryFunctionName,
-              ' function'
-            );
-            return [
-              4 /*yield*/,
-              OpenAi_1.OpenAi.generateResponse(
-                [summarizeConversationRequest],
-                model,
-                this.history,
-                this.functions,
-                this.messageModerators,
-                this.params.logLevel
-              ),
-            ];
-          case 1:
-            _b.sent();
-            referenceSummaryRequest =
-              "If there's a file mentioned in the conversation summary, find and read the file to better respond to my next request. If that doesn't find anything, call the ".concat(
-                PackageFunctions_1.searchLibrariesFunctionName,
-                ' function on other keywords in the conversation summary to find a file to read'
-              );
-            return [
-              4 /*yield*/,
-              OpenAi_1.OpenAi.generateResponse(
-                [referenceSummaryRequest],
-                model,
-                this.history,
-                this.functions,
-                this.messageModerators,
-                this.params.logLevel
-              ),
-            ];
-          case 2:
-            _b.sent();
-            return [2 /*return*/];
         }
-      });
-    });
-  };
-  Conversation.prototype.summarizeConversationHistory = function (summary) {
-    this.clearHistory();
-    this.history.push([{ role: 'assistant', content: 'Previous conversation summary: '.concat(summary) }]);
-  };
-  Conversation.prototype.clearHistory = function () {
-    this.history = new MessageHistory_1.MessageHistory();
-    this.history.push(this.systemMessages);
-  };
-  Conversation.prototype.addSystemMessagesToHistory = function (messages, unshift) {
-    var _a, _b, _c;
-    if (unshift === void 0) {
-      unshift = false;
-    }
-    var chatCompletions = messages.map(function (message) {
-      return { role: 'system', content: message };
-    });
-    if (unshift) {
-      (_a = this.history.getMessages()).unshift.apply(_a, chatCompletions);
-      this.history.prune();
-      (_b = this.systemMessages).unshift.apply(_b, chatCompletions);
-    } else {
-      this.history.push(chatCompletions);
-      (_c = this.systemMessages).push.apply(_c, chatCompletions);
-    }
-  };
-  Conversation.prototype.addAssistantMessagesToHistory = function (messages, unshift) {
-    var _a;
-    if (unshift === void 0) {
-      unshift = false;
-    }
-    var chatCompletions = messages.map(function (message) {
-      return { role: 'assistant', content: message };
-    });
-    if (unshift) {
-      (_a = this.history.getMessages()).unshift.apply(_a, chatCompletions);
-      this.history.prune();
-    } else {
-      this.history.push(chatCompletions);
-    }
-  };
-  Conversation.prototype.addUserMessagesToHistory = function (messages, unshift) {
-    var _a;
-    if (unshift === void 0) {
-      unshift = false;
-    }
-    var chatCompletions = messages.map(function (message) {
-      return { role: 'user', content: message };
-    });
-    if (unshift) {
-      (_a = this.history.getMessages()).unshift.apply(_a, chatCompletions);
-      this.history.prune();
-    } else {
-      this.history.push(chatCompletions);
-    }
-  };
-  Conversation.prototype.generateResponse = function (messages, model) {
-    return __awaiter(this, void 0, void 0, function () {
-      return __generator(this, function (_a) {
-        switch (_a.label) {
-          case 0:
-            return [4 /*yield*/, this.enforceTokenLimit(messages, model)];
-          case 1:
-            _a.sent();
-            return [
-              4 /*yield*/,
-              OpenAi_1.OpenAi.generateResponse(
-                messages,
-                model,
-                this.history,
-                this.functions,
-                this.messageModerators,
-                this.params.logLevel
-              ),
-            ];
-          case 2:
-            return [2 /*return*/, _a.sent()];
+        if (!functionInstructionsAdded) {
+            return;
         }
-      });
-    });
-  };
-  Conversation.prototype.generateCode = function (description, model) {
-    return __awaiter(this, void 0, void 0, function () {
-      var code;
-      return __generator(this, function (_a) {
-        switch (_a.label) {
-          case 0:
-            this.logger.info('Generating code for description:\n'.concat(description.join('\n')));
-            return [
-              4 /*yield*/,
-              OpenAi_1.OpenAi.generateCode(
-                description,
-                model,
-                this.history,
-                this.functions,
-                this.messageModerators,
-                !this.generatedCode,
-                this.params.logLevel
-              ),
-            ];
-          case 1:
-            code = _a.sent();
-            this.logger.info('Generated code:\n'.concat(code.slice(0, 150)).concat(code.length > 150 ? '...' : ''));
-            this.generatedCode = true;
-            return [2 /*return*/, code];
+        this.addSystemMessagesToHistory([functionInstructions]);
+    };
+    Conversation.prototype.addMessageModerators = function (messageModerators) {
+        var _a;
+        (_a = this.messageModerators).push.apply(_a, messageModerators);
+    };
+    Conversation.prototype.enforceTokenLimit = function (messages, model) {
+        var _a;
+        return __awaiter(this, void 0, void 0, function () {
+            var resolvedModel, encoder, conversation, encoded, summarizeConversationRequest, referenceSummaryRequest;
+            return __generator(this, function (_b) {
+                switch (_b.label) {
+                    case 0:
+                        if (((_a = this.params.limits) === null || _a === void 0 ? void 0 : _a.enforceLimits) === false) {
+                            return [2 /*return*/];
+                        }
+                        resolvedModel = model ? model : OpenAi_1.DEFAULT_MODEL;
+                        encoder = (0, tiktoken_1.encoding_for_model)(resolvedModel);
+                        conversation = this.history.toString() + messages.join('. ');
+                        encoded = encoder.encode(conversation);
+                        console.log("current tokens: ".concat(encoded.length));
+                        if (encoded.length < this.tokenLimit) {
+                            return [2 /*return*/];
+                        }
+                        summarizeConversationRequest = "First, call the ".concat(exports.summarizeConversationHistoryFunctionName, " function");
+                        return [4 /*yield*/, OpenAi_1.OpenAi.generateResponse([summarizeConversationRequest], model, this.history, this.functions, this.messageModerators, this.params.logLevel)];
+                    case 1:
+                        _b.sent();
+                        referenceSummaryRequest = "If there's a file mentioned in the conversation summary, find and read the file to better respond to my next request. If that doesn't find anything, call the ".concat(PackageFunctions_1.searchLibrariesFunctionName, " function on other keywords in the conversation summary to find a file to read");
+                        return [4 /*yield*/, OpenAi_1.OpenAi.generateResponse([referenceSummaryRequest], model, this.history, this.functions, this.messageModerators, this.params.logLevel)];
+                    case 2:
+                        _b.sent();
+                        return [2 /*return*/];
+                }
+            });
+        });
+    };
+    Conversation.prototype.summarizeConversationHistory = function (summary) {
+        this.clearHistory();
+        this.history.push([{ role: 'assistant', content: "Previous conversation summary: ".concat(summary) }]);
+    };
+    Conversation.prototype.clearHistory = function () {
+        this.history = new MessageHistory_1.MessageHistory();
+        this.history.push(this.systemMessages);
+    };
+    Conversation.prototype.addSystemMessagesToHistory = function (messages, unshift) {
+        var _a, _b, _c;
+        if (unshift === void 0) { unshift = false; }
+        var chatCompletions = messages.map(function (message) {
+            return { role: 'system', content: message };
+        });
+        if (unshift) {
+            (_a = this.history.getMessages()).unshift.apply(_a, chatCompletions);
+            this.history.prune();
+            (_b = this.systemMessages).unshift.apply(_b, chatCompletions);
         }
-      });
-    });
-  };
-  Conversation.prototype.updateCodeFromFile = function (
-    codeToUpdateFilePath,
-    dependencyCodeFilePaths,
-    description,
-    model
-  ) {
-    return __awaiter(this, void 0, void 0, function () {
-      var codeToUpdate, dependencyDescription, _i, dependencyCodeFilePaths_1, dependencyCodeFilePath, dependencCode;
-      return __generator(this, function (_a) {
-        switch (_a.label) {
-          case 0:
-            return [4 /*yield*/, util_node_1.Fs.readFile(codeToUpdateFilePath)];
-          case 1:
-            codeToUpdate = _a.sent();
-            dependencyDescription = 'Assume the following exists:\n';
-            (_i = 0), (dependencyCodeFilePaths_1 = dependencyCodeFilePaths);
-            _a.label = 2;
-          case 2:
-            if (!(_i < dependencyCodeFilePaths_1.length)) return [3 /*break*/, 5];
-            dependencyCodeFilePath = dependencyCodeFilePaths_1[_i];
-            return [4 /*yield*/, util_node_1.Fs.readFile(dependencyCodeFilePath)];
-          case 3:
-            dependencCode = _a.sent();
-            dependencyDescription += dependencCode + '\n\n';
-            _a.label = 4;
-          case 4:
-            _i++;
-            return [3 /*break*/, 2];
-          case 5:
-            this.logger.info('Updating code from file: '.concat(codeToUpdateFilePath));
-            return [4 /*yield*/, this.updateCode(codeToUpdate, dependencyDescription + description, model)];
-          case 6:
-            return [2 /*return*/, _a.sent()];
+        else {
+            this.history.push(chatCompletions);
+            (_c = this.systemMessages).push.apply(_c, chatCompletions);
         }
-      });
-    });
-  };
-  Conversation.prototype.updateCode = function (code, description, model) {
-    return __awaiter(this, void 0, void 0, function () {
-      var updatedCode;
-      return __generator(this, function (_a) {
-        switch (_a.label) {
-          case 0:
-            this.logger.info(
-              'Updating code:\n'
-                .concat(code.slice(0, 150))
-                .concat(code.length > 150 ? '...' : '', '\nFrom description: ')
-                .concat(description)
-            );
-            return [
-              4 /*yield*/,
-              OpenAi_1.OpenAi.updateCode(
-                code,
-                description,
-                model,
-                this.history,
-                this.functions,
-                this.messageModerators,
-                !this.generatedCode,
-                this.params.logLevel
-              ),
-            ];
-          case 1:
-            updatedCode = _a.sent();
-            this.logger.info(
-              'Updated code:\n'.concat(updatedCode.slice(0, 150)).concat(updatedCode.length > 150 ? '...' : '')
-            );
-            this.generatedCode = true;
-            return [2 /*return*/, updatedCode];
+    };
+    Conversation.prototype.addAssistantMessagesToHistory = function (messages, unshift) {
+        var _a;
+        if (unshift === void 0) { unshift = false; }
+        var chatCompletions = messages.map(function (message) {
+            return { role: 'assistant', content: message };
+        });
+        if (unshift) {
+            (_a = this.history.getMessages()).unshift.apply(_a, chatCompletions);
+            this.history.prune();
         }
-      });
-    });
-  };
-  Conversation.prototype.generateList = function (description, model) {
-    return __awaiter(this, void 0, void 0, function () {
-      var list;
-      return __generator(this, function (_a) {
-        switch (_a.label) {
-          case 0:
-            return [
-              4 /*yield*/,
-              OpenAi_1.OpenAi.generateList(
-                description,
-                model,
-                this.history,
-                this.functions,
-                this.messageModerators,
-                !this.generatedList,
-                this.params.logLevel
-              ),
-            ];
-          case 1:
-            list = _a.sent();
-            this.generatedList = true;
-            return [2 /*return*/, list];
+        else {
+            this.history.push(chatCompletions);
         }
-      });
-    });
-  };
-  return Conversation;
-})();
+    };
+    Conversation.prototype.addUserMessagesToHistory = function (messages, unshift) {
+        var _a;
+        if (unshift === void 0) { unshift = false; }
+        var chatCompletions = messages.map(function (message) {
+            return { role: 'user', content: message };
+        });
+        if (unshift) {
+            (_a = this.history.getMessages()).unshift.apply(_a, chatCompletions);
+            this.history.prune();
+        }
+        else {
+            this.history.push(chatCompletions);
+        }
+    };
+    Conversation.prototype.generateResponse = function (messages, model) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, this.enforceTokenLimit(messages, model)];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, OpenAi_1.OpenAi.generateResponse(messages, model, this.history, this.functions, this.messageModerators, this.params.logLevel)];
+                    case 2: return [2 /*return*/, _a.sent()];
+                }
+            });
+        });
+    };
+    Conversation.prototype.generateCode = function (description, model) {
+        return __awaiter(this, void 0, void 0, function () {
+            var code;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.logger.info("Generating code for description:\n".concat(description.join('\n')));
+                        return [4 /*yield*/, OpenAi_1.OpenAi.generateCode(description, model, this.history, this.functions, this.messageModerators, !this.generatedCode, this.params.logLevel)];
+                    case 1:
+                        code = _a.sent();
+                        this.logger.info("Generated code:\n".concat(code.slice(0, 150)).concat(code.length > 150 ? '...' : ''));
+                        this.generatedCode = true;
+                        return [2 /*return*/, code];
+                }
+            });
+        });
+    };
+    Conversation.prototype.updateCodeFromFile = function (codeToUpdateFilePath, dependencyCodeFilePaths, description, model) {
+        return __awaiter(this, void 0, void 0, function () {
+            var codeToUpdate, dependencyDescription, _i, dependencyCodeFilePaths_1, dependencyCodeFilePath, dependencCode;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, util_node_1.Fs.readFile(codeToUpdateFilePath)];
+                    case 1:
+                        codeToUpdate = _a.sent();
+                        dependencyDescription = "Assume the following exists:\n";
+                        _i = 0, dependencyCodeFilePaths_1 = dependencyCodeFilePaths;
+                        _a.label = 2;
+                    case 2:
+                        if (!(_i < dependencyCodeFilePaths_1.length)) return [3 /*break*/, 5];
+                        dependencyCodeFilePath = dependencyCodeFilePaths_1[_i];
+                        return [4 /*yield*/, util_node_1.Fs.readFile(dependencyCodeFilePath)];
+                    case 3:
+                        dependencCode = _a.sent();
+                        dependencyDescription += dependencCode + '\n\n';
+                        _a.label = 4;
+                    case 4:
+                        _i++;
+                        return [3 /*break*/, 2];
+                    case 5:
+                        this.logger.info("Updating code from file: ".concat(codeToUpdateFilePath));
+                        return [4 /*yield*/, this.updateCode(codeToUpdate, dependencyDescription + description, model)];
+                    case 6: return [2 /*return*/, _a.sent()];
+                }
+            });
+        });
+    };
+    Conversation.prototype.updateCode = function (code, description, model) {
+        return __awaiter(this, void 0, void 0, function () {
+            var updatedCode;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.logger.info("Updating code:\n".concat(code.slice(0, 150)).concat(code.length > 150 ? '...' : '', "\nFrom description: ").concat(description));
+                        return [4 /*yield*/, OpenAi_1.OpenAi.updateCode(code, description, model, this.history, this.functions, this.messageModerators, !this.generatedCode, this.params.logLevel)];
+                    case 1:
+                        updatedCode = _a.sent();
+                        this.logger.info("Updated code:\n".concat(updatedCode.slice(0, 150)).concat(updatedCode.length > 150 ? '...' : ''));
+                        this.generatedCode = true;
+                        return [2 /*return*/, updatedCode];
+                }
+            });
+        });
+    };
+    Conversation.prototype.generateList = function (description, model) {
+        return __awaiter(this, void 0, void 0, function () {
+            var list;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, OpenAi_1.OpenAi.generateList(description, model, this.history, this.functions, this.messageModerators, !this.generatedList, this.params.logLevel)];
+                    case 1:
+                        list = _a.sent();
+                        this.generatedList = true;
+                        return [2 /*return*/, list];
+                }
+            });
+        });
+    };
+    return Conversation;
+}());
 exports.Conversation = Conversation;
 exports.summarizeConversationHistoryFunctionName = 'summarizeConversationHistory';
 var summarizeConversationHistoryFunction = function (conversation) {
-  return {
-    definition: {
-      name: exports.summarizeConversationHistoryFunctionName,
-      description: 'Clear the conversation history and summarize what was in it',
-      parameters: {
-        type: 'object',
-        properties: {
-          summary: {
-            type: 'string',
-            description: 'A 1-3 sentence summary of the current chat history',
-          },
+    return {
+        definition: {
+            name: exports.summarizeConversationHistoryFunctionName,
+            description: 'Clear the conversation history and summarize what was in it',
+            parameters: {
+                type: 'object',
+                properties: {
+                    summary: {
+                        type: 'string',
+                        description: 'A 1-3 sentence summary of the current chat history',
+                    },
+                },
+                required: ['summary'],
+            },
         },
-        required: ['summary'],
-      },
-    },
-    call: function (params) {
-      return __awaiter(void 0, void 0, void 0, function () {
-        return __generator(this, function (_a) {
-          return [2 /*return*/, conversation.summarizeConversationHistory(params.summary)];
-        });
-      });
-    },
-  };
+        call: function (params) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
+            return [2 /*return*/, conversation.summarizeConversationHistory(params.summary)];
+        }); }); },
+    };
 };
 exports.summarizeConversationHistoryFunction = summarizeConversationHistoryFunction;
-//# sourceMappingURL=Conversation.js.map
+//# sourceMappingURL=Conversation.js.map