Skip to content

Commit 2ea872a

Browse files
authored
fix TEE Log plugin errors at agent startup (elizaOS#2415)
* fix TEE Log configuration error * fix error telling 'Cannot open database because the directory does not exist' * fix error telling 'Invalid TEE_MODE: undefined. Must be one of: LOCAL, DOCKER, PRODUCTION' at start up
1 parent cd0388d commit 2ea872a

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

packages/plugin-tee-log/src/services/teeLogManager.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ export class TeeLogManager {
9898
const sgxAttestation = await sgxAttestationProvider.generateAttestation(userReport);
9999
return JSON.stringify(sgxAttestation);
100100
} else if (this.teeType === TeeType.TDX_DSTACK) {
101-
const tdxAttestationProvider = new TdxAttestationProvider();
101+
const tdxAttestationProvider = new TdxAttestationProvider(this.teeMode);
102102
const tdxAttestation = await tdxAttestationProvider.generateAttestation(userReport);
103103
return JSON.stringify(tdxAttestation);
104104
} else {

packages/plugin-tee-log/src/services/teeLogService.ts

+5-2
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,10 @@ import { SqliteTeeLogDAO } from "../adapters/sqliteDAO";
44
import { TeeType, TeeLogDAO, TeeAgent, TeeLog, TeeLogQuery, PageQuery } from "../types";
55
import { TeeLogManager } from "./teeLogManager";
66
import Database from "better-sqlite3";
7+
import path from "path";
78

89
export class TeeLogService extends Service implements ITeeLogService {
9-
private readonly dbPath = "./data/tee_log.sqlite";
10+
private readonly dbPath = path.resolve("agent/data/tee_log.sqlite");
1011

1112
private initialized: boolean = false;
1213
private enableTeeLog: boolean = false;
@@ -46,8 +47,10 @@ export class TeeLogService extends Service implements ITeeLogService {
4647
const teeMode = runtime.getSetting("TEE_MODE");
4748
const walletSecretSalt = runtime.getSetting("WALLET_SECRET_SALT");
4849

50+
this.teeMode = teeMode ? TEEMode[teeMode as keyof typeof TEEMode] : TEEMode.OFF;
51+
4952
const useSgxGramine = runInSgx && enableValues.includes(runInSgx.toLowerCase());
50-
const useTdxDstack = !teeMode && teeMode !== TEEMode.OFF && walletSecretSalt;
53+
const useTdxDstack = teeMode && teeMode !== TEEMode.OFF && walletSecretSalt;
5154

5255
if (useSgxGramine && useTdxDstack) {
5356
throw new Error("Cannot configure both SGX and TDX at the same time.");

0 commit comments

Comments
 (0)