Adds chat with RAG deployment task (#54)

galadriel-ai · Apr 2, 2024 · 7cc5e1d · 7cc5e1d
1 parent 07f0e1e
commit 7cc5e1d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/contracts/hardhat.config.ts b/contracts/hardhat.config.ts
@@ -1,6 +1,7 @@
 import {HardhatUserConfig} from "hardhat/config";
 import "@nomicfoundation/hardhat-toolbox";
 import "./tasks/whitelist";
+import "./tasks/deployChatWithRAG";
 
 require('dotenv').config()
 

diff --git a/contracts/tasks/deployChatWithRAG.ts b/contracts/tasks/deployChatWithRAG.ts
@@ -0,0 +1,14 @@
+import { task } from "hardhat/config";
+
+task("deployChatWithRAG", "Deploys the chat contract with knowledge base")
+  .addParam("oracleAddress", "The address of the Oracle contract")
+  .addParam("cid", "Knowledge base CID")
+  .setAction(async (taskArgs, hre) => {
+    const oracleAddress = taskArgs.oracleAddress;
+    const knowledgeBaseCID = taskArgs.cid;
+    const contract = await hre.ethers.deployContract("ChatGpt", [oracleAddress, knowledgeBaseCID], {});
+    await contract.waitForDeployment();
+    console.log(`RAG deployed to: ${contract.target}`);
+  });
+
+