Function pythonPromptFunction

src/prompts/processors/python.ts:20–41

(
  filePath: string,
  functionName: string,
  context: {
    vars: Record<string, VarValue>;
    provider?: ApiProvider;
    config?: unknown;
  },
)

 * @returns The prompts
 */
export const pythonPromptFunction = async (
  filePath: string,
  functionName: string,
  context: {
    vars: Record<string, VarValue>;
    provider?: ApiProvider;
    config?: unknown;
  },
) => {
  invariant(context.provider?.id, 'provider.id is required');
  const transformedContext: PromptFunctionContext = {
    vars: context.vars,
    provider: {
      id:
        typeof context.provider?.id === 'function' ? context.provider?.id() : context.provider?.id,
      label: context.provider?.label,
    },
    config: context.config ?? {},
  };

  return runPython(filePath, functionName, [transformedContext]);
};

/**
 * Legacy Python prompt function. Runs the whole python file.