@genkit-ai/dotprompt/lib/prompt.mjs

Genkit AI framework `.prompt` file format and management library.

import {
  __async,
  __objRest,
  __spreadProps,
  __spreadValues
} from "./chunk-DFMI36TU.mjs";
import {
  definePrompt,
  generate,
  generateStream,
  toGenerateRequest
} from "@genkit-ai/ai";
import { GenkitError } from "@genkit-ai/core";
import { parseSchema } from "@genkit-ai/core/schema";
import { createHash } from "crypto";
import fm from "front-matter";
import z from "zod";
import {
  toFrontmatter,
  toMetadata
} from "./metadata.js";
import { lookupPrompt, registryDefinitionKey } from "./registry.js";
import { compile } from "./template.js";
class Dotprompt {
  static parse(name, source) {
    try {
      const fmResult = fm(source.trimStart(), {
        allowUnsafe: false
      });
      return new Dotprompt(
        __spreadProps(__spreadValues({}, toMetadata(fmResult.attributes)), { name }),
        fmResult.body
      );
    } catch (e) {
      throw new GenkitError({
        source: "Dotprompt",
        status: "INVALID_ARGUMENT",
        message: `Error parsing YAML frontmatter of '${name}' prompt: ${e.stack}`
      });
    }
  }
  static fromAction(action) {
    var _b, _c, _d, _e;
    const _a = action.__action.metadata.prompt, { template } = _a, options = __objRest(_a, ["template"]);
    const pm = options;
    if ((_b = pm.input) == null ? void 0 : _b.schema) {
      pm.input.jsonSchema = (_c = options.input) == null ? void 0 : _c.schema;
      delete pm.input.schema;
    }
    if ((_d = pm.output) == null ? void 0 : _d.schema) {
      pm.output.jsonSchema = (_e = options.output) == null ? void 0 : _e.schema;
    }
    const prompt = new Dotprompt(options, template);
    return prompt;
  }
  constructor(options, template) {
    this.name = options.name || "untitledPrompt";
    this.variant = options.variant;
    this.model = options.model;
    this.input = options.input || { schema: z.any() };
    this.output = options.output;
    this.tools = options.tools;
    this.config = options.config;
    this.candidates = options.candidates;
    this.template = template;
    this.hash = createHash("sha256").update(JSON.stringify(this)).digest("hex");
    this._render = compile(this.template, options);
  }
  renderText(input, options) {
    const result = this.renderMessages(input, options);
    if (result.length !== 1) {
      throw new Error("Multi-message prompt can't be rendered as text.");
    }
    let out = "";
    for (const part of result[0].content) {
      if (!part.text) {
        throw new Error("Multimodal prompt can't be rendered as text.");
      }
      out += part.text;
    }
    return out;
  }
  renderMessages(input, options) {
    var _a, _b, _c;
    input = parseSchema(input, {
      schema: (_a = this.input) == null ? void 0 : _a.schema,
      jsonSchema: (_b = this.input) == null ? void 0 : _b.jsonSchema
    });
    return this._render(__spreadValues(__spreadValues({}, (_c = this.input) == null ? void 0 : _c.default), input), options);
  }
  toJSON() {
    return __spreadProps(__spreadValues({}, toFrontmatter(this)), { template: this.template });
  }
  define(options) {
    var _a, _b;
    definePrompt(
      {
        name: registryDefinitionKey(this.name, this.variant, options == null ? void 0 : options.ns),
        description: "Defined by Dotprompt",
        inputSchema: (_a = this.input) == null ? void 0 : _a.schema,
        inputJsonSchema: (_b = this.input) == null ? void 0 : _b.jsonSchema,
        metadata: {
          type: "prompt",
          prompt: this.toJSON()
        }
      },
      (input) => __async(this, null, function* () {
        return toGenerateRequest(this.render({ input }));
      })
    );
  }
  _generateOptions(options) {
    var _a, _b, _c, _d, _e, _f;
    const messages = this.renderMessages(options.input, {
      history: options.history,
      context: options.context
    });
    return {
      model: options.model || this.model,
      config: __spreadValues(__spreadValues({}, this.config), options.config),
      history: messages.slice(0, messages.length - 1),
      prompt: messages[messages.length - 1].content,
      context: options.context,
      candidates: options.candidates || this.candidates || 1,
      output: {
        format: ((_a = options.output) == null ? void 0 : _a.format) || ((_b = this.output) == null ? void 0 : _b.format) || void 0,
        schema: ((_c = options.output) == null ? void 0 : _c.schema) || ((_d = this.output) == null ? void 0 : _d.schema),
        jsonSchema: ((_e = options.output) == null ? void 0 : _e.jsonSchema) || ((_f = this.output) == null ? void 0 : _f.jsonSchema)
      },
      tools: (options.tools || []).concat(this.tools || []),
      streamingCallback: options.streamingCallback,
      returnToolRequests: options.returnToolRequests
    };
  }
  render(opt) {
    return this._generateOptions(opt);
  }
  generate(opt) {
    return __async(this, null, function* () {
      return generate(this.render(opt));
    });
  }
  generateStream(opt) {
    return __async(this, null, function* () {
      return generateStream(this.render(opt));
    });
  }
}
class DotpromptRef {
  constructor(name, options) {
    this.name = name;
    this.variant = options == null ? void 0 : options.variant;
    this.dir = options == null ? void 0 : options.dir;
  }
  loadPrompt() {
    return __async(this, null, function* () {
      if (this._prompt)
        return this._prompt;
      this._prompt = yield lookupPrompt(
        this.name,
        this.variant,
        this.dir
      );
      return this._prompt;
    });
  }
  generate(opt) {
    return __async(this, null, function* () {
      const prompt = yield this.loadPrompt();
      return prompt.generate(opt);
    });
  }
  render(opt) {
    return __async(this, null, function* () {
      const prompt = yield this.loadPrompt();
      return prompt.render(opt);
    });
  }
}
function defineDotprompt(options, template) {
  const prompt = new Dotprompt(options, template);
  prompt.define();
  return prompt;
}
export {
  Dotprompt,
  DotpromptRef,
  defineDotprompt
};
//# sourceMappingURL=prompt.mjs.map