Initial public release of Warp.

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

crates/graphql/src/api/queries/rerank_fragments.rs

@@ -0,0 +1,68 @@
+use crate::{
+    error::UserFacingError, full_source_code_embedding::ContentHash,
+    request_context::RequestContext, schema,
+};
+
+#[derive(cynic::QueryVariables, Debug)]
+pub struct RerankFragmentsVariables {
+    pub fragments: Vec<RerankFragmentInput>,
+    pub query: String,
+    pub request_context: RequestContext,
+}
+
+#[derive(cynic::InputObject, Debug)]
+pub struct RerankFragmentInput {
+    pub content: String,
+    pub content_hash: ContentHash,
+    pub location: FragmentLocationInput,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+#[cynic(graphql_type = "RootQuery", variables = "RerankFragmentsVariables")]
+pub struct RerankFragments {
+    #[arguments(input: { fragments: $fragments, query: $query }, requestContext: $request_context)]
+    pub rerank_fragments: RerankFragmentsResult,
+}
+crate::client::define_operation! {
+    rerank_fragments(RerankFragmentsVariables) -> RerankFragments;
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+pub struct RerankFragmentsError {
+    pub error: String,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+pub struct RerankFragmentsOutput {
+    pub ranked_fragments: Vec<RerankFragment>,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+pub struct RerankFragment {
+    pub content: String,
+    pub content_hash: ContentHash,
+    pub location: FragmentLocation,
+}
+
+#[derive(cynic::QueryFragment, Debug)]
+pub struct FragmentLocation {
+    pub byte_end: i32,
+    pub byte_start: i32,
+    pub file_path: String,
+}
+
+#[derive(cynic::InlineFragments, Debug)]
+pub enum RerankFragmentsResult {
+    RerankFragmentsOutput(RerankFragmentsOutput),
+    RerankFragmentsError(RerankFragmentsError),
+    UserFacingError(UserFacingError),
+    #[cynic(fallback)]
+    Unknown,
+}
+
+#[derive(cynic::InputObject, Debug)]
+pub struct FragmentLocationInput {
+    pub byte_end: i32,
+    pub byte_start: i32,
+    pub file_path: String,
+}