do_memory_mcp/server/tools/embeddings/
generate.rs

1//! Generate embedding tool handler for MCP server
2//!
3//! This module provides the tool for generating embedding vectors
4//! for text using the configured embedding provider.
5
6use crate::server::MemoryMCPServer;
7use anyhow::Result;
8use serde_json::json;
9use std::sync::Arc;
10use tracing::debug;
11
12impl MemoryMCPServer {
13    /// Execute the generate_embedding tool
14    ///
15    /// # Arguments
16    ///
17    /// * `input` - Parameters for embedding generation
18    ///
19    /// # Returns
20    ///
21    /// Returns the generated embedding vector with metadata
22    pub async fn execute_generate_embedding(
23        &self,
24        input: crate::mcp::tools::embeddings::GenerateEmbeddingInput,
25    ) -> Result<serde_json::Value> {
26        self.track_tool_usage("generate_embedding").await;
27
28        // Start monitoring request
29        let request_id = format!(
30            "generate_embedding_{}",
31            std::time::SystemTime::now()
32                .duration_since(std::time::UNIX_EPOCH)
33                .unwrap_or_default()
34                .as_nanos()
35        );
36        self.monitoring
37            .start_request(request_id.clone(), "generate_embedding".to_string())
38            .await;
39
40        debug!("Generating embedding for text ({} chars)", input.text.len());
41
42        let tool = crate::mcp::tools::embeddings::EmbeddingTools::new(Arc::clone(&self.memory));
43
44        let result = tool.execute_generate_embedding(input).await;
45
46        // End monitoring request
47        self.monitoring
48            .end_request(&request_id, result.is_ok(), None)
49            .await;
50
51        let output = result?;
52
53        // Convert result to JSON
54        Ok(json!(output))
55    }
56}
57
58#[cfg(test)]
59mod tests {
60    use super::*;
61
62    #[test]
63    #[allow(clippy::manual_async_fn)]
64    fn test_generate_embedding_signature_compile() {
65        // This test ensures the method signature compiles correctly
66        use crate::mcp::tools::embeddings::GenerateEmbeddingInput;
67        fn method_signature(
68            _server: &MemoryMCPServer,
69            _input: GenerateEmbeddingInput,
70        ) -> impl std::future::Future<Output = Result<serde_json::Value>> {
71            async { Ok(json!({})) }
72        }
73        let _ = method_signature; // Use the function to avoid unused warnings
74    }
75}
do_memory_mcp/server/tools/embeddings/generate.rs

do_memory_mcp/server/tools/embeddings/
generate.rs
