Initial public release of Warp.

Repo-Sync-Origin: warpdotdev/warp-internal@12af1d983b

app/src/ai/predict/generate_am_query_suggestions/api/mod.rs

@@ -0,0 +1,13 @@
+//! This module contains Rust types for the GenerateAMQuerySuggestions endpoint in warp-server that
+//! serves Agent Mode.
+//!
+//! These types are manually transposed from the API schema defined in go
+//! (warp-server/model/types/generate_am_query_suggestions/(request.go|response.go|common.go)).
+//!
+//! Documentation on the types here is directly borrowed from the documentation on the go schema;
+//! see the go schema for the source-of-truth.
+mod request;
+mod response;
+
+pub use request::*;
+pub use response::*;

app/src/ai/predict/generate_am_query_suggestions/api/request.rs

@@ -0,0 +1,25 @@
+//! Rust version of `GenerateAMQuerySuggestions` and its fields.
+//!
+//! These types correspond to the warp-go types defined in
+//! warp-server/model/types/generate_am_query_suggestions/request.go.
+//!
+//! Documentation on the types here is directly borrowed from the documentation on the go schema;
+//! see the go schema for the source-of-truth.
+
+use serde::{Deserialize, Serialize};
+
+#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq)]
+/// Top-level request type for the `GenerateAMQuerySuggestion` API endpoint.
+pub struct GenerateAMQuerySuggestionsRequest {
+    /// The previous blocks that were run in the session. Each item in the array is expected to correspond to 1 block.
+    /// TODO(advait): we've purposely switched over to a free-form string for faster iteration here. We should
+    /// switch back to strongly typed fields, once we've figured out the right schema/context/API for this.
+    pub context_messages: Vec<String>,
+
+    /// System/platform relevant context for system prompt.
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub system_context: Option<String>,
+
+    /// Exit code for the command run.
+    pub exit_code: i32,
+}

app/src/ai/predict/generate_am_query_suggestions/api/response.rs

@@ -0,0 +1,65 @@
+//! Rust version of `GenerateAMQuerySuggestionsResponse` and its fields.
+//!
+//! These types correspond to the warp-go types defined in
+//! warp-server/model/types/generate_am_query_suggestions/response.go.
+//!
+//! Documentation on the types here is directly borrowed from the documentation on the go schema;
+//! see the go schema for the source-of-truth.
+use serde::{Deserialize, Serialize};
+
+use crate::ai::agent::FileLocations;
+
+/// Top-level response type for the `GenerateAMQuerySuggestions` API endpoint.
+#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq)]
+pub struct GenerateAMQuerySuggestionsResponse {
+    pub id: String,
+    pub suggestion: Option<Suggestion>,
+}
+
+/// Represents a particular type of suggestion.
+#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
+#[serde(rename_all = "lowercase")]
+pub enum Suggestion {
+    Simple(SimpleQuery),
+    Coding(CodingQuery),
+}
+
+/// Simple suggestion structure.
+#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq)]
+pub struct SimpleQuery {
+    pub query: String,
+    /// If the query is a complex task, we should use Dispatch to create a plan first.
+    pub should_plan_task: bool,
+}
+
+impl GenerateAMQuerySuggestionsResponse {
+    /// Check whether the response contains a valid code delegation, which should include:
+    /// 1) the coding_query field is non-empty.
+    /// 2) the attached file locations to the coding_query is non-empty.
+    pub fn is_valid_code_delegation(&self) -> bool {
+        matches!(&self.suggestion, Some(Suggestion::Coding(coding_query)) if !coding_query.files.is_empty())
+    }
+}
+
+#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq)]
+pub struct CodingQuery {
+    pub files: Vec<GeneratedFileLocations>,
+    pub query: String,
+}
+
+#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq)]
+pub struct GeneratedFileLocations {
+    pub file_name: String,
+    pub line_numbers: Option<Vec<usize>>,
+}
+
+impl From<GeneratedFileLocations> for FileLocations {
+    fn from(value: GeneratedFileLocations) -> Self {
+        Self {
+            name: value.file_name,
+            // We are explicitly disgarding the line_numbers right now.
+            // TODO(kevin): Convert them into Range<usize>.
+            lines: Vec::new(),
+        }
+    }
+}