coze-coding-dev-sdk 0.2.2 → 0.2.5

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.
Files changed (83) hide show
  1. package/README.md +0 -13
  2. package/dist/cli/commands/chat.d.ts.map +1 -1
  3. package/dist/cli/commands/chat.js +0 -1
  4. package/dist/cli/commands/chat.js.map +1 -1
  5. package/dist/cli/commands/db.d.ts +3 -0
  6. package/dist/cli/commands/db.d.ts.map +1 -0
  7. package/dist/cli/commands/db.js +91 -0
  8. package/dist/cli/commands/db.js.map +1 -0
  9. package/dist/cli/commands/image.d.ts.map +1 -1
  10. package/dist/cli/commands/image.js +1 -2
  11. package/dist/cli/commands/image.js.map +1 -1
  12. package/dist/cli/commands/search.d.ts.map +1 -1
  13. package/dist/cli/commands/search.js +1 -2
  14. package/dist/cli/commands/search.js.map +1 -1
  15. package/dist/cli/commands/video.d.ts.map +1 -1
  16. package/dist/cli/commands/video.js +1 -2
  17. package/dist/cli/commands/video.js.map +1 -1
  18. package/dist/cli/commands/voice.d.ts.map +1 -1
  19. package/dist/cli/commands/voice.js +1 -2
  20. package/dist/cli/commands/voice.js.map +1 -1
  21. package/dist/cli/index.js +3 -1
  22. package/dist/cli/index.js.map +1 -1
  23. package/dist/cli/utils.d.ts +0 -1
  24. package/dist/cli/utils.d.ts.map +1 -1
  25. package/dist/cli/utils.js +0 -29
  26. package/dist/cli/utils.js.map +1 -1
  27. package/dist/core/client.d.ts +1 -5
  28. package/dist/core/client.d.ts.map +1 -1
  29. package/dist/core/client.js +1 -81
  30. package/dist/core/client.js.map +1 -1
  31. package/dist/database/client.d.ts +13 -0
  32. package/dist/database/client.d.ts.map +1 -0
  33. package/dist/database/client.js +142 -0
  34. package/dist/database/client.js.map +1 -0
  35. package/dist/database/index.d.ts +4 -0
  36. package/dist/database/index.d.ts.map +1 -0
  37. package/dist/database/index.js +20 -0
  38. package/dist/database/index.js.map +1 -0
  39. package/dist/database/migration.d.ts +7 -0
  40. package/dist/database/migration.d.ts.map +1 -0
  41. package/dist/database/migration.js +188 -0
  42. package/dist/database/migration.js.map +1 -0
  43. package/dist/database/models.d.ts +26 -0
  44. package/dist/database/models.d.ts.map +1 -0
  45. package/dist/database/models.js +15 -0
  46. package/dist/database/models.js.map +1 -0
  47. package/dist/image/client.d.ts +1 -1
  48. package/dist/image/client.d.ts.map +1 -1
  49. package/dist/image/client.js +2 -2
  50. package/dist/image/client.js.map +1 -1
  51. package/dist/index.d.ts +2 -0
  52. package/dist/index.d.ts.map +1 -1
  53. package/dist/index.js +2 -0
  54. package/dist/index.js.map +1 -1
  55. package/dist/s3/client.d.ts +66 -0
  56. package/dist/s3/client.d.ts.map +1 -0
  57. package/dist/s3/client.js +253 -0
  58. package/dist/s3/client.js.map +1 -0
  59. package/dist/s3/index.d.ts +3 -0
  60. package/dist/s3/index.d.ts.map +1 -0
  61. package/dist/s3/index.js +19 -0
  62. package/dist/s3/index.js.map +1 -0
  63. package/dist/s3/models.d.ts +23 -0
  64. package/dist/s3/models.d.ts.map +1 -0
  65. package/dist/s3/models.js +15 -0
  66. package/dist/s3/models.js.map +1 -0
  67. package/dist/search/client.d.ts +1 -1
  68. package/dist/search/client.d.ts.map +1 -1
  69. package/dist/search/client.js +2 -2
  70. package/dist/search/client.js.map +1 -1
  71. package/dist/video/client.d.ts +1 -1
  72. package/dist/video/client.d.ts.map +1 -1
  73. package/dist/video/client.js +2 -2
  74. package/dist/video/client.js.map +1 -1
  75. package/dist/voice/asr.d.ts +2 -2
  76. package/dist/voice/asr.d.ts.map +1 -1
  77. package/dist/voice/asr.js +3 -3
  78. package/dist/voice/asr.js.map +1 -1
  79. package/dist/voice/tts.d.ts +1 -1
  80. package/dist/voice/tts.d.ts.map +1 -1
  81. package/dist/voice/tts.js +2 -2
  82. package/dist/voice/tts.js.map +1 -1
  83. package/package.json +10 -3
@@ -0,0 +1,142 @@
1
+ "use strict";
2
+ /**
3
+ * Database client module
4
+ */
5
+ var __importDefault = (this && this.__importDefault) || function (mod) {
6
+ return (mod && mod.__esModule) ? mod : { "default": mod };
7
+ };
8
+ Object.defineProperty(exports, "__esModule", { value: true });
9
+ exports.loadEnv = loadEnv;
10
+ exports.getDbUrl = getDbUrl;
11
+ exports.getPool = getPool;
12
+ exports.getDb = getDb;
13
+ exports.getClient = getClient;
14
+ exports.closePool = closePool;
15
+ const node_postgres_1 = require("drizzle-orm/node-postgres");
16
+ const pg_1 = __importDefault(require("pg"));
17
+ const child_process_1 = require("child_process");
18
+ const models_1 = require("./models");
19
+ let envLoaded = false;
20
+ function loadEnv() {
21
+ if (envLoaded || process.env.PGDATABASE_URL) {
22
+ return;
23
+ }
24
+ try {
25
+ try {
26
+ require('dotenv').config();
27
+ if (process.env.PGDATABASE_URL) {
28
+ envLoaded = true;
29
+ return;
30
+ }
31
+ }
32
+ catch {
33
+ // dotenv not available
34
+ }
35
+ const pythonCode = `
36
+ import os
37
+ import sys
38
+ try:
39
+ from coze_workload_identity import Client
40
+ client = Client()
41
+ env_vars = client.get_project_env_vars()
42
+ client.close()
43
+ for env_var in env_vars:
44
+ print(f"{env_var.key}={env_var.value}")
45
+ except Exception as e:
46
+ print(f"# Error: {e}", file=sys.stderr)
47
+ `;
48
+ const output = (0, child_process_1.execSync)(`python3 -c '${pythonCode.replace(/'/g, "'\"'\"'")}'`, {
49
+ encoding: 'utf-8',
50
+ timeout: 10000,
51
+ stdio: ['pipe', 'pipe', 'pipe'],
52
+ });
53
+ const lines = output.trim().split('\n');
54
+ for (const line of lines) {
55
+ if (line.startsWith('#'))
56
+ continue;
57
+ const eqIndex = line.indexOf('=');
58
+ if (eqIndex > 0) {
59
+ const key = line.substring(0, eqIndex);
60
+ let value = line.substring(eqIndex + 1);
61
+ if ((value.startsWith("'") && value.endsWith("'")) ||
62
+ (value.startsWith('"') && value.endsWith('"'))) {
63
+ value = value.slice(1, -1);
64
+ }
65
+ if (!process.env[key]) {
66
+ process.env[key] = value;
67
+ }
68
+ }
69
+ }
70
+ envLoaded = true;
71
+ }
72
+ catch {
73
+ // Silently fail
74
+ }
75
+ }
76
+ function getDbUrl() {
77
+ loadEnv();
78
+ let url = process.env.PGDATABASE_URL || '';
79
+ if (!url) {
80
+ throw new Error('Database URL not configured. Set PGDATABASE_URL environment variable.\n' +
81
+ 'Did you create a database? You can create one via the Coze Coding platform.');
82
+ }
83
+ if (url.startsWith('postgres://')) {
84
+ url = 'postgresql://' + url.substring('postgres://'.length);
85
+ }
86
+ return url;
87
+ }
88
+ let pool = null;
89
+ let db = null;
90
+ async function createPoolWithRetry(config) {
91
+ const url = getDbUrl();
92
+ const newPool = new pg_1.default.Pool({
93
+ connectionString: url,
94
+ max: config?.maxConnections ?? models_1.DatabaseConfig.MAX_CONNECTIONS,
95
+ idleTimeoutMillis: config?.idleTimeoutMillis ?? models_1.DatabaseConfig.IDLE_TIMEOUT_MILLIS,
96
+ connectionTimeoutMillis: config?.connectionTimeoutMillis ?? models_1.DatabaseConfig.CONNECTION_TIMEOUT_MILLIS,
97
+ });
98
+ const startTime = Date.now();
99
+ let lastError = null;
100
+ while (Date.now() - startTime < models_1.DatabaseConfig.MAX_RETRY_TIME) {
101
+ try {
102
+ const client = await newPool.connect();
103
+ await client.query('SELECT 1');
104
+ client.release();
105
+ return newPool;
106
+ }
107
+ catch (e) {
108
+ lastError = e;
109
+ const elapsed = ((Date.now() - startTime) / 1000).toFixed(1);
110
+ console.warn(`Database connection failed, retrying... (elapsed: ${elapsed}s)`);
111
+ await new Promise((resolve) => setTimeout(resolve, Math.min(1000, models_1.DatabaseConfig.MAX_RETRY_TIME - (Date.now() - startTime))));
112
+ }
113
+ }
114
+ console.error(`Database connection failed after ${models_1.DatabaseConfig.MAX_RETRY_TIME / 1000}s: ${lastError?.message}`);
115
+ throw lastError;
116
+ }
117
+ async function getPool(config) {
118
+ loadEnv();
119
+ if (!pool) {
120
+ pool = await createPoolWithRetry(config);
121
+ }
122
+ return pool;
123
+ }
124
+ async function getDb(schema) {
125
+ if (!db) {
126
+ const p = await getPool();
127
+ db = schema ? (0, node_postgres_1.drizzle)(p, { schema }) : (0, node_postgres_1.drizzle)(p);
128
+ }
129
+ return db;
130
+ }
131
+ async function getClient() {
132
+ const p = await getPool();
133
+ return p.connect();
134
+ }
135
+ async function closePool() {
136
+ if (pool) {
137
+ await pool.end();
138
+ pool = null;
139
+ db = null;
140
+ }
141
+ }
142
+ //# sourceMappingURL=client.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/database/client.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;AASH,0BAyDC;AAED,4BAiBC;AAsCD,0BAMC;AAED,sBAQC;AAED,8BAGC;AAED,8BAMC;AAtJD,6DAAoD;AACpD,4CAAoB;AACpB,iDAAyC;AACzC,qCAA8D;AAE9D,IAAI,SAAS,GAAG,KAAK,CAAC;AAEtB,SAAgB,OAAO;IACrB,IAAI,SAAS,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;QAC5C,OAAO;IACT,CAAC;IAED,IAAI,CAAC;QACH,IAAI,CAAC;YACH,OAAO,CAAC,QAAQ,CAAC,CAAC,MAAM,EAAE,CAAC;YAC3B,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;gBAC/B,SAAS,GAAG,IAAI,CAAC;gBACjB,OAAO;YACT,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,uBAAuB;QACzB,CAAC;QAED,MAAM,UAAU,GAAG;;;;;;;;;;;;CAYtB,CAAC;QAEE,MAAM,MAAM,GAAG,IAAA,wBAAQ,EAAC,eAAe,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC,GAAG,EAAE;YAC7E,QAAQ,EAAE,OAAO;YACjB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;SAChC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACxC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS;YACnC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAClC,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;gBAChB,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;gBACvC,IAAI,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;gBACxC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;oBAC9C,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;oBACnD,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC7B,CAAC;gBACD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;oBACtB,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;gBAC3B,CAAC;YACH,CAAC;QACH,CAAC;QAED,SAAS,GAAG,IAAI,CAAC;IACnB,CAAC;IAAC,MAAM,CAAC;QACP,gBAAgB;IAClB,CAAC;AACH,CAAC;AAED,SAAgB,QAAQ;IACtB,OAAO,EAAE,CAAC;IAEV,IAAI,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE,CAAC;IAE3C,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,MAAM,IAAI,KAAK,CACb,yEAAyE;YACzE,6EAA6E,CAC9E,CAAC;IACJ,CAAC;IAED,IAAI,GAAG,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QAClC,GAAG,GAAG,eAAe,GAAG,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IAC9D,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC;AAED,IAAI,IAAI,GAAmB,IAAI,CAAC;AAChC,IAAI,EAAE,GAAsC,IAAI,CAAC;AAEjD,KAAK,UAAU,mBAAmB,CAAC,MAA2B;IAC5D,MAAM,GAAG,GAAG,QAAQ,EAAE,CAAC;IAEvB,MAAM,OAAO,GAAG,IAAI,YAAE,CAAC,IAAI,CAAC;QAC1B,gBAAgB,EAAE,GAAG;QACrB,GAAG,EAAE,MAAM,EAAE,cAAc,IAAI,uBAAc,CAAC,eAAe;QAC7D,iBAAiB,EAAE,MAAM,EAAE,iBAAiB,IAAI,uBAAc,CAAC,mBAAmB;QAClF,uBAAuB,EAAE,MAAM,EAAE,uBAAuB,IAAI,uBAAc,CAAC,yBAAyB;KACrG,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,IAAI,SAAS,GAAiB,IAAI,CAAC;IAEnC,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,GAAG,uBAAc,CAAC,cAAc,EAAE,CAAC;QAC9D,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;YACvC,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC/B,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,SAAS,GAAG,CAAU,CAAC;YACvB,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC7D,OAAO,CAAC,IAAI,CAAC,qDAAqD,OAAO,IAAI,CAAC,CAAC;YAC/E,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5B,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,uBAAc,CAAC,cAAc,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,CAC9F,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,CAAC,KAAK,CAAC,oCAAoC,uBAAc,CAAC,cAAc,GAAG,IAAI,MAAM,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;IAClH,MAAM,SAAS,CAAC;AAClB,CAAC;AAEM,KAAK,UAAU,OAAO,CAAC,MAA2B;IACvD,OAAO,EAAE,CAAC;IACV,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,IAAI,GAAG,MAAM,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAEM,KAAK,UAAU,KAAK,CACzB,MAAU;IAEV,IAAI,CAAC,EAAE,EAAE,CAAC;QACR,MAAM,CAAC,GAAG,MAAM,OAAO,EAAE,CAAC;QAC1B,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,IAAA,uBAAO,EAAC,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,IAAA,uBAAO,EAAC,CAAC,CAAC,CAAC;IACpD,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAEM,KAAK,UAAU,SAAS;IAC7B,MAAM,CAAC,GAAG,MAAM,OAAO,EAAE,CAAC;IAC1B,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;AACrB,CAAC;AAEM,KAAK,UAAU,SAAS;IAC7B,IAAI,IAAI,EAAE,CAAC;QACT,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;QACjB,IAAI,GAAG,IAAI,CAAC;QACZ,EAAE,GAAG,IAAI,CAAC;IACZ,CAAC;AACH,CAAC"}
@@ -0,0 +1,4 @@
1
+ export * from './client';
2
+ export * from './migration';
3
+ export * from './models';
4
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/database/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,aAAa,CAAC;AAC5B,cAAc,UAAU,CAAC"}
@@ -0,0 +1,20 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
+ for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
+ };
16
+ Object.defineProperty(exports, "__esModule", { value: true });
17
+ __exportStar(require("./client"), exports);
18
+ __exportStar(require("./migration"), exports);
19
+ __exportStar(require("./models"), exports);
20
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/database/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAAyB;AACzB,8CAA4B;AAC5B,2CAAyB"}
@@ -0,0 +1,7 @@
1
+ /**
2
+ * Database migration module
3
+ */
4
+ import { GenerateModelsOptions, UpgradeOptions } from './models';
5
+ export declare function generateModels(options?: GenerateModelsOptions): Promise<void>;
6
+ export declare function upgrade(options?: UpgradeOptions): Promise<void>;
7
+ //# sourceMappingURL=migration.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"migration.d.ts","sourceRoot":"","sources":["../../src/database/migration.ts"],"names":[],"mappings":"AAAA;;GAEG;AAMH,OAAO,EAAE,qBAAqB,EAAE,cAAc,EAAkB,MAAM,UAAU,CAAC;AAqBjF,wBAAsB,cAAc,CAAC,OAAO,GAAE,qBAA0B,GAAG,OAAO,CAAC,IAAI,CAAC,CA2FvF;AAED,wBAAsB,OAAO,CAAC,OAAO,GAAE,cAAmB,GAAG,OAAO,CAAC,IAAI,CAAC,CA6DzE"}
@@ -0,0 +1,188 @@
1
+ "use strict";
2
+ /**
3
+ * Database migration module
4
+ */
5
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
6
+ if (k2 === undefined) k2 = k;
7
+ var desc = Object.getOwnPropertyDescriptor(m, k);
8
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
9
+ desc = { enumerable: true, get: function() { return m[k]; } };
10
+ }
11
+ Object.defineProperty(o, k2, desc);
12
+ }) : (function(o, m, k, k2) {
13
+ if (k2 === undefined) k2 = k;
14
+ o[k2] = m[k];
15
+ }));
16
+ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
17
+ Object.defineProperty(o, "default", { enumerable: true, value: v });
18
+ }) : function(o, v) {
19
+ o["default"] = v;
20
+ });
21
+ var __importStar = (this && this.__importStar) || (function () {
22
+ var ownKeys = function(o) {
23
+ ownKeys = Object.getOwnPropertyNames || function (o) {
24
+ var ar = [];
25
+ for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
26
+ return ar;
27
+ };
28
+ return ownKeys(o);
29
+ };
30
+ return function (mod) {
31
+ if (mod && mod.__esModule) return mod;
32
+ var result = {};
33
+ if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
34
+ __setModuleDefault(result, mod);
35
+ return result;
36
+ };
37
+ })();
38
+ Object.defineProperty(exports, "__esModule", { value: true });
39
+ exports.generateModels = generateModels;
40
+ exports.upgrade = upgrade;
41
+ const child_process_1 = require("child_process");
42
+ const fs = __importStar(require("fs"));
43
+ const path = __importStar(require("path"));
44
+ const os = __importStar(require("os"));
45
+ const models_1 = require("./models");
46
+ const client_1 = require("./client");
47
+ /**
48
+ * Fix Drizzle bug: introspect generates numbers beyond JS precision
49
+ * Replace .generatedByDefaultAsIdentity({...}) with .generatedByDefaultAsIdentity()
50
+ */
51
+ function fixGeneratedIdentity(filePath) {
52
+ if (!fs.existsSync(filePath))
53
+ return;
54
+ let content = fs.readFileSync(filePath, 'utf-8');
55
+ const fixed = content.replace(/\.generatedByDefaultAsIdentity\(\{[^}]*\}\)/g, '.generatedByDefaultAsIdentity()');
56
+ if (fixed !== content) {
57
+ fs.writeFileSync(filePath, fixed);
58
+ }
59
+ }
60
+ async function generateModels(options = {}) {
61
+ const { outputPath = models_1.DatabaseConfig.DEFAULT_SCHEMA_PATH, verbose = false, } = options;
62
+ (0, client_1.loadEnv)();
63
+ const dbUrl = (0, client_1.getDbUrl)();
64
+ const outputDir = path.dirname(outputPath);
65
+ if (outputDir && !fs.existsSync(outputDir)) {
66
+ fs.mkdirSync(outputDir, { recursive: true });
67
+ }
68
+ // Use /tmp for drizzle-kit output, config in outputDir for module resolution
69
+ const tempOutDir = fs.mkdtempSync(path.join(os.tmpdir(), 'drizzle-'));
70
+ const tempConfigPath = path.join(outputDir, `drizzle.config.ts`);
71
+ const drizzleConfig = `
72
+ import { defineConfig } from 'drizzle-kit';
73
+
74
+ export default defineConfig({
75
+ dialect: 'postgresql',
76
+ out: '${tempOutDir}',
77
+ dbCredentials: {
78
+ url: '${dbUrl}',
79
+ },
80
+ });
81
+ `;
82
+ try {
83
+ fs.writeFileSync(tempConfigPath, drizzleConfig);
84
+ if (verbose) {
85
+ console.log('Introspecting database schema...');
86
+ }
87
+ const cmd = `npx drizzle-kit introspect --config=${tempConfigPath}`;
88
+ if (verbose) {
89
+ console.log(`Running: ${cmd}`);
90
+ }
91
+ (0, child_process_1.execSync)(cmd, {
92
+ encoding: 'utf-8',
93
+ stdio: verbose ? 'inherit' : 'pipe',
94
+ timeout: 60000,
95
+ });
96
+ // Fix Drizzle bug in generated schema
97
+ const generatedSchema = path.join(tempOutDir, 'schema.ts');
98
+ const generatedRelations = path.join(tempOutDir, 'relations.ts');
99
+ if (fs.existsSync(generatedSchema)) {
100
+ fixGeneratedIdentity(generatedSchema);
101
+ fs.copyFileSync(generatedSchema, outputPath);
102
+ if (verbose) {
103
+ console.log(`✓ Copied schema.ts to ${outputDir}/`);
104
+ }
105
+ }
106
+ if (fs.existsSync(generatedRelations)) {
107
+ const relationsPath = path.join(outputDir, 'relations.ts');
108
+ fs.copyFileSync(generatedRelations, relationsPath);
109
+ if (verbose) {
110
+ console.log(`✓ Copied relations.ts to ${outputDir}/`);
111
+ }
112
+ }
113
+ // Git add generated files
114
+ try {
115
+ (0, child_process_1.execSync)('git add -A', {
116
+ encoding: 'utf-8',
117
+ stdio: 'pipe',
118
+ cwd: process.cwd(),
119
+ });
120
+ }
121
+ catch {
122
+ // Ignore git errors (might not be in a git repo)
123
+ }
124
+ if (verbose) {
125
+ console.log(`✓ Schema generated successfully`);
126
+ }
127
+ }
128
+ finally {
129
+ // Cleanup temp files
130
+ if (fs.existsSync(tempConfigPath)) {
131
+ fs.unlinkSync(tempConfigPath);
132
+ }
133
+ if (fs.existsSync(tempOutDir)) {
134
+ fs.rmSync(tempOutDir, { recursive: true });
135
+ }
136
+ }
137
+ }
138
+ async function upgrade(options = {}) {
139
+ const { schemaPath = models_1.DatabaseConfig.DEFAULT_SCHEMA_PATH, projectPath = process.cwd(), verbose = false, } = options;
140
+ (0, client_1.loadEnv)();
141
+ const dbUrl = (0, client_1.getDbUrl)();
142
+ const fullSchemaPath = path.isAbsolute(schemaPath)
143
+ ? schemaPath
144
+ : path.join(projectPath, schemaPath);
145
+ if (!fs.existsSync(fullSchemaPath)) {
146
+ throw new Error(`Schema file not found: ${fullSchemaPath}`);
147
+ }
148
+ // Config in schema directory for module resolution
149
+ const schemaDir = path.dirname(fullSchemaPath);
150
+ const tempConfigPath = path.join(schemaDir, `drizzle.config.ts`);
151
+ const drizzleConfig = `
152
+ import { defineConfig } from 'drizzle-kit';
153
+
154
+ export default defineConfig({
155
+ dialect: 'postgresql',
156
+ schema: '${fullSchemaPath}',
157
+ dbCredentials: {
158
+ url: '${dbUrl}',
159
+ },
160
+ });
161
+ `;
162
+ try {
163
+ fs.writeFileSync(tempConfigPath, drizzleConfig);
164
+ if (verbose) {
165
+ console.log('Pushing schema changes to database...');
166
+ }
167
+ // Use --force to skip confirmation
168
+ const cmd = `npx drizzle-kit push --force --config=${tempConfigPath}`;
169
+ if (verbose) {
170
+ console.log(`Running: ${cmd}`);
171
+ }
172
+ (0, child_process_1.execSync)(cmd, {
173
+ encoding: 'utf-8',
174
+ stdio: verbose ? 'inherit' : 'pipe',
175
+ timeout: 120000,
176
+ cwd: projectPath,
177
+ });
178
+ if (verbose) {
179
+ console.log('✅ Database migration completed successfully');
180
+ }
181
+ }
182
+ finally {
183
+ if (fs.existsSync(tempConfigPath)) {
184
+ fs.unlinkSync(tempConfigPath);
185
+ }
186
+ }
187
+ }
188
+ //# sourceMappingURL=migration.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"migration.js","sourceRoot":"","sources":["../../src/database/migration.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA2BH,wCA2FC;AAED,0BA6DC;AAnLD,iDAAyC;AACzC,uCAAyB;AACzB,2CAA6B;AAC7B,uCAAyB;AACzB,qCAAiF;AACjF,qCAA6C;AAE7C;;;GAGG;AACH,SAAS,oBAAoB,CAAC,QAAgB;IAC5C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;QAAE,OAAO;IAErC,IAAI,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACjD,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAC3B,8CAA8C,EAC9C,iCAAiC,CAClC,CAAC;IAEF,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACpC,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,cAAc,CAAC,UAAiC,EAAE;IACtE,MAAM,EACJ,UAAU,GAAG,uBAAc,CAAC,mBAAmB,EAC/C,OAAO,GAAG,KAAK,GAChB,GAAG,OAAO,CAAC;IAEZ,IAAA,gBAAO,GAAE,CAAC;IACV,MAAM,KAAK,GAAG,IAAA,iBAAQ,GAAE,CAAC;IAEzB,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC3C,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3C,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED,6EAA6E;IAC7E,MAAM,UAAU,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,UAAU,CAAC,CAAC,CAAC;IACtE,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;IAEjE,MAAM,aAAa,GAAG;;;;;UAKd,UAAU;;YAER,KAAK;;;CAGhB,CAAC;IAEA,IAAI,CAAC;QACH,EAAE,CAAC,aAAa,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;QAEhD,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,GAAG,GAAG,uCAAuC,cAAc,EAAE,CAAC;QACpE,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,EAAE,CAAC,CAAC;QACjC,CAAC;QAED,IAAA,wBAAQ,EAAC,GAAG,EAAE;YACZ,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM;YACnC,OAAO,EAAE,KAAK;SACf,CAAC,CAAC;QAEH,sCAAsC;QACtC,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAC3D,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;QAEjE,IAAI,EAAE,CAAC,UAAU,CAAC,eAAe,CAAC,EAAE,CAAC;YACnC,oBAAoB,CAAC,eAAe,CAAC,CAAC;YACtC,EAAE,CAAC,YAAY,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;YAC7C,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,GAAG,CAAC,yBAAyB,SAAS,GAAG,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;QAED,IAAI,EAAE,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;YAC3D,EAAE,CAAC,YAAY,CAAC,kBAAkB,EAAE,aAAa,CAAC,CAAC;YACnD,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,GAAG,CAAC,4BAA4B,SAAS,GAAG,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QAED,0BAA0B;QAC1B,IAAI,CAAC;YACH,IAAA,wBAAQ,EAAC,YAAY,EAAE;gBACrB,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,MAAM;gBACb,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE;aACnB,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,iDAAiD;QACnD,CAAC;QAED,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;YAAS,CAAC;QACT,qBAAqB;QACrB,IAAI,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;YAClC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YAC9B,EAAE,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,OAAO,CAAC,UAA0B,EAAE;IACxD,MAAM,EACJ,UAAU,GAAG,uBAAc,CAAC,mBAAmB,EAC/C,WAAW,GAAG,OAAO,CAAC,GAAG,EAAE,EAC3B,OAAO,GAAG,KAAK,GAChB,GAAG,OAAO,CAAC;IAEZ,IAAA,gBAAO,GAAE,CAAC;IACV,MAAM,KAAK,GAAG,IAAA,iBAAQ,GAAE,CAAC;IAEzB,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;QAChD,CAAC,CAAC,UAAU;QACZ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;IAEvC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,0BAA0B,cAAc,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,mDAAmD;IACnD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IAC/C,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;IACjE,MAAM,aAAa,GAAG;;;;;aAKX,cAAc;;YAEf,KAAK;;;CAGhB,CAAC;IAEA,IAAI,CAAC;QACH,EAAE,CAAC,aAAa,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;QAEhD,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;QACvD,CAAC;QAED,mCAAmC;QACnC,MAAM,GAAG,GAAG,yCAAyC,cAAc,EAAE,CAAC;QACtE,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,EAAE,CAAC,CAAC;QACjC,CAAC;QAED,IAAA,wBAAQ,EAAC,GAAG,EAAE;YACZ,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM;YACnC,OAAO,EAAE,MAAM;YACf,GAAG,EAAE,WAAW;SACjB,CAAC,CAAC;QAEH,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;YAAS,CAAC;QACT,IAAI,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;YAClC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;AACH,CAAC"}
@@ -0,0 +1,26 @@
1
+ /**
2
+ * Database module types
3
+ */
4
+ export declare class DatabaseConfig {
5
+ static readonly MAX_CONNECTIONS = 100;
6
+ static readonly IDLE_TIMEOUT_MILLIS = 30000;
7
+ static readonly CONNECTION_TIMEOUT_MILLIS = 30000;
8
+ static readonly MAX_RETRY_TIME = 20000;
9
+ static readonly DEFAULT_SCHEMA_PATH = "src/storage/database/shared/schema.ts";
10
+ }
11
+ export interface DatabasePoolConfig {
12
+ connectionString?: string;
13
+ maxConnections?: number;
14
+ idleTimeoutMillis?: number;
15
+ connectionTimeoutMillis?: number;
16
+ }
17
+ export interface UpgradeOptions {
18
+ schemaPath?: string;
19
+ projectPath?: string;
20
+ verbose?: boolean;
21
+ }
22
+ export interface GenerateModelsOptions {
23
+ outputPath?: string;
24
+ verbose?: boolean;
25
+ }
26
+ //# sourceMappingURL=models.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../src/database/models.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,qBAAa,cAAc;IACzB,MAAM,CAAC,QAAQ,CAAC,eAAe,OAAO;IACtC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,SAAS;IAC5C,MAAM,CAAC,QAAQ,CAAC,yBAAyB,SAAS;IAClD,MAAM,CAAC,QAAQ,CAAC,cAAc,SAAS;IACvC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,2CAA2C;CAC/E;AAED,MAAM,WAAW,kBAAkB;IACjC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,uBAAuB,CAAC,EAAE,MAAM,CAAC;CAClC;AAED,MAAM,WAAW,cAAc;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,qBAAqB;IACpC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB"}
@@ -0,0 +1,15 @@
1
+ "use strict";
2
+ /**
3
+ * Database module types
4
+ */
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.DatabaseConfig = void 0;
7
+ class DatabaseConfig {
8
+ }
9
+ exports.DatabaseConfig = DatabaseConfig;
10
+ DatabaseConfig.MAX_CONNECTIONS = 100;
11
+ DatabaseConfig.IDLE_TIMEOUT_MILLIS = 30000;
12
+ DatabaseConfig.CONNECTION_TIMEOUT_MILLIS = 30000;
13
+ DatabaseConfig.MAX_RETRY_TIME = 20000;
14
+ DatabaseConfig.DEFAULT_SCHEMA_PATH = 'src/storage/database/shared/schema.ts';
15
+ //# sourceMappingURL=models.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"models.js","sourceRoot":"","sources":["../../src/database/models.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAEH,MAAa,cAAc;;AAA3B,wCAMC;AALiB,8BAAe,GAAG,GAAG,CAAC;AACtB,kCAAmB,GAAG,KAAK,CAAC;AAC5B,wCAAyB,GAAG,KAAK,CAAC;AAClC,6BAAc,GAAG,KAAK,CAAC;AACvB,kCAAmB,GAAG,uCAAuC,CAAC"}
@@ -3,7 +3,7 @@ import { Config } from '../core/config';
3
3
  import { ImageGenerationRequest, ImageGenerationResponse, ImageGenerationResponseHelper } from './models';
4
4
  export declare class ImageGenerationClient extends BaseClient {
5
5
  private model;
6
- constructor(config?: Config, customHeaders?: Record<string, string>, verbose?: boolean);
6
+ constructor(config?: Config, customHeaders?: Record<string, string>);
7
7
  generate(request: ImageGenerationRequest): Promise<ImageGenerationResponse>;
8
8
  batchGenerate(requests: ImageGenerationRequest[]): Promise<ImageGenerationResponse[]>;
9
9
  getResponseHelper(response: ImageGenerationResponse): ImageGenerationResponseHelper;
@@ -1 +1 @@
1
- {"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/image/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAExC,OAAO,EAEL,sBAAsB,EACtB,uBAAuB,EACvB,6BAA6B,EAE9B,MAAM,UAAU,CAAC;AAElB,qBAAa,qBAAsB,SAAQ,UAAU;IACnD,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,CAAC,EAAE,MAAM,EAAE,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO,GAAE,OAAe;IAKvF,QAAQ,CACZ,OAAO,EAAE,sBAAsB,GAC9B,OAAO,CAAC,uBAAuB,CAAC;IA8B7B,aAAa,CACjB,QAAQ,EAAE,sBAAsB,EAAE,GACjC,OAAO,CAAC,uBAAuB,EAAE,CAAC;IAKrC,iBAAiB,CACf,QAAQ,EAAE,uBAAuB,GAChC,6BAA6B;CAGjC"}
1
+ {"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/image/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAExC,OAAO,EAEL,sBAAsB,EACtB,uBAAuB,EACvB,6BAA6B,EAE9B,MAAM,UAAU,CAAC;AAElB,qBAAa,qBAAsB,SAAQ,UAAU;IACnD,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,CAAC,EAAE,MAAM,EAAE,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAK7D,QAAQ,CACZ,OAAO,EAAE,sBAAsB,GAC9B,OAAO,CAAC,uBAAuB,CAAC;IA8B7B,aAAa,CACjB,QAAQ,EAAE,sBAAsB,EAAE,GACjC,OAAO,CAAC,uBAAuB,EAAE,CAAC;IAKrC,iBAAiB,CACf,QAAQ,EAAE,uBAAuB,GAChC,6BAA6B;CAGjC"}
@@ -5,8 +5,8 @@ const client_1 = require("../core/client");
5
5
  const exceptions_1 = require("../core/exceptions");
6
6
  const models_1 = require("./models");
7
7
  class ImageGenerationClient extends client_1.BaseClient {
8
- constructor(config, customHeaders, verbose = false) {
9
- super(config, customHeaders, verbose);
8
+ constructor(config, customHeaders) {
9
+ super(config, customHeaders);
10
10
  this.model = models_1.ImageConfig.DEFAULT_MODEL;
11
11
  }
12
12
  async generate(request) {
@@ -1 +1 @@
1
- {"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/image/client.ts"],"names":[],"mappings":";;;AAAA,2CAA4C;AAE5C,mDAA8C;AAC9C,qCAMkB;AAElB,MAAa,qBAAsB,SAAQ,mBAAU;IAGnD,YAAY,MAAe,EAAE,aAAsC,EAAE,UAAmB,KAAK;QAC3F,KAAK,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK,GAAG,oBAAW,CAAC,aAAa,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,OAA+B;QAE/B,MAAM,UAAU,GAAG,IAAA,uCAA8B,EAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAEvE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAC7B,MAAM,EACN,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,4BAA4B,EAClD,UAAU,CACX,CAAC;QAEF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,qBAAQ,CAChB,uBAAuB,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,eAAe,EAAE,EAC9D,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,EACjD,IAAI,CAAC,KAAK,CACX,CAAC;QACJ,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC7B,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,MAAM,IAAI,qBAAQ,CAChB,4BAA4B,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,eAAe,EAAE,EACnE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,EACjD,IAAI,CAAC,KAAK,CACX,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,QAAkC;QAElC,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QACnE,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAED,iBAAiB,CACf,QAAiC;QAEjC,OAAO,IAAI,sCAA6B,CAAC,QAAQ,CAAC,CAAC;IACrD,CAAC;CACF;AApDD,sDAoDC"}
1
+ {"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/image/client.ts"],"names":[],"mappings":";;;AAAA,2CAA4C;AAE5C,mDAA8C;AAC9C,qCAMkB;AAElB,MAAa,qBAAsB,SAAQ,mBAAU;IAGnD,YAAY,MAAe,EAAE,aAAsC;QACjE,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAC7B,IAAI,CAAC,KAAK,GAAG,oBAAW,CAAC,aAAa,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,OAA+B;QAE/B,MAAM,UAAU,GAAG,IAAA,uCAA8B,EAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAEvE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAC7B,MAAM,EACN,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,4BAA4B,EAClD,UAAU,CACX,CAAC;QAEF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,qBAAQ,CAChB,uBAAuB,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,eAAe,EAAE,EAC9D,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,EACjD,IAAI,CAAC,KAAK,CACX,CAAC;QACJ,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC7B,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,MAAM,IAAI,qBAAQ,CAChB,4BAA4B,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,eAAe,EAAE,EACnE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,EACjD,IAAI,CAAC,KAAK,CACX,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,QAAkC;QAElC,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QACnE,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAED,iBAAiB,CACf,QAAiC;QAEjC,OAAO,IAAI,sCAA6B,CAAC,QAAQ,CAAC,CAAC;IACrD,CAAC;CACF;AApDD,sDAoDC"}
package/dist/index.d.ts CHANGED
@@ -4,5 +4,7 @@ export * from './search';
4
4
  export * from './voice';
5
5
  export * from './video';
6
6
  export * from './llm';
7
+ export * from './database';
8
+ export * from './s3';
7
9
  export { VERSION } from './version';
8
10
  //# sourceMappingURL=index.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,QAAQ,CAAC;AACvB,cAAc,SAAS,CAAC;AACxB,cAAc,UAAU,CAAC;AACzB,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,cAAc,OAAO,CAAC;AACtB,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,QAAQ,CAAC;AACvB,cAAc,SAAS,CAAC;AACxB,cAAc,UAAU,CAAC;AACzB,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,cAAc,OAAO,CAAC;AACtB,cAAc,YAAY,CAAC;AAC3B,cAAc,MAAM,CAAC;AACrB,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC"}
package/dist/index.js CHANGED
@@ -21,6 +21,8 @@ __exportStar(require("./search"), exports);
21
21
  __exportStar(require("./voice"), exports);
22
22
  __exportStar(require("./video"), exports);
23
23
  __exportStar(require("./llm"), exports);
24
+ __exportStar(require("./database"), exports);
25
+ __exportStar(require("./s3"), exports);
24
26
  var version_1 = require("./version");
25
27
  Object.defineProperty(exports, "VERSION", { enumerable: true, get: function () { return version_1.VERSION; } });
26
28
  //# sourceMappingURL=index.js.map
package/dist/index.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,yCAAuB;AACvB,0CAAwB;AACxB,2CAAyB;AACzB,0CAAwB;AACxB,0CAAwB;AACxB,wCAAsB;AACtB,qCAAoC;AAA3B,kGAAA,OAAO,OAAA"}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,yCAAuB;AACvB,0CAAwB;AACxB,2CAAyB;AACzB,0CAAwB;AACxB,0CAAwB;AACxB,wCAAsB;AACtB,6CAA2B;AAC3B,uCAAqB;AACrB,qCAAoC;AAA3B,kGAAA,OAAO,OAAA"}
@@ -0,0 +1,66 @@
1
+ /**
2
+ * S3 Storage client module
3
+ */
4
+ import { Readable } from 'stream';
5
+ import { S3StorageConfig, ListFilesResult } from './models';
6
+ export declare class S3Storage {
7
+ private endpointUrl;
8
+ private accessKey;
9
+ private secretKey;
10
+ private bucketName;
11
+ private region;
12
+ private client;
13
+ constructor(config?: S3StorageConfig);
14
+ private getClient;
15
+ private generateObjectKey;
16
+ private resolveBucket;
17
+ private validateFileName;
18
+ uploadFile(options: {
19
+ fileContent: Buffer;
20
+ fileName: string;
21
+ contentType?: string;
22
+ bucket?: string;
23
+ }): Promise<string>;
24
+ readFile(options: {
25
+ fileKey: string;
26
+ bucket?: string;
27
+ }): Promise<Buffer>;
28
+ deleteFile(options: {
29
+ fileKey: string;
30
+ bucket?: string;
31
+ }): Promise<boolean>;
32
+ fileExists(options: {
33
+ fileKey: string;
34
+ bucket?: string;
35
+ }): Promise<boolean>;
36
+ listFiles(options?: {
37
+ prefix?: string;
38
+ bucket?: string;
39
+ maxKeys?: number;
40
+ continuationToken?: string;
41
+ }): Promise<ListFilesResult>;
42
+ generatePresignedUrl(options: {
43
+ key: string;
44
+ bucket?: string;
45
+ expireTime?: number;
46
+ }): Promise<string>;
47
+ streamUploadFile(options: {
48
+ stream: Readable;
49
+ fileName: string;
50
+ contentType?: string;
51
+ bucket?: string;
52
+ }): Promise<string>;
53
+ uploadFromUrl(options: {
54
+ url: string;
55
+ bucket?: string;
56
+ timeout?: number;
57
+ }): Promise<string>;
58
+ chunkUploadFile(options: {
59
+ chunks: AsyncIterable<Buffer>;
60
+ fileName: string;
61
+ contentType?: string;
62
+ bucket?: string;
63
+ }): Promise<string>;
64
+ }
65
+ export default S3Storage;
66
+ //# sourceMappingURL=client.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/s3/client.ts"],"names":[],"mappings":"AAAA;;GAEG;AAWH,OAAO,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAGlC,OAAO,EAAE,eAAe,EAAY,eAAe,EAAwB,MAAM,UAAU,CAAC;AAM5F,qBAAa,SAAS;IACpB,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAyB;gBAE3B,MAAM,GAAE,eAAoB;IAQxC,OAAO,CAAC,SAAS;IA8BjB,OAAO,CAAC,iBAAiB;IAOzB,OAAO,CAAC,aAAa;IAQrB,OAAO,CAAC,gBAAgB;IAclB,UAAU,CAAC,OAAO,EAAE;QACxB,WAAW,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC;QACjB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,MAAM,CAAC;IAoBb,QAAQ,CAAC,OAAO,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,MAAM,CAAC;IAoBxE,UAAU,CAAC,OAAO,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,CAAC;IAY3E,UAAU,CAAC,OAAO,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,CAAC;IAmB3E,SAAS,CAAC,OAAO,GAAE;QACvB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,iBAAiB,CAAC,EAAE,MAAM,CAAC;KACvB,GAAG,OAAO,CAAC,eAAe,CAAC;IA4B3B,oBAAoB,CAAC,OAAO,EAAE;QAClC,GAAG,EAAE,MAAM,CAAC;QACZ,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,GAAG,OAAO,CAAC,MAAM,CAAC;IAmDb,gBAAgB,CAAC,OAAO,EAAE;QAC9B,MAAM,EAAE,QAAQ,CAAC;QACjB,QAAQ,EAAE,MAAM,CAAC;QACjB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,MAAM,CAAC;IAqBb,aAAa,CAAC,OAAO,EAAE;QAC3B,GAAG,EAAE,MAAM,CAAC;QACZ,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,GAAG,OAAO,CAAC,MAAM,CAAC;IAwBb,eAAe,CAAC,OAAO,EAAE;QAC7B,MAAM,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;QAC9B,QAAQ,EAAE,MAAM,CAAC;QACjB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,MAAM,CAAC;CAsBpB;AAED,eAAe,SAAS,CAAC"}