Skip to content

Commit 123decd

Browse files
committed
prefetches prompts if applicable
1 parent 67480bf commit 123decd

File tree

2 files changed

+14
-5
lines changed

2 files changed

+14
-5
lines changed

crates/chat-cli/src/cli/chat/cli/prompts.rs

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -230,7 +230,7 @@ pub enum PromptsSubcommand {
230230
impl PromptsSubcommand {
231231
pub async fn execute(self, session: &mut ChatSession) -> Result<ChatState, ChatError> {
232232
let PromptsSubcommand::Get {
233-
orig_input: _,
233+
orig_input,
234234
name,
235235
arguments,
236236
} = self
@@ -286,10 +286,8 @@ impl PromptsSubcommand {
286286
session.pending_prompts.clear();
287287
session.pending_prompts.append(&mut VecDeque::from(prompts.messages));
288288

289-
execute!(session.stderr, style::Print("\n"))?;
290-
291-
Ok(ChatState::PromptUser {
292-
skip_printing_tools: true,
289+
Ok(ChatState::HandleInput {
290+
input: orig_input.unwrap_or_default(),
293291
})
294292
}
295293

crates/chat-cli/src/mcp_client/client.rs

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,17 @@ where
152152
service_method: list_tools,
153153
result_field: tools,
154154
messenger_method: send_tools_list_result,
155+
service: service_clone.clone(),
156+
messenger: messenger_clone,
157+
server_name: server_name
158+
};
159+
160+
paginated_fetch! {
161+
final_result_type: ListPromptsResult,
162+
content_type: rmcp::model::Prompt,
163+
service_method: list_prompts,
164+
result_field: prompts,
165+
messenger_method: send_prompts_list_result,
155166
service: service_clone,
156167
messenger: messenger_clone,
157168
server_name: server_name

0 commit comments

Comments
 (0)