fix last items for cleanup

elizaOS · Jan 4, 2025 · 6f24839 · 6f24839
1 parent aadf8cb
commit 6f24839
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 9 deletions.
diff --git a/docs/docs/packages/plugins.md b/docs/docs/packages/plugins.md
@@ -449,6 +449,10 @@ const response = await runtime.triggerAction("INVOKE_CONTRACT", {
 
 Integrates [Dstack SDK](https://github.com/Dstack-TEE/dstack) to enable TEE (Trusted Execution Environment) functionality and deploy secure & privacy-enhanced Eliza Agents:
 
+**Actions:**
+
+- `REMOTE_ATTESTATION` - Generate a Remote Attestation Quote based on `runtime.agentId` when the agent is prompted for a remote attestation. The quote is uploaded to the [proof.t16z.com](https://proof.t16z.com) service and the agent is informed of the attestation report URL.
+
 **Providers:**
 
 - `deriveKeyProvider` - Allows for secure key derivation within a TEE environment. It supports deriving keys for both Solana (Ed25519) and Ethereum (ECDSA) chains.
@@ -526,8 +530,12 @@ docker run --rm -p 8090:8090 phalanetwork/tappd-simulator:latest
 When using the provider through the runtime environment, ensure the following settings are configured:
 
 ```env
- # Optional, for simulator purposes if testing on mac or windows. Leave empty for Linux x86 machines.
-DSTACK_SIMULATOR_ENDPOINT="http://host.docker.internal:8090"
+# TEE_MODE options:
+# - LOCAL: Uses simulator at localhost:8090 (for local development)
+# - DOCKER: Uses simulator at host.docker.internal:8090 (for docker development)
+# - PRODUCTION: No simulator, uses production endpoints
+# Defaults to OFF if not specified
+TEE_MODE=OFF # LOCAL | DOCKER | PRODUCTION
 WALLET_SECRET_SALT=your-secret-salt // Required to single agent deployments
 ```
 

diff --git a/packages/plugin-tee/src/actions/remoteAttestation.ts b/packages/plugin-tee/src/actions/remoteAttestation.ts
@@ -1,6 +1,6 @@
 import type { IAgentRuntime, Memory, State, HandlerCallback } from "@elizaos/core";
 import { RemoteAttestationProvider } from "../providers/remoteAttestationProvider";
-import { fetch, type BodyInit } from 'undici'
+import { fetch, type BodyInit } from "undici";
 
 function hexToUint8Array(hex: string) {
     hex = hex.trim();
@@ -58,11 +58,9 @@ export const remoteAttestationAction = {
             const response = await uploadUint8Array(attestationData);
             const data = await response.json();
             callback({
-                text: `Uploading attestation to proof.t16z.com...`,
-                action: "NONE",
-            });
-            callback({
-                text: `Attestation report for ${agentName}: https://proof.t16z.com/reports/${data.checksum}`,
+                text: `Here's my 🧾 RA Quote 🫡
+
+                https://proof.t16z.com/reports/${data.checksum}`,
                 action: "NONE",
             });
             return true;
@@ -85,7 +83,7 @@ export const remoteAttestationAction = {
             {
                 user: "{{user2}}",
                 content: {
-                    text: "One second, let me generate and upload my attestation report",
+                    text: "Of course, one second...",
                     action: "REMOTE_ATTESTATION",
                 },
             }

diff --git a/packages/plugin-tee/tsup.config.ts b/packages/plugin-tee/tsup.config.ts
@@ -24,5 +24,6 @@ export default defineConfig({
         "@solana/buffer-layout",
         "stream",
         "buffer",
+        "undici",
     ],
 });