chat.cjs 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. exports.isChatMessageChunk = exports.isChatMessage = exports.ChatMessageChunk = exports.ChatMessage = void 0;
  4. const base_js_1 = require("./base.cjs");
  5. /**
  6. * Represents a chat message in a conversation.
  7. */
  8. class ChatMessage extends base_js_1.BaseMessage {
  9. static lc_name() {
  10. return "ChatMessage";
  11. }
  12. static _chatMessageClass() {
  13. return ChatMessage;
  14. }
  15. constructor(fields, role) {
  16. if (typeof fields === "string") {
  17. // eslint-disable-next-line no-param-reassign, @typescript-eslint/no-non-null-assertion
  18. fields = { content: fields, role: role };
  19. }
  20. super(fields);
  21. Object.defineProperty(this, "role", {
  22. enumerable: true,
  23. configurable: true,
  24. writable: true,
  25. value: void 0
  26. });
  27. this.role = fields.role;
  28. }
  29. _getType() {
  30. return "generic";
  31. }
  32. static isInstance(message) {
  33. return message._getType() === "generic";
  34. }
  35. get _printableFields() {
  36. return {
  37. ...super._printableFields,
  38. role: this.role,
  39. };
  40. }
  41. }
  42. exports.ChatMessage = ChatMessage;
  43. /**
  44. * Represents a chunk of a chat message, which can be concatenated with
  45. * other chat message chunks.
  46. */
  47. class ChatMessageChunk extends base_js_1.BaseMessageChunk {
  48. static lc_name() {
  49. return "ChatMessageChunk";
  50. }
  51. constructor(fields, role) {
  52. if (typeof fields === "string") {
  53. // eslint-disable-next-line no-param-reassign, @typescript-eslint/no-non-null-assertion
  54. fields = { content: fields, role: role };
  55. }
  56. super(fields);
  57. Object.defineProperty(this, "role", {
  58. enumerable: true,
  59. configurable: true,
  60. writable: true,
  61. value: void 0
  62. });
  63. this.role = fields.role;
  64. }
  65. _getType() {
  66. return "generic";
  67. }
  68. concat(chunk) {
  69. return new ChatMessageChunk({
  70. content: (0, base_js_1.mergeContent)(this.content, chunk.content),
  71. additional_kwargs: (0, base_js_1._mergeDicts)(this.additional_kwargs, chunk.additional_kwargs),
  72. response_metadata: (0, base_js_1._mergeDicts)(this.response_metadata, chunk.response_metadata),
  73. role: this.role,
  74. id: this.id ?? chunk.id,
  75. });
  76. }
  77. get _printableFields() {
  78. return {
  79. ...super._printableFields,
  80. role: this.role,
  81. };
  82. }
  83. }
  84. exports.ChatMessageChunk = ChatMessageChunk;
  85. function isChatMessage(x) {
  86. return x._getType() === "generic";
  87. }
  88. exports.isChatMessage = isChatMessage;
  89. function isChatMessageChunk(x) {
  90. return x._getType() === "generic";
  91. }
  92. exports.isChatMessageChunk = isChatMessageChunk;