fix(deploy): improve deploy script

Author: dineshpinto

.dockerignore

@@ -17,6 +17,7 @@ wheels/
 *.egg-info
 
 # Other folders
+.github
 docs/
 examples/
 tests/

.env.example

@@ -22,7 +22,7 @@ GCP__MACHINE_TYPE=c3-standard-4
 GCP__SERVICE_ACCOUNT=confidential-sa@verifiable-ai-hackathon.iam.gserviceaccount.com 
 GCP__TEE_IMAGE_REFERENCE=ghcr.io/flare-foundation/flare-ai-kit:main
 # For production, use `confidential-space-250301`
-GCP__IMAGE=confidential-space-debug-250301
+GCP__CONFIDENTIAL_IMAGE=confidential-space-debug-250301
 # For production, use `false`
 GCP__TEE_CONTAINER_LOG_REDIRECT=true
 GCP__CONFIDENTIAL_COMPUTE_TYPE=TDX

README.md

@@ -22,41 +22,64 @@ SDK for building **verifiable AI Agents** on Flare using Confidential Space.
 The kit is composed of modular engines for agents, social feeds, onchain data, and consensus.
 
 ```mermaid
-graph TD
-    A(Flare AI Kit)
-    A --> B{Agent Framework - PydanticAI}
-    A --> C{VectorRAG Engine - PostgreSQL + pgvector}
-    A --> D{GraphRAG Engine - Neo4j}
-    A --> E{Secure Enclave - Confidential Space + TDX}
-    A --> F{Ecosystem Engine}
-    A --> G{Social Engine}
-    A --> H{Consensus Engine}
-
-    B --> I(Gemini, GPT, Grok +200 models)
-
-    C --o DevHub/News/Governance
-
-    D --o MainnetTxData
-
-    E --> vTPM
-    vTPM --o RA-TLS
-
-    F --o M(Protocols)
-    M --o FTSO
-    M --o FDC
-    M --o FAssets
-
-    F --o L(Applications)
-    L --o OpenOcean
-    L --o Kinetic
-    L --o SparkDEX
-    L --o Cyclo
-    L --o ...
+flowchart TD
+    A["Flare AI Kit"]
+
+    %% Agent Framework subgraph
+    subgraph AgentFramework["Agent Framework"]
+        B["Google ADK"]
+        B --o LLM["Gemini<br>GPT<br>Grok<br>+200 models"]
+    end
+
+    %% VectorRAG Engine subgraph
+    subgraph VectorRAG["VectorRAG Engine"]
+        C["Qdrant"]
+        C --o SOURCE[DevHub<br>Flare News<br>Flare Governance]
+    end
+
+    %% Secure Enclave subgraph
+    subgraph SecureEnclave["Secure Enclave"]
+        E["Confidential Space"]
+        E --> HW[Intel TDX]
+        HW --o RA[RA-verify<br>RA-TLS]
+    end
+
+    %% Ecosystem Engine subgraph
+    subgraph EcosystemEngine["Ecosystem Engine"]
+        F[Ecosystem Engine]
+        F --> PR[Protocols]
+        PR --o PROTOS["FTSO<br>FDC<br>FAssets"]
+        F --> AP[Applications]
+        AP --o APPS[SparkDEX<br>OpenOcean<br>Kinetic<br>Cyclo]
+    end
+
+    %% Social Engine subgraph
+    subgraph SocialEngine["Social Engine"]
+        G["Social"]
+        G --o SOC["X<br>Telegram<br>Farcaster<br>Slack"]
+    end
+
+    %% Consensus Engine subgraph
+    subgraph ConsensusEngine["Consensus Engine"]
+        H["Consensus"]
+        H --o ALGOS[Majority<br>Tournament<br>Clustering]
+    end
+
+    %% Connections to Flare AI Kit central node
+    A --> B
+    A --> C
+    A --> E
+    A --> F
+    A --> G
+    A --> H
 ```
 
 ## 📦 Getting Started
 
-**Prerequisites** Python >= 3.12 and [uv](https://github.com/astral-sh/uv).
+**Prerequisites**
+
+- [uv](https://github.com/astral-sh/uv) with Python >= 3.12
+- [Docker](https://www.docker.com).
 
 1. **Clone the repo:**
 
@@ -94,6 +117,13 @@ uv run pyright
 uv run pytest
 ```
 
+## 🚧 Build with Docker
+
+```bash
+docker build -t flare-ai-kit .
+docker run -rm --env-file .env flare-ai-kit
+```
+
 ## ☁️ Deploy to Confidential Space
 
 **Prerequisites:** Authenticated [gcloud CLI](https://cloud.google.com/sdk/docs/install).

gcloud-deploy.sh

@@ -26,7 +26,7 @@ fi
 : "${GCP__MACHINE_TYPE:?Please set GCP__MACHINE_TYPE}"
 : "${GCP__TEE_IMAGE_REFERENCE:?Please set GCP__TEE_IMAGE_REFERENCE}"
 : "${GCP__SERVICE_ACCOUNT:?Please set GCP__SERVICE_ACCOUNT}"
-: "${GCP__IMAGE:?Please set GCP__IMAGE}"
+: "${GCP__CONFIDENTIAL_IMAGE:?Please set GCP__CONFIDENTIAL_IMAGE}"
 : "${GCP__CONFIDENTIAL_COMPUTE_TYPE:?Please set GCP__CONFIDENTIAL_COMPUTE_TYPE}"
 : "${GCP__SCOPES:?Please set GCP__SCOPES}"
 : "${GCP__TAGS:?Please set GCP__TAGS}"
@@ -36,7 +36,7 @@ echo "--> Creating instance '$GCP__INSTANCE_NAME' in project '$GCP__PROJECT' wit
 echo "      - Zone: $GCP__ZONE"
 echo "      - Machine Type: $GCP__MACHINE_TYPE"
 echo "      - Service Account: $GCP__SERVICE_ACCOUNT"
-echo "      - Image: $GCP__IMAGE"
+echo "      - Image: $GCP__CONFIDENTIAL_IMAGE"
 echo "      - TEE Image Reference: $GCP__TEE_IMAGE_REFERENCE"
 echo "      - Confidential Compute Type: $GCP__CONFIDENTIAL_COMPUTE_TYPE"
 echo "      - Scopes: $GCP__SCOPES"
@@ -77,7 +77,7 @@ COMMAND=(
   --service-account="$GCP__SERVICE_ACCOUNT"
   --scopes="$GCP__SCOPES"
   --tags="$GCP__TAGS"
-  --create-disk=auto-delete=yes,boot=yes,device-name="$GCP__INSTANCE_NAME",image=projects/confidential-space-images/global/images/"$GCP__IMAGE",mode=rw,size=11,type=pd-balanced
+  --create-disk=auto-delete=yes,boot=yes,device-name="$GCP__INSTANCE_NAME",image=projects/confidential-space-images/global/images/"$GCP__CONFIDENTIAL_IMAGE",mode=rw,size=11,type=pd-balanced
   --shielded-secure-boot
   --shielded-vtpm
   --shielded-integrity-monitoring