Lifei/sub recipe desktop temp (#3576)

crates/goose-server/src/openapi.rs

@@ -305,6 +305,7 @@ derive_utoipa!(ResourceContents as ResourceContentsSchema);
         super::routes::config_management::providers,
         super::routes::config_management::upsert_permissions,
         super::routes::agent::get_tools,
+        super::routes::agent::add_sub_recipes,
         super::routes::reply::confirm_permission,
         super::routes::context::manage_context,
         super::routes::session::list_sessions,
@@ -394,6 +395,8 @@ derive_utoipa!(ResourceContents as ResourceContentsSchema);
         goose::recipe::SubRecipe,
         goose::agents::types::RetryConfig,
         goose::agents::types::SuccessCheck,
+        super::routes::agent::AddSubRecipesRequest,
+        super::routes::agent::AddSubRecipesResponse,
     ))
 )]
 pub struct ApiDoc;

crates/goose-server/src/routes/agent.rs

@@ -6,7 +6,6 @@ use axum::{
     routing::{get, post},
     Json, Router,
 };
-use goose::config::Config;
 use goose::config::PermissionManager;
 use goose::model::ModelConfig;
 use goose::providers::create;
@@ -15,6 +14,7 @@ use goose::{
     agents::{extension::ToolInfo, extension_manager::get_parameter_names},
     config::permission::PermissionLevel,
 };
+use goose::{config::Config, recipe::SubRecipe};
 use serde::{Deserialize, Serialize};
 use std::collections::HashMap;
 use std::sync::Arc;
@@ -35,6 +35,16 @@ struct ExtendPromptResponse {
     success: bool,
 }
 
+#[derive(Deserialize, utoipa::ToSchema)]
+pub struct AddSubRecipesRequest {
+    sub_recipes: Vec<SubRecipe>,
+}
+
+#[derive(Serialize, utoipa::ToSchema)]
+pub struct AddSubRecipesResponse {
+    success: bool,
+}
+
 #[derive(Deserialize)]
 struct ProviderFile {
     name: String,
@@ -88,6 +98,30 @@ async fn get_versions() -> Json<VersionsResponse> {
     })
 }
 
+#[utoipa::path(
+    post,
+    path = "/agent/add_sub_recipes",
+    request_body = AddSubRecipesRequest,
+    responses(
+        (status = 200, description = "added sub recipes to agent successfully", body = AddSubRecipesResponse),
+        (status = 401, description = "Unauthorized - invalid secret key"),
+    ),
+)]
+async fn add_sub_recipes(
+    State(state): State<Arc<AppState>>,
+    headers: HeaderMap,
+    Json(payload): Json<AddSubRecipesRequest>,
+) -> Result<Json<AddSubRecipesResponse>, StatusCode> {
+    verify_secret_key(&headers, &state)?;
+
+    let agent = state
+        .get_agent()
+        .await
+        .map_err(|_| StatusCode::PRECONDITION_FAILED)?;
+    agent.add_sub_recipes(payload.sub_recipes.clone()).await;
+    Ok(Json(AddSubRecipesResponse { success: true }))
+}
+
 async fn extend_prompt(
     State(state): State<Arc<AppState>>,
     headers: HeaderMap,
@@ -318,5 +352,6 @@ pub fn routes(state: Arc<AppState>) -> Router {
             post(update_router_tool_selector),
         )
         .route("/agent/session_config", post(update_session_config))
+        .route("/agent/add_sub_recipes", post(add_sub_recipes))
         .with_state(state)
 }