Adds system prompt as constructor arg

galadriel-ai · Jul 5, 2024 · eafa75c · eafa75c
1 parent 359c06b
commit eafa75c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/contracts/contracts/AiTownAgent.sol b/contracts/contracts/AiTownAgent.sol
@@ -29,14 +29,18 @@ contract AiTownAgent {
     // @notice Address of the oracle contract
     address public oracleAddress;
 
+    // @notice System prompt for the agent
+    string public systemPrompt;
+
     // @notice Event emitted when the oracle address is updated
     event OracleAddressUpdated(address indexed newOracleAddress);
 
     // @param initialOracleAddress Initial address of the oracle contract
     // @param systemPrompt System prompt for the run
-    constructor(address initialOracleAddress) {
+    constructor(address initialOracleAddress, string memory _systemPrompt) {
         owner = msg.sender;
         oracleAddress = initialOracleAddress;
+        systemPrompt = _systemPrompt;
     }
 
     // @notice Ensures the caller is the contract owner
@@ -56,7 +60,7 @@ contract AiTownAgent {
         emit OracleAddressUpdated(newOracleAddress);
     }
 
-    function startChat(string memory conversationId, string memory systemPrompt, string memory message) public returns (uint) {
+    function startChat(string memory conversationId, string memory message) public returns (uint) {
         ChatRun storage run = chatRuns[chatRunsCount];
         conversationIdToChatRun[conversationId] = chatRunsCount;
         run.owner = msg.sender;