@skmtc/core 0.0.502 → 0.0.504
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/esm/dsl/operation/OperationInsertable.d.ts +2 -2
- package/esm/dsl/operation/OperationInsertable.d.ts.map +1 -1
- package/esm/dsl/operation/toOperationInsertable.d.ts +2 -8
- package/esm/dsl/operation/toOperationInsertable.d.ts.map +1 -1
- package/esm/run/handleEnrichment.d.ts +1 -1
- package/esm/run/handleEnrichment.d.ts.map +1 -1
- package/esm/run/handleEnrichment.js +4 -2
- package/package.json +1 -1
- package/script/dsl/operation/OperationInsertable.d.ts +2 -2
- package/script/dsl/operation/OperationInsertable.d.ts.map +1 -1
- package/script/dsl/operation/toOperationInsertable.d.ts +2 -8
- package/script/dsl/operation/toOperationInsertable.d.ts.map +1 -1
- package/script/run/handleEnrichment.d.ts +1 -1
- package/script/run/handleEnrichment.d.ts.map +1 -1
- package/script/run/handleEnrichment.js +4 -2
|
@@ -4,7 +4,7 @@ import type { GenerateContext } from '../../context/GenerateContext.js';
|
|
|
4
4
|
import type { Identifier } from '../Identifier.js';
|
|
5
5
|
import type { EnrichmentRequest } from '../../types/EnrichmentRequest.js';
|
|
6
6
|
import type { z } from 'zod';
|
|
7
|
-
type
|
|
7
|
+
export type OperationInsertableArgs<EnrichmentType> = {
|
|
8
8
|
context: GenerateContext;
|
|
9
9
|
settings: ContentSettings<EnrichmentType>;
|
|
10
10
|
operation: OasOperation;
|
|
@@ -41,7 +41,7 @@ export type OperationGateway<EnrichmentType> = {
|
|
|
41
41
|
export type OperationInsertable<V, EnrichmentType> = {
|
|
42
42
|
prototype: V;
|
|
43
43
|
} & {
|
|
44
|
-
new ({ context, settings, operation }:
|
|
44
|
+
new ({ context, settings, operation }: OperationInsertableArgs<EnrichmentType>): V;
|
|
45
45
|
id: string;
|
|
46
46
|
type: 'operation';
|
|
47
47
|
_class: 'OperationInsertable';
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"OperationInsertable.d.ts","sourceRoot":"","sources":["../../../src/dsl/operation/OperationInsertable.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAA;AACpE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAA;AAC5D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAA;AACvE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAA;AAClD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAA;AACzE,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAE5B,
|
|
1
|
+
{"version":3,"file":"OperationInsertable.d.ts","sourceRoot":"","sources":["../../../src/dsl/operation/OperationInsertable.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAA;AACpE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAA;AAC5D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAA;AACvE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAA;AAClD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAA;AACzE,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAE5B,MAAM,MAAM,uBAAuB,CAAC,cAAc,IAAI;IACpD,OAAO,EAAE,eAAe,CAAA;IACxB,QAAQ,EAAE,eAAe,CAAC,cAAc,CAAC,CAAA;IACzC,SAAS,EAAE,YAAY,CAAA;CACxB,CAAA;AAED,MAAM,MAAM,sBAAsB,GAAG;IACnC,kBAAkB,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,IAAI,CAAA;CACtD,CAAA;AAED,MAAM,MAAM,6BAA6B,CAAC,cAAc,IAAI;IAC1D,OAAO,EAAE,eAAe,CAAA;IACxB,QAAQ,EAAE,eAAe,CAAC,cAAc,CAAC,CAAA;CAC1C,CAAA;AAED,MAAM,MAAM,wBAAwB,CAAC,cAAc,IAAI;IACrD,SAAS,EAAE,YAAY,CAAA;IACvB,WAAW,EAAE,cAAc,CAAA;IAC3B,OAAO,EAAE,eAAe,CAAA;CACzB,CAAA;AAED,KAAK,iBAAiB,GAAG;IACvB,SAAS,EAAE,YAAY,CAAA;IACvB,OAAO,EAAE,eAAe,CAAA;CACzB,CAAA;AAED,MAAM,MAAM,gBAAgB,CAAC,cAAc,IAAI;IAC7C,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,6BAA6B,CAAC,cAAc,CAAC,GAAG,sBAAsB,CAAA;IAClG,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,EAAE,WAAW,CAAA;IACjB,MAAM,EAAE,kBAAkB,CAAA;IAE1B,YAAY,EAAE,MAAM,UAAU,CAAA;IAC9B,YAAY,EAAE,MAAM,MAAM,CAAA;IAC1B,mBAAmB,CAAC,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,iBAAiB,CAAC,cAAc,CAAC,GAAG,SAAS,CAAA;IAChG,kBAAkB,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;IACnD,aAAa,EAAE,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,iBAAiB,KAAK,cAAc,CAAA;IAC5E,WAAW,EAAE,CAAC,EACZ,SAAS,EACT,WAAW,EACX,OAAO,EACR,EAAE,wBAAwB,CAAC,cAAc,CAAC,KAAK,OAAO,CAAA;IAEvD,QAAQ,EAAE,OAAO,CAAA;CAClB,CAAA;AAED,MAAM,MAAM,mBAAmB,CAAC,CAAC,EAAE,cAAc,IAAI;IAAE,SAAS,EAAE,CAAC,CAAA;CAAE,GAAG;IACtE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,uBAAuB,CAAC,cAAc,CAAC,GAAG,CAAC,CAAA;IAClF,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,EAAE,WAAW,CAAA;IACjB,MAAM,EAAE,qBAAqB,CAAA;IAC7B,YAAY,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,UAAU,CAAA;IACrD,YAAY,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,MAAM,CAAA;IACjD,mBAAmB,CAAC,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,iBAAiB,CAAC,cAAc,CAAC,GAAG,SAAS,CAAA;IAChG,kBAAkB,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;IACnD,aAAa,EAAE,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,iBAAiB,KAAK,cAAc,CAAA;IAC5E,WAAW,EAAE,CAAC,EACZ,SAAS,EACT,WAAW,EACX,OAAO,EACR,EAAE,wBAAwB,CAAC,cAAc,CAAC,KAAK,OAAO,CAAA;IAEvD,QAAQ,EAAE,OAAO,CAAA;CAGlB,GAAG,QAAQ,CAAA"}
|
|
@@ -1,15 +1,9 @@
|
|
|
1
1
|
import type { GenerateContext } from '../../context/GenerateContext.js';
|
|
2
2
|
import type { OasOperation } from '../../oas/operation/Operation.js';
|
|
3
|
-
import type { ContentSettings } from '../ContentSettings.js';
|
|
4
3
|
import type { Identifier } from '../Identifier.js';
|
|
5
4
|
import type { EnrichmentRequest } from '../../types/EnrichmentRequest.js';
|
|
6
|
-
import type { IsSupportedOperationArgs } from './OperationInsertable.js';
|
|
5
|
+
import type { IsSupportedOperationArgs, OperationInsertableArgs } from './OperationInsertable.js';
|
|
7
6
|
import type { z } from 'zod';
|
|
8
|
-
export type OperationInsertableArgs<EnrichmentType> = {
|
|
9
|
-
context: GenerateContext;
|
|
10
|
-
settings: ContentSettings<EnrichmentType>;
|
|
11
|
-
operation: OasOperation;
|
|
12
|
-
};
|
|
13
7
|
export type ToOperationInsertableArgs<EnrichmentType> = {
|
|
14
8
|
id: string;
|
|
15
9
|
toIdentifier: (operation: OasOperation) => Identifier;
|
|
@@ -25,7 +19,7 @@ type ToEnrichmentsArgs = {
|
|
|
25
19
|
};
|
|
26
20
|
export declare const toOperationInsertable: <EnrichmentType>(config: ToOperationInsertableArgs<EnrichmentType>) => {
|
|
27
21
|
new (args: OperationInsertableArgs<EnrichmentType>): {
|
|
28
|
-
settings: ContentSettings<EnrichmentType>;
|
|
22
|
+
settings: import("../ContentSettings.js").ContentSettings<EnrichmentType>;
|
|
29
23
|
operation: OasOperation;
|
|
30
24
|
generatorKey: import("../../types/GeneratorKeys.js").GeneratorKey;
|
|
31
25
|
insertOperation<V extends import("../../mod.js").GeneratedValue, ET>(insertable: import("./OperationInsertable.js").OperationInsertable<V, ET>, operation: OasOperation): import("../Inserted.js").Inserted<V, "force", ET>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"toOperationInsertable.d.ts","sourceRoot":"","sources":["../../../src/dsl/operation/toOperationInsertable.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAA;AACvE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAA;
|
|
1
|
+
{"version":3,"file":"toOperationInsertable.d.ts","sourceRoot":"","sources":["../../../src/dsl/operation/toOperationInsertable.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAA;AACvE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAA;AAEpE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAA;AAClD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAA;AACzE,OAAO,KAAK,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,MAAM,0BAA0B,CAAA;AACjG,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAE5B,MAAM,MAAM,yBAAyB,CAAC,cAAc,IAAI;IACtD,EAAE,EAAE,MAAM,CAAA;IACV,YAAY,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,UAAU,CAAA;IACrD,YAAY,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,MAAM,CAAA;IACjD,WAAW,CAAC,EAAE,CAAC,EACb,SAAS,EACT,WAAW,EACX,OAAO,EACR,EAAE,wBAAwB,CAAC,cAAc,CAAC,KAAK,OAAO,CAAA;IACvD,mBAAmB,CAAC,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,iBAAiB,CAAC,cAAc,CAAC,GAAG,SAAS,CAAA;IAChG,kBAAkB,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;IACnD,QAAQ,CAAC,EAAE,OAAO,CAAA;CACnB,CAAA;AAED,KAAK,iBAAiB,GAAG;IACvB,SAAS,EAAE,YAAY,CAAA;IACvB,OAAO,EAAE,eAAe,CAAA;CACzB,CAAA;AAED,eAAO,MAAM,qBAAqB,GAAI,cAAc,UAC1C,yBAAyB,CAAC,cAAc,CAAC;eA4B7B,uBAAuB,CAAC,cAAc,CAAC;;;;;;;;;;;;;;;;;;8BA9CjC,YAAY,KAAK,UAAU;8BAC3B,YAAY,KAAK,MAAM;sCAMf,YAAY,KAAK,iBAAiB,gBAAgB,GAAG,SAAS;8BACtE,CAAC,CAAC,OAAO,gBAAgB;4CAsBD,iBAAiB,KAAG,cAAc;oGAxBpC,OAAO;;CAoDxD,CAAA"}
|
|
@@ -1,3 +1,3 @@
|
|
|
1
1
|
import type { EnrichmentRequest } from '../types/EnrichmentRequest.js';
|
|
2
|
-
export declare const handleEnrichment: <EnrichmentType>({ prompt, content, responseSchema }: EnrichmentRequest<EnrichmentType>) => Promise<string>;
|
|
2
|
+
export declare const handleEnrichment: <EnrichmentType>({ prompt, content, responseSchema: zodResponseSchema }: EnrichmentRequest<EnrichmentType>) => Promise<string>;
|
|
3
3
|
//# sourceMappingURL=handleEnrichment.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"handleEnrichment.d.ts","sourceRoot":"","sources":["../../src/run/handleEnrichment.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAA;AAetE,eAAO,MAAM,gBAAgB,GAAU,cAAc,
|
|
1
|
+
{"version":3,"file":"handleEnrichment.d.ts","sourceRoot":"","sources":["../../src/run/handleEnrichment.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAA;AAetE,eAAO,MAAM,gBAAgB,GAAU,cAAc,0DAIlD,iBAAiB,CAAC,cAAc,CAAC,oBAuCnC,CAAA"}
|
|
@@ -9,7 +9,9 @@ const genAI = new GoogleGenerativeAI(apiKey);
|
|
|
9
9
|
const model = genAI.getGenerativeModel({
|
|
10
10
|
model: 'gemini-1.5-flash'
|
|
11
11
|
});
|
|
12
|
-
export const handleEnrichment = async ({ prompt, content, responseSchema }) => {
|
|
12
|
+
export const handleEnrichment = async ({ prompt, content, responseSchema: zodResponseSchema }) => {
|
|
13
|
+
const responseSchema = zodToJsonSchema(zodResponseSchema);
|
|
14
|
+
console.log('RESPONSE SCHEMA', responseSchema);
|
|
13
15
|
const chatSession = model.startChat({
|
|
14
16
|
generationConfig: {
|
|
15
17
|
temperature: 1,
|
|
@@ -17,7 +19,7 @@ export const handleEnrichment = async ({ prompt, content, responseSchema }) => {
|
|
|
17
19
|
topK: 40,
|
|
18
20
|
maxOutputTokens: 8192,
|
|
19
21
|
responseMimeType: 'application/json',
|
|
20
|
-
responseSchema
|
|
22
|
+
responseSchema
|
|
21
23
|
},
|
|
22
24
|
history: [
|
|
23
25
|
{
|
package/package.json
CHANGED
|
@@ -4,7 +4,7 @@ import type { GenerateContext } from '../../context/GenerateContext.js';
|
|
|
4
4
|
import type { Identifier } from '../Identifier.js';
|
|
5
5
|
import type { EnrichmentRequest } from '../../types/EnrichmentRequest.js';
|
|
6
6
|
import type { z } from 'zod';
|
|
7
|
-
type
|
|
7
|
+
export type OperationInsertableArgs<EnrichmentType> = {
|
|
8
8
|
context: GenerateContext;
|
|
9
9
|
settings: ContentSettings<EnrichmentType>;
|
|
10
10
|
operation: OasOperation;
|
|
@@ -41,7 +41,7 @@ export type OperationGateway<EnrichmentType> = {
|
|
|
41
41
|
export type OperationInsertable<V, EnrichmentType> = {
|
|
42
42
|
prototype: V;
|
|
43
43
|
} & {
|
|
44
|
-
new ({ context, settings, operation }:
|
|
44
|
+
new ({ context, settings, operation }: OperationInsertableArgs<EnrichmentType>): V;
|
|
45
45
|
id: string;
|
|
46
46
|
type: 'operation';
|
|
47
47
|
_class: 'OperationInsertable';
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"OperationInsertable.d.ts","sourceRoot":"","sources":["../../../src/dsl/operation/OperationInsertable.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAA;AACpE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAA;AAC5D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAA;AACvE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAA;AAClD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAA;AACzE,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAE5B,
|
|
1
|
+
{"version":3,"file":"OperationInsertable.d.ts","sourceRoot":"","sources":["../../../src/dsl/operation/OperationInsertable.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAA;AACpE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAA;AAC5D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAA;AACvE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAA;AAClD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAA;AACzE,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAE5B,MAAM,MAAM,uBAAuB,CAAC,cAAc,IAAI;IACpD,OAAO,EAAE,eAAe,CAAA;IACxB,QAAQ,EAAE,eAAe,CAAC,cAAc,CAAC,CAAA;IACzC,SAAS,EAAE,YAAY,CAAA;CACxB,CAAA;AAED,MAAM,MAAM,sBAAsB,GAAG;IACnC,kBAAkB,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,IAAI,CAAA;CACtD,CAAA;AAED,MAAM,MAAM,6BAA6B,CAAC,cAAc,IAAI;IAC1D,OAAO,EAAE,eAAe,CAAA;IACxB,QAAQ,EAAE,eAAe,CAAC,cAAc,CAAC,CAAA;CAC1C,CAAA;AAED,MAAM,MAAM,wBAAwB,CAAC,cAAc,IAAI;IACrD,SAAS,EAAE,YAAY,CAAA;IACvB,WAAW,EAAE,cAAc,CAAA;IAC3B,OAAO,EAAE,eAAe,CAAA;CACzB,CAAA;AAED,KAAK,iBAAiB,GAAG;IACvB,SAAS,EAAE,YAAY,CAAA;IACvB,OAAO,EAAE,eAAe,CAAA;CACzB,CAAA;AAED,MAAM,MAAM,gBAAgB,CAAC,cAAc,IAAI;IAC7C,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,6BAA6B,CAAC,cAAc,CAAC,GAAG,sBAAsB,CAAA;IAClG,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,EAAE,WAAW,CAAA;IACjB,MAAM,EAAE,kBAAkB,CAAA;IAE1B,YAAY,EAAE,MAAM,UAAU,CAAA;IAC9B,YAAY,EAAE,MAAM,MAAM,CAAA;IAC1B,mBAAmB,CAAC,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,iBAAiB,CAAC,cAAc,CAAC,GAAG,SAAS,CAAA;IAChG,kBAAkB,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;IACnD,aAAa,EAAE,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,iBAAiB,KAAK,cAAc,CAAA;IAC5E,WAAW,EAAE,CAAC,EACZ,SAAS,EACT,WAAW,EACX,OAAO,EACR,EAAE,wBAAwB,CAAC,cAAc,CAAC,KAAK,OAAO,CAAA;IAEvD,QAAQ,EAAE,OAAO,CAAA;CAClB,CAAA;AAED,MAAM,MAAM,mBAAmB,CAAC,CAAC,EAAE,cAAc,IAAI;IAAE,SAAS,EAAE,CAAC,CAAA;CAAE,GAAG;IACtE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,uBAAuB,CAAC,cAAc,CAAC,GAAG,CAAC,CAAA;IAClF,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,EAAE,WAAW,CAAA;IACjB,MAAM,EAAE,qBAAqB,CAAA;IAC7B,YAAY,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,UAAU,CAAA;IACrD,YAAY,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,MAAM,CAAA;IACjD,mBAAmB,CAAC,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,iBAAiB,CAAC,cAAc,CAAC,GAAG,SAAS,CAAA;IAChG,kBAAkB,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;IACnD,aAAa,EAAE,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,iBAAiB,KAAK,cAAc,CAAA;IAC5E,WAAW,EAAE,CAAC,EACZ,SAAS,EACT,WAAW,EACX,OAAO,EACR,EAAE,wBAAwB,CAAC,cAAc,CAAC,KAAK,OAAO,CAAA;IAEvD,QAAQ,EAAE,OAAO,CAAA;CAGlB,GAAG,QAAQ,CAAA"}
|
|
@@ -1,15 +1,9 @@
|
|
|
1
1
|
import type { GenerateContext } from '../../context/GenerateContext.js';
|
|
2
2
|
import type { OasOperation } from '../../oas/operation/Operation.js';
|
|
3
|
-
import type { ContentSettings } from '../ContentSettings.js';
|
|
4
3
|
import type { Identifier } from '../Identifier.js';
|
|
5
4
|
import type { EnrichmentRequest } from '../../types/EnrichmentRequest.js';
|
|
6
|
-
import type { IsSupportedOperationArgs } from './OperationInsertable.js';
|
|
5
|
+
import type { IsSupportedOperationArgs, OperationInsertableArgs } from './OperationInsertable.js';
|
|
7
6
|
import type { z } from 'zod';
|
|
8
|
-
export type OperationInsertableArgs<EnrichmentType> = {
|
|
9
|
-
context: GenerateContext;
|
|
10
|
-
settings: ContentSettings<EnrichmentType>;
|
|
11
|
-
operation: OasOperation;
|
|
12
|
-
};
|
|
13
7
|
export type ToOperationInsertableArgs<EnrichmentType> = {
|
|
14
8
|
id: string;
|
|
15
9
|
toIdentifier: (operation: OasOperation) => Identifier;
|
|
@@ -25,7 +19,7 @@ type ToEnrichmentsArgs = {
|
|
|
25
19
|
};
|
|
26
20
|
export declare const toOperationInsertable: <EnrichmentType>(config: ToOperationInsertableArgs<EnrichmentType>) => {
|
|
27
21
|
new (args: OperationInsertableArgs<EnrichmentType>): {
|
|
28
|
-
settings: ContentSettings<EnrichmentType>;
|
|
22
|
+
settings: import("../ContentSettings.js").ContentSettings<EnrichmentType>;
|
|
29
23
|
operation: OasOperation;
|
|
30
24
|
generatorKey: import("../../types/GeneratorKeys.js").GeneratorKey;
|
|
31
25
|
insertOperation<V extends import("../../mod.js").GeneratedValue, ET>(insertable: import("./OperationInsertable.js").OperationInsertable<V, ET>, operation: OasOperation): import("../Inserted.js").Inserted<V, "force", ET>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"toOperationInsertable.d.ts","sourceRoot":"","sources":["../../../src/dsl/operation/toOperationInsertable.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAA;AACvE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAA;
|
|
1
|
+
{"version":3,"file":"toOperationInsertable.d.ts","sourceRoot":"","sources":["../../../src/dsl/operation/toOperationInsertable.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAA;AACvE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAA;AAEpE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAA;AAClD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAA;AACzE,OAAO,KAAK,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,MAAM,0BAA0B,CAAA;AACjG,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAE5B,MAAM,MAAM,yBAAyB,CAAC,cAAc,IAAI;IACtD,EAAE,EAAE,MAAM,CAAA;IACV,YAAY,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,UAAU,CAAA;IACrD,YAAY,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,MAAM,CAAA;IACjD,WAAW,CAAC,EAAE,CAAC,EACb,SAAS,EACT,WAAW,EACX,OAAO,EACR,EAAE,wBAAwB,CAAC,cAAc,CAAC,KAAK,OAAO,CAAA;IACvD,mBAAmB,CAAC,EAAE,CAAC,SAAS,EAAE,YAAY,KAAK,iBAAiB,CAAC,cAAc,CAAC,GAAG,SAAS,CAAA;IAChG,kBAAkB,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;IACnD,QAAQ,CAAC,EAAE,OAAO,CAAA;CACnB,CAAA;AAED,KAAK,iBAAiB,GAAG;IACvB,SAAS,EAAE,YAAY,CAAA;IACvB,OAAO,EAAE,eAAe,CAAA;CACzB,CAAA;AAED,eAAO,MAAM,qBAAqB,GAAI,cAAc,UAC1C,yBAAyB,CAAC,cAAc,CAAC;eA4B7B,uBAAuB,CAAC,cAAc,CAAC;;;;;;;;;;;;;;;;;;8BA9CjC,YAAY,KAAK,UAAU;8BAC3B,YAAY,KAAK,MAAM;sCAMf,YAAY,KAAK,iBAAiB,gBAAgB,GAAG,SAAS;8BACtE,CAAC,CAAC,OAAO,gBAAgB;4CAsBD,iBAAiB,KAAG,cAAc;oGAxBpC,OAAO;;CAoDxD,CAAA"}
|
|
@@ -1,3 +1,3 @@
|
|
|
1
1
|
import type { EnrichmentRequest } from '../types/EnrichmentRequest.js';
|
|
2
|
-
export declare const handleEnrichment: <EnrichmentType>({ prompt, content, responseSchema }: EnrichmentRequest<EnrichmentType>) => Promise<string>;
|
|
2
|
+
export declare const handleEnrichment: <EnrichmentType>({ prompt, content, responseSchema: zodResponseSchema }: EnrichmentRequest<EnrichmentType>) => Promise<string>;
|
|
3
3
|
//# sourceMappingURL=handleEnrichment.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"handleEnrichment.d.ts","sourceRoot":"","sources":["../../src/run/handleEnrichment.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAA;AAetE,eAAO,MAAM,gBAAgB,GAAU,cAAc,
|
|
1
|
+
{"version":3,"file":"handleEnrichment.d.ts","sourceRoot":"","sources":["../../src/run/handleEnrichment.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAA;AAetE,eAAO,MAAM,gBAAgB,GAAU,cAAc,0DAIlD,iBAAiB,CAAC,cAAc,CAAC,oBAuCnC,CAAA"}
|
|
@@ -35,7 +35,9 @@ const genAI = new generative_ai_1.GoogleGenerativeAI(apiKey);
|
|
|
35
35
|
const model = genAI.getGenerativeModel({
|
|
36
36
|
model: 'gemini-1.5-flash'
|
|
37
37
|
});
|
|
38
|
-
const handleEnrichment = async ({ prompt, content, responseSchema }) => {
|
|
38
|
+
const handleEnrichment = async ({ prompt, content, responseSchema: zodResponseSchema }) => {
|
|
39
|
+
const responseSchema = (0, zod_to_json_schema_1.zodToJsonSchema)(zodResponseSchema);
|
|
40
|
+
console.log('RESPONSE SCHEMA', responseSchema);
|
|
39
41
|
const chatSession = model.startChat({
|
|
40
42
|
generationConfig: {
|
|
41
43
|
temperature: 1,
|
|
@@ -43,7 +45,7 @@ const handleEnrichment = async ({ prompt, content, responseSchema }) => {
|
|
|
43
45
|
topK: 40,
|
|
44
46
|
maxOutputTokens: 8192,
|
|
45
47
|
responseMimeType: 'application/json',
|
|
46
|
-
responseSchema
|
|
48
|
+
responseSchema
|
|
47
49
|
},
|
|
48
50
|
history: [
|
|
49
51
|
{
|