Initial public release of Warp.

Repo-Sync-Origin: warpdotdev/warp-internal@12af1d983b
2026-04-28 08:43:33 -05:00
commit 0dbd3d567a
4982 changed files with 1431549 additions and 0 deletions
@@ -0,0 +1,95 @@
+use warp_completer::util::parse_current_commands_and_tokens;
+
+use crate::{Context, test_utils::CompletionContext};
+
+use super::*;
+
+async fn mock_parsed_input_token(buffer_text: String) -> ParsedTokensSnapshot {
+    let completion_context = CompletionContext::new();
+    parse_current_commands_and_tokens(buffer_text, &completion_context).await
+}
+
+#[test]
+fn test_input_detection() {
+    futures::executor::block_on(async move {
+        let classifier = HeuristicClassifier;
+
+        let mut context = Context {
+            current_input_type: InputType::AI,
+            is_agent_follow_up: false,
+        };
+
+        let token = mock_parsed_input_token("cargo --version".to_string()).await;
+        assert_eq!(
+            classifier.detect_input_type(token, &context).await,
+            InputType::Shell
+        );
+
+        // We have to override the first token description here given the mocked completion
+        // parser will parse the first token always as commands.
+        //
+        // Mock the case where cargo is not installed. We should still parse this as Shell input.
+        let mut token = mock_parsed_input_token("cargo --version".to_string()).await;
+        token.parsed_tokens[0].token_description = None;
+        assert_eq!(
+            classifier.detect_input_type(token, &context).await,
+            InputType::Shell
+        );
+
+        let mut token = mock_parsed_input_token("rvm install 3.3".to_string()).await;
+        token.parsed_tokens[0].token_description = None;
+        assert_eq!(
+            classifier.detect_input_type(token, &context).await,
+            InputType::Shell
+        );
+
+        // Short queries with NL should be parsed as AI input when already in AI input.
+        let mut token = mock_parsed_input_token("Explain this".to_string()).await;
+        token.parsed_tokens[0].token_description = None;
+        assert_eq!(
+            classifier.detect_input_type(token.clone(), &context).await,
+            InputType::AI
+        );
+
+        context.current_input_type = InputType::Shell;
+
+        // Typing "fix this" after an error block is a common use case.
+        let mut token = mock_parsed_input_token("fix this".to_string()).await;
+        token.parsed_tokens[0].token_description = None;
+        assert_eq!(
+            classifier.detect_input_type(token, &context).await,
+            InputType::AI,
+        );
+
+        // Short queries with punctuation should be parsed as AI input.
+        let token = mock_parsed_input_token("What went wrong?".to_string()).await;
+        assert_eq!(
+            classifier.detect_input_type(token, &context).await,
+            InputType::AI
+        );
+        // Short queries with contractions should be parsed as AI input.
+        let mut token = mock_parsed_input_token("What's the reason".to_string()).await;
+        token.parsed_tokens[0].token_description = None;
+        assert_eq!(
+            classifier.detect_input_type(token, &context).await,
+            InputType::AI
+        );
+
+        // Short queries with quotations should be parsed as AI input.
+        let mut token =
+            mock_parsed_input_token("The message is \"utils::future ... ok\"".to_string()).await;
+        token.parsed_tokens[0].token_description = None;
+        assert_eq!(
+            classifier.detect_input_type(token, &context).await,
+            InputType::AI
+        );
+
+        // String tokens with special shell syntax should not be treated as negative NL signal.
+        let mut token = mock_parsed_input_token("The type is \"<>\"".to_string()).await;
+        token.parsed_tokens[0].token_description = None;
+        assert_eq!(
+            classifier.detect_input_type(token, &context).await,
+            InputType::AI
+        );
+    });
+}