@hashgraphonline/conversational-agent
Version:
Hashgraph Online conversational AI agent implementing HCS-10 communication, HCS-2 registries, and content inscription on Hedera. https://hol.org
101 lines (100 loc) • 3.62 kB
JavaScript
import { BaseHederaQueryTool } from "hedera-agent-kit";
import { z } from "zod";
import crypto from "crypto";
import { getUploadPostageBatchId, hexToBytes, errorHasStatus, getErrorMessage, getResponseWithStructuredContent } from "./index62.js";
import { BAD_REQUEST_STATUS } from "./index63.js";
import { Wallet } from "@ethereumjs/wallet";
const UpdateFeedSchema = z.object({
data: z.string(),
memoryTopic: z.string(),
postageBatchId: z.string().optional()
});
class UpdateFeedTool extends BaseHederaQueryTool {
constructor(params) {
const { bee, config, ...rest } = params;
super(rest);
this.name = "swarm-update-feed";
this.description = `Update the feed of a given topic with new data.
data: Arbitrary string to upload.
memoryTopic: If provided, uploads the data to a feed with this topic. It is the label of the memory that can be used later to retrieve the data instead of its content hash. If not a hex string, it will be hashed to create a feed topic.
postageBatchId: The postage stamp batch ID which will be used to perform the upload, if it is provided.`;
this.namespace = "swarm";
this.specificInputSchema = UpdateFeedSchema;
this.bee = bee;
this.config = config;
}
async executeQuery(input) {
const { data, memoryTopic, postageBatchId: inputPostageBatchId } = input;
if (!data) {
this.logger.error(
"Missing required parameter: data."
);
throw new Error("Missing required parameter: data.");
} else if (!memoryTopic) {
this.logger.error(
"Missing required parameter: topic."
);
throw new Error("Missing required parameter: topic.");
}
let postageBatchId = "";
try {
postageBatchId = await getUploadPostageBatchId(
inputPostageBatchId,
this.bee,
this.config
);
} catch (error) {
let errorMessage = "Update feed failed.";
if (error instanceof Error) {
errorMessage = error.message;
}
this.logger.error(errorMessage);
throw new Error(errorMessage);
}
const binaryData = Buffer.from(data);
if (!this.config.beeFeedPK) {
this.logger.error("Feed private key not configured.");
throw new Error("Feed private key not configured.");
}
let topic = memoryTopic;
if (topic.startsWith("0x")) {
topic = topic.slice(2);
}
const isHexString = /^[0-9a-fA-F]{64}$/.test(memoryTopic);
if (!isHexString) {
const hash = crypto.createHash("sha256").update(memoryTopic).digest("hex");
topic = hash;
}
const topicBytes = hexToBytes(topic);
const feedPrivateKey = hexToBytes(this.config.beeFeedPK);
const signer = new Wallet(feedPrivateKey);
const owner = signer.getAddressString().slice(2);
let result;
try {
const feedWriter = this.bee.makeFeedWriter(topicBytes, feedPrivateKey);
result = await feedWriter.uploadPayload(postageBatchId, binaryData);
} catch (error) {
let errorMessage = "Unable to update feed.";
if (errorHasStatus(error, BAD_REQUEST_STATUS)) {
errorMessage = getErrorMessage(error);
}
this.logger.error(
errorMessage,
error
);
throw new Error(errorMessage);
}
const reference = result.reference.toString();
return getResponseWithStructuredContent({
reference,
topicString: memoryTopic,
topic,
feedUrl: `${this.bee.url}/feeds/${owner}/${topic}`,
message: "Data successfully uploaded to Swarm and linked to feed."
});
}
}
export {
UpdateFeedTool
};
//# sourceMappingURL=index57.js.map