@mastra/rag 0.0.2-alpha.49 → 0.0.2-alpha.50

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/CHANGELOG.md CHANGED
@@ -1,5 +1,11 @@
1
1
  # @mastra/rag
2
2
 
3
+ ## 0.0.2-alpha.50
4
+
5
+ ### Patch Changes
6
+
7
+ - 1616f70: Added transaction handling to pgvector
8
+
3
9
  ## 0.0.2-alpha.49
4
10
 
5
11
  ### Patch Changes
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/pg/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAGrE,OAAO,EAAE,MAAM,EAAqC,MAAM,UAAU,CAAC;AAErE,qBAAa,QAAS,SAAQ,YAAY;IACxC,OAAO,CAAC,IAAI,CAAU;gBAEV,gBAAgB,EAAE,MAAM;IAoB9B,KAAK,CACT,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,MAAM,EAAE,EACrB,IAAI,GAAE,MAAW,EACjB,MAAM,CAAC,EAAE,MAAM,EACf,aAAa,GAAE,OAAe,EAC9B,QAAQ,GAAE,MAAU,GACnB,OAAO,CAAC,WAAW,EAAE,CAAC;IAuEnB,MAAM,CACV,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EAAE,EAAE,EACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,EAChC,GAAG,CAAC,EAAE,MAAM,EAAE,GACb,OAAO,CAAC,MAAM,EAAE,CAAC;IA8Bd,WAAW,CACf,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,EACjB,MAAM,GAAE,QAAQ,GAAG,WAAW,GAAG,YAAuB,GACvD,OAAO,CAAC,IAAI,CAAC;IAqDV,WAAW,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAiBhC,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;IA6DrD,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAa7C,aAAa,CAAC,SAAS,EAAE,MAAM;IAY/B,UAAU;CAGjB"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/pg/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAGrE,OAAO,EAAE,MAAM,EAAqC,MAAM,UAAU,CAAC;AAErE,qBAAa,QAAS,SAAQ,YAAY;IACxC,OAAO,CAAC,IAAI,CAAU;gBAEV,gBAAgB,EAAE,MAAM;IAoB9B,KAAK,CACT,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,MAAM,EAAE,EACrB,IAAI,GAAE,MAAW,EACjB,MAAM,CAAC,EAAE,MAAM,EACf,aAAa,GAAE,OAAe,EAC9B,QAAQ,GAAE,MAAU,GACnB,OAAO,CAAC,WAAW,EAAE,CAAC;IAuEnB,MAAM,CACV,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EAAE,EAAE,EACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,EAChC,GAAG,CAAC,EAAE,MAAM,EAAE,GACb,OAAO,CAAC,MAAM,EAAE,CAAC;IAiCd,WAAW,CACf,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,EACjB,MAAM,GAAE,QAAQ,GAAG,WAAW,GAAG,YAAuB,GACvD,OAAO,CAAC,IAAI,CAAC;IAqDV,WAAW,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAiBhC,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;IA6DrD,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAa7C,aAAa,CAAC,SAAS,EAAE,MAAM;IAY/B,UAAU;CAGjB"}
@@ -2224,23 +2224,26 @@ var PgVector = /*#__PURE__*/function (_MastraVector) {
2224
2224
  _context2.next = 8;
2225
2225
  break;
2226
2226
  case 15:
2227
+ _context2.next = 17;
2228
+ return client.query('COMMIT');
2229
+ case 17:
2227
2230
  return _context2.abrupt("return", vectorIds);
2228
- case 18:
2229
- _context2.prev = 18;
2231
+ case 20:
2232
+ _context2.prev = 20;
2230
2233
  _context2.t0 = _context2["catch"](3);
2231
- _context2.next = 22;
2234
+ _context2.next = 24;
2232
2235
  return client.query('ROLLBACK');
2233
- case 22:
2236
+ case 24:
2234
2237
  throw _context2.t0;
2235
- case 23:
2236
- _context2.prev = 23;
2238
+ case 25:
2239
+ _context2.prev = 25;
2237
2240
  client.release();
2238
- return _context2.finish(23);
2239
- case 26:
2241
+ return _context2.finish(25);
2242
+ case 28:
2240
2243
  case "end":
2241
2244
  return _context2.stop();
2242
2245
  }
2243
- }, _callee2, this, [[3, 18, 23, 26]]);
2246
+ }, _callee2, this, [[3, 20, 25, 28]]);
2244
2247
  }));
2245
2248
  function upsert(_x7, _x8, _x9, _x10) {
2246
2249
  return _upsert.apply(this, arguments);
@@ -3880,8 +3883,7 @@ var createFilter = function createFilter(filter, vectorFilterType) {
3880
3883
  };
3881
3884
  }
3882
3885
  };
3883
- // Separate function to handle vector query search
3884
- // Can be imported and used in custom tools
3886
+ // Helper function to handle vector query search
3885
3887
  var vectorQuerySearch = /*#__PURE__*/function () {
3886
3888
  var _ref4 = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(_ref3) {
3887
3889
  var indexName, vectorStore, queryText, options, _ref3$queryFilter, queryFilter, topK, _ref3$includeVectors, includeVectors, _yield$embed, embedding, results;
@@ -4109,7 +4111,7 @@ var createGraphRAGTool = function createGraphRAGTool(_ref8) {
4109
4111
  }()
4110
4112
  });
4111
4113
  };
4112
- var createDocumentChunker = function createDocumentChunker(_ref10) {
4114
+ var createDocumentChunkerTool = function createDocumentChunkerTool(_ref10) {
4113
4115
  var doc = _ref10.doc,
4114
4116
  _ref10$params = _ref10.params,
4115
4117
  params = _ref10$params === void 0 ? {
@@ -4157,9 +4159,8 @@ exports.PineconeVector = PineconeVector;
4157
4159
  exports.QdrantVector = QdrantVector;
4158
4160
  exports.Reranker = Reranker;
4159
4161
  exports.UpstashVector = UpstashVector;
4160
- exports.createDocumentChunker = createDocumentChunker;
4162
+ exports.createDocumentChunkerTool = createDocumentChunkerTool;
4161
4163
  exports.createGraphRAGTool = createGraphRAGTool;
4162
4164
  exports.createVectorQueryTool = createVectorQueryTool;
4163
4165
  exports.embed = embed;
4164
- exports.vectorQuerySearch = vectorQuerySearch;
4165
4166
  //# sourceMappingURL=rag.cjs.development.js.map