feat(anda_engine_server): add extra_info

Author: zensh

anda_engine_server/Cargo.toml

@@ -3,7 +3,7 @@ name = "anda_engine_server"
 description = "A http server to serve multiple Anda engines."
 repository = "https://github.com/ldclabs/anda/tree/main/anda_engine_server"
 publish = true
-version.workspace = true
+version = "0.9.9"
 edition.workspace = true
 keywords.workspace = true
 categories.workspace = true

anda_engine_server/src/handler.rs

@@ -26,6 +26,7 @@ pub struct AppState {
     pub(crate) engines: Arc<BTreeMap<Principal, Engine>>,
     pub(crate) default_engine: Principal,
     pub(crate) start_time_ms: u64,
+    pub(crate) extra_info: Arc<BTreeMap<String, Json>>,
 }
 
 /// GET /.well-known/information
@@ -49,6 +50,7 @@ pub async fn get_information(
         default_engine: app.default_engine,
         start_time_ms: app.start_time_ms,
         caller,
+        extra_info: app.extra_info.as_ref().clone(),
     };
 
     match Content::from(&headers) {

anda_engine_server/src/lib.rs

@@ -1,4 +1,4 @@
-use anda_core::BoxError;
+use anda_core::{BoxError, Json};
 use anda_engine::engine::Engine;
 use axum::{Router, routing};
 use candid::Principal;
@@ -25,6 +25,7 @@ pub struct ServerBuilder {
     engines: BTreeMap<Principal, Engine>,
     default_engine: Option<Principal>,
     middlewares: Vec<Arc<dyn HttpMiddleware>>,
+    extra_info: BTreeMap<String, Json>,
 }
 
 impl Default for ServerBuilder {
@@ -46,6 +47,7 @@ impl ServerBuilder {
             engines: BTreeMap::new(),
             default_engine: None,
             middlewares: Vec::new(),
+            extra_info: BTreeMap::new(),
         }
     }
 
@@ -69,6 +71,11 @@ impl ServerBuilder {
         self
     }
 
+    pub fn with_extra_info(mut self, extra_info: BTreeMap<String, Json>) -> Self {
+        self.extra_info = extra_info;
+        self
+    }
+
     pub fn with_engines(
         mut self,
         mut engines: BTreeMap<Principal, Engine>,
@@ -163,6 +170,7 @@ impl ServerBuilder {
             engines: Arc::new(self.engines),
             default_engine,
             start_time_ms: unix_ms(),
+            extra_info: Arc::new(self.extra_info),
         };
 
         // Build a router that is still "missing" an `AppState`.

anda_engine_server/src/types.rs

@@ -1,11 +1,14 @@
+use anda_core::Json;
 use anda_engine::engine::AgentInfo;
 use candid::Principal;
 use serde::{Deserialize, Serialize};
+use std::collections::BTreeMap;
 
 #[derive(Clone, Debug, Deserialize, Serialize)]
 pub struct AppInformation {
     pub engines: Vec<AgentInfo>,
     pub default_engine: Principal,
     pub caller: Principal,
     pub start_time_ms: u64,
+    pub extra_info: BTreeMap<String, Json>,
 }