@adobe/genstudio-uix-sdk
Version:
GenStudio UI Extensibility SDK
70 lines (69 loc) • 3 kB
JavaScript
/*
Copyright 2025 Adobe. All rights reserved.
This file is licensed to you under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License. You may obtain a copy
of the License at http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software distributed under
the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
OF ANY KIND, either express or implied. See the License for the specific language
governing permissions and limitations under the License.
*/
import { AdditionalContextTypes } from "./GenerationContext";
export class GenerationContextError extends Error {
constructor(message) {
super(message);
this.name = "GenerationContextError";
}
}
export class GenerationContextService {
/**
* Sets additional context on the prompt
* @param connection - The guest connection to the host
* @param additionalContext - The additional context object
* @returns void
* @throws Error if connection is missing
*/
static async setAdditionalContext(connection, additionalContext) {
const { extensionId, additionalContextType, additionalContextValues } = additionalContext;
const validations = [
[!connection, "Connection is required to set additional context"],
[!extensionId, "Invalid extension ID"],
[!additionalContextType, "Context type is required"],
[!Object.values(AdditionalContextTypes).includes(additionalContextType), "Invalid context type"],
[
!additionalContextValues.every(value => value.id && value.description),
"Invalid context value format"
]
];
const failedValidation = validations.find(([condition]) => condition);
if (failedValidation) {
throw new GenerationContextError(failedValidation[1]);
}
try {
// @ts-ignore Remote API is handled through postMessage
await connection.host.api.create.updateAdditionalContext(additionalContext);
}
catch (error) {
throw new GenerationContextError("Failed to set additional context");
}
}
/**
* Gets the generation context
* @param connection - The guest connection to the host
* @returns The generation context
* @throws Error if connection is missing
*/
static async getGenerationContext(connection) {
if (!connection) {
throw new GenerationContextError("Connection is required to get generation context");
}
try {
// @ts-ignore Remote API is handled through postMessage
return await connection.host.api.create.getGenerationContext();
}
catch (error) {
throw new GenerationContextError("Failed to get generation context");
}
}
}
//# sourceMappingURL=GenerationContextService.js.map