Skip to content

Latest commit

 

History

History
230 lines (177 loc) · 6.13 KB

README.md

File metadata and controls

230 lines (177 loc) · 6.13 KB

TEE Agent

Concept

This is a tee agent running in a nitro enclave instance. The code of the agent is signed by aws and its public key is generated randomly at start. No one but the agent has access to the corresponding private key.

Verify agent

Verify the code attestation

The attestation is signed by aws and contains the code hash. To verify the attestation you need the aws root certificate.

Here is a rust crate to parse and verify the attestation tee-attestation-verifier

Verify the code hash

The code hash is a representation of the code of the agent.

To rebuild the code hash, you have to follow the steps belows to rebuild the enclave image from source, within an ec2 instance. Then run the enclave image and measure the PCRs value. The second PCR value correspond to the code hash of the agent that was published.

see article on aws code attestation

see also nitriding doc

Setup ec2 nitro instance

You can do it with UI (recommended) or with CLI:

aws ec2 run-instances \
--image-id ami-0b5eea76982371e91 \
--count 1 \
--instance-type m5.xlarge \
--key-name <key-name> \
--security-group-ids sg-0a006bcdd6147509f \
--tag-specifications 'ResourceType=instance,Tags=[{Key=Name,Value=my-nitro-tee-1}]' \
--enclave-options 'Enabled=true'

If you go with the UI:

  • Pick a instance good enough such as md5.xlarge, enable nitro
  • Make sure that the ports needed by your application are open by checking the security group, in "security" tab in the instance screen.

Setup nitro cli

see doc

sudo /setup.sh

Allocate more memory for the enclave if necessary

sudo nano /etc/nitro_enclaves/allocator.yaml

sudo systemctl restart nitro-enclaves-allocator.service

Run enclave

  • clone this repo in the instance
make

Start gvproxy

Gvproxy is a reverse proxy that allows the enclave to communicate with the outside world.

Use screen to run gvproxy in background

screen
./gvproxy.sh

Run in production mode

sudo ./run-enclave.sh <image_eif>

Test

Get current instance ip

curl ifconfig.me

call config endpoint to confirm that the enclave is running

curl http://127.0.0.1:7047/api/v1/config

Measure enclave PCRS to verify code hash

nitro-cli describe-enclaves

Setup environment variables

The env variable system is similar to variables in a smart contract. An operator address is hardcoded before enclave initialization. After deployment, the operator can set the environment variables using a SECP256k1 signature. Message is (tee_name, function, var_name, new_value). Variables can also have immutable property, and therefore be set only once. The difference with a smart contract is that the environment variables can be set to hidden. Hidden variables can be changed to public, but not the other way. Similarly, mutable variables can be changed to immutable, but not the other way.

Example:

curl --location 'http://127.0.0.1:7047/api/v1//variables' \
--header 'Content-Type: application/json' \
--data '{
    "variables": [ {
    "name": "anthropic_api_key" ,
    "value": "sk-proj-f-xxxxxx",
    "signature": "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
}]
}
'

Prompting Chat Agent

All responses are signed using the TEE Agent key.

        result = {}
        result["model"] = model
        result["response"] = prompt_openrouter(model, system_prompt, messages, tools, tool_choice)
        result["messages"] = messages
        result["tools"] = tools
        result["system_prompt"] = system_prompt
        result["signature"] = sign_ethereum_message(str(result), private_key.to_string().hex())

Query endpoint

curl --location 'http://127.0.0.1:7047/api/v1/prompt' \
--header 'Content-Type: application/json' \
--data '{
    "system_prompt": "You are a helpful AI assistant focused on technical writing and documentation.",
    "messages": [
        {
            "role": "user",
            "content": "give me the weather in paris"
        }
    ],
    "tools": [
        {
            "type": "function",
            "function": {
                "name": "get_weather",
                "description": "Get the current weather",
                "parameters": {
                    "type": "object",
                    "properties": {
                        "location": {
                            "type": "string"
                        },
                        "unit": {
                            "type": "string",
                            "enum": [
                                "celsius",
                                "fahrenheit"
                            ]
                        }
                    },
                    "required": [
                        "location"
                    ]
                }
            }
        }
    ]
}'

Signing and broadcasting transaction

POST /api/v1/sign-transaction

ETH transfer Request

{
   "transaction": {
       "to": "0x86693CbE3f2581DF512635823F46f0d593D96dbA",
       "value": 3,
       "data": "0x"
   },
   "rpc_url": "https://sepolia.base.org"
}

ERC20 transfer Request

{
   "transaction": {
       "to": "0x1c23A3912247f104D707dfEf41F4B7842B241d4B",
       "value": 0,
       "data": "0xa9059cbb00000000000000000000000000693cbe3f2581df512635823f46f0d593d96dba0000000000000000000000000000000000000000000000000000048c27395000",
       "gas": 60000
   },
   "rpc_url": "https://sepolia.base.org"
}

Response

{
   "signed_transaction": "0xf8668180830f43cb8252089486693cbe3f2581df512635823f46f0d593d96dba03808302948ca01a05e840f30baa629670d7c7c68b1ed3898a6f5fbd5ff3581663f3de323ccf1ba03d43ae83cdf13530ee45741169d71c6fef92bf6daa0015b4716e92380c7df827",
   "transaction_hash": "0xde41dd0a0eac568e51306acff9742fe6503f4c57c0735f43b813a82be260bc48"
}