nominal_api/proto/
nominal.ai.v1.rs

1// This file is @generated by prost-build.
2#[derive(Clone, Copy, PartialEq, ::prost::Message)]
3pub struct GetProviderStatusRequest {}
4#[derive(Clone, Copy, PartialEq, ::prost::Message)]
5pub struct GetProviderStatusResponse {
6    /// Timestamp when the last status was determined
7    #[prost(message, optional, tag = "1")]
8    pub timestamp: ::core::option::Option<
9        super::super::super::google::protobuf::Timestamp,
10    >,
11    /// Status of the most recent health check probe
12    #[prost(message, optional, tag = "2")]
13    pub last_status: ::core::option::Option<ProviderStatus>,
14    /// Aggregated status over the last 30 minutes (DEGRADED if any check failed or exceeded thresholds)
15    /// Deprecated: Use aggregated_status instead. This field is kept for backward compatibility.
16    #[deprecated]
17    #[prost(message, optional, tag = "3")]
18    pub aggregated_status_over_last_30m: ::core::option::Option<ProviderStatus>,
19    /// Aggregated status over the last iterations, configurable in the BE (DEGRADED if any check failed or exceeded thresholds)
20    #[prost(message, optional, tag = "4")]
21    pub aggregated_status: ::core::option::Option<ProviderStatus>,
22}
23#[derive(Clone, Copy, PartialEq, ::prost::Message)]
24pub struct ProviderStatus {
25    #[prost(oneof = "provider_status::Status", tags = "1, 2")]
26    pub status: ::core::option::Option<provider_status::Status>,
27}
28/// Nested message and enum types in `ProviderStatus`.
29pub mod provider_status {
30    #[derive(Clone, Copy, PartialEq, ::prost::Oneof)]
31    pub enum Status {
32        #[prost(message, tag = "1")]
33        Healthy(super::Healthy),
34        #[prost(message, tag = "2")]
35        Degraded(super::Degraded),
36    }
37}
38#[derive(Clone, Copy, PartialEq, ::prost::Message)]
39pub struct Healthy {}
40#[derive(Clone, Copy, PartialEq, ::prost::Message)]
41pub struct Degraded {
42    #[prost(enumeration = "DegradationReason", tag = "1")]
43    pub reason: i32,
44}
45#[derive(Clone, Copy, PartialEq, ::prost::Message)]
46pub struct ProviderMetrics {
47    #[prost(int32, tag = "1")]
48    pub time_to_first_token_ms: i32,
49    #[prost(int32, tag = "2")]
50    pub total_time_ms: i32,
51}
52#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, PartialOrd, Ord, ::prost::Enumeration)]
53#[repr(i32)]
54pub enum DegradationReason {
55    Unspecified = 0,
56    HighLatency = 1,
57    Failures = 2,
58    HighLatencyAndFailures = 3,
59}
60impl DegradationReason {
61    /// String value of the enum field names used in the ProtoBuf definition.
62    ///
63    /// The values are not transformed in any way and thus are considered stable
64    /// (if the ProtoBuf definition does not change) and safe for programmatic use.
65    pub fn as_str_name(&self) -> &'static str {
66        match self {
67            Self::Unspecified => "DEGRADATION_REASON_UNSPECIFIED",
68            Self::HighLatency => "DEGRADATION_REASON_HIGH_LATENCY",
69            Self::Failures => "DEGRADATION_REASON_FAILURES",
70            Self::HighLatencyAndFailures => {
71                "DEGRADATION_REASON_HIGH_LATENCY_AND_FAILURES"
72            }
73        }
74    }
75    /// Creates an enum from field names used in the ProtoBuf definition.
76    pub fn from_str_name(value: &str) -> ::core::option::Option<Self> {
77        match value {
78            "DEGRADATION_REASON_UNSPECIFIED" => Some(Self::Unspecified),
79            "DEGRADATION_REASON_HIGH_LATENCY" => Some(Self::HighLatency),
80            "DEGRADATION_REASON_FAILURES" => Some(Self::Failures),
81            "DEGRADATION_REASON_HIGH_LATENCY_AND_FAILURES" => {
82                Some(Self::HighLatencyAndFailures)
83            }
84            _ => None,
85        }
86    }
87}
88/// Generated client implementations.
89pub mod model_provider_health_service_client {
90    #![allow(
91        unused_variables,
92        dead_code,
93        missing_docs,
94        clippy::wildcard_imports,
95        clippy::let_unit_value,
96    )]
97    use tonic::codegen::*;
98    use tonic::codegen::http::Uri;
99    /// ModelProviderHealthService monitors the health and performance of the backing LLM model provider.
100    /// It runs lightweight health checks every 5 minutes to measure provider responsiveness and reliability,
101    /// independent of the complexity of user prompts.
102    #[derive(Debug, Clone)]
103    pub struct ModelProviderHealthServiceClient<T> {
104        inner: tonic::client::Grpc<T>,
105    }
106    impl ModelProviderHealthServiceClient<tonic::transport::Channel> {
107        /// Attempt to create a new client by connecting to a given endpoint.
108        pub async fn connect<D>(dst: D) -> Result<Self, tonic::transport::Error>
109        where
110            D: TryInto<tonic::transport::Endpoint>,
111            D::Error: Into<StdError>,
112        {
113            let conn = tonic::transport::Endpoint::new(dst)?.connect().await?;
114            Ok(Self::new(conn))
115        }
116    }
117    impl<T> ModelProviderHealthServiceClient<T>
118    where
119        T: tonic::client::GrpcService<tonic::body::Body>,
120        T::Error: Into<StdError>,
121        T::ResponseBody: Body<Data = Bytes> + std::marker::Send + 'static,
122        <T::ResponseBody as Body>::Error: Into<StdError> + std::marker::Send,
123    {
124        pub fn new(inner: T) -> Self {
125            let inner = tonic::client::Grpc::new(inner);
126            Self { inner }
127        }
128        pub fn with_origin(inner: T, origin: Uri) -> Self {
129            let inner = tonic::client::Grpc::with_origin(inner, origin);
130            Self { inner }
131        }
132        pub fn with_interceptor<F>(
133            inner: T,
134            interceptor: F,
135        ) -> ModelProviderHealthServiceClient<InterceptedService<T, F>>
136        where
137            F: tonic::service::Interceptor,
138            T::ResponseBody: Default,
139            T: tonic::codegen::Service<
140                http::Request<tonic::body::Body>,
141                Response = http::Response<
142                    <T as tonic::client::GrpcService<tonic::body::Body>>::ResponseBody,
143                >,
144            >,
145            <T as tonic::codegen::Service<
146                http::Request<tonic::body::Body>,
147            >>::Error: Into<StdError> + std::marker::Send + std::marker::Sync,
148        {
149            ModelProviderHealthServiceClient::new(
150                InterceptedService::new(inner, interceptor),
151            )
152        }
153        /// Compress requests with the given encoding.
154        ///
155        /// This requires the server to support it otherwise it might respond with an
156        /// error.
157        #[must_use]
158        pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
159            self.inner = self.inner.send_compressed(encoding);
160            self
161        }
162        /// Enable decompressing responses.
163        #[must_use]
164        pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
165            self.inner = self.inner.accept_compressed(encoding);
166            self
167        }
168        /// Limits the maximum size of a decoded message.
169        ///
170        /// Default: `4MB`
171        #[must_use]
172        pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
173            self.inner = self.inner.max_decoding_message_size(limit);
174            self
175        }
176        /// Limits the maximum size of an encoded message.
177        ///
178        /// Default: `usize::MAX`
179        #[must_use]
180        pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
181            self.inner = self.inner.max_encoding_message_size(limit);
182            self
183        }
184        /// GetProviderStatus returns the current health status of the model provider based on recent health checks.
185        /// The status is HEALTHY if all checks in the last 30 minutes passed latency thresholds,
186        /// or DEGRADED if any checks exceeded latency thresholds or failed entirely.
187        pub async fn get_provider_status(
188            &mut self,
189            request: impl tonic::IntoRequest<super::GetProviderStatusRequest>,
190        ) -> std::result::Result<
191            tonic::Response<super::GetProviderStatusResponse>,
192            tonic::Status,
193        > {
194            self.inner
195                .ready()
196                .await
197                .map_err(|e| {
198                    tonic::Status::unknown(
199                        format!("Service was not ready: {}", e.into()),
200                    )
201                })?;
202            let codec = tonic::codec::ProstCodec::default();
203            let path = http::uri::PathAndQuery::from_static(
204                "/nominal.ai.v1.ModelProviderHealthService/GetProviderStatus",
205            );
206            let mut req = request.into_request();
207            req.extensions_mut()
208                .insert(
209                    GrpcMethod::new(
210                        "nominal.ai.v1.ModelProviderHealthService",
211                        "GetProviderStatus",
212                    ),
213                );
214            self.inner.unary(req, path, codec).await
215        }
216    }
217}
218/// CreateOrUpdateKnowledgeBaseRequest is a request to create a knowledge from an attachment in the attachment's
219/// workspace. Subsequent calls for the same attachment will overwrite the existing knowledge base.
220#[derive(Clone, PartialEq, ::prost::Message)]
221pub struct CreateOrUpdateKnowledgeBaseRequest {
222    #[prost(string, tag = "1")]
223    pub attachment_rid: ::prost::alloc::string::String,
224    /// summary of the knowledge base, will be used by the LLM to decide when to use it
225    #[prost(string, tag = "2")]
226    pub summary_description: ::prost::alloc::string::String,
227    #[prost(enumeration = "KnowledgeBaseType", optional, tag = "3")]
228    pub r#type: ::core::option::Option<i32>,
229}
230/// CreateOrUpdateKnowledgeBaseResponse is the response to creating/updating a knowledge base
231#[derive(Clone, PartialEq, ::prost::Message)]
232pub struct CreateOrUpdateKnowledgeBaseResponse {
233    #[prost(string, tag = "1")]
234    pub knowledge_base_rid: ::prost::alloc::string::String,
235}
236/// KnowledgeBase represents a knowledge base entry
237#[derive(Clone, PartialEq, ::prost::Message)]
238pub struct KnowledgeBase {
239    #[prost(string, tag = "1")]
240    pub knowledge_base_rid: ::prost::alloc::string::String,
241    #[prost(string, tag = "2")]
242    pub attachment_rid: ::prost::alloc::string::String,
243    #[prost(string, tag = "3")]
244    pub workspace_rid: ::prost::alloc::string::String,
245    #[prost(string, tag = "4")]
246    pub summary_description: ::prost::alloc::string::String,
247    #[prost(enumeration = "KnowledgeBaseType", tag = "5")]
248    pub r#type: i32,
249    #[prost(int32, tag = "6")]
250    pub version: i32,
251}
252#[derive(Clone, PartialEq, ::prost::Message)]
253pub struct ListRequest {
254    #[prost(string, tag = "1")]
255    pub workspace_rid: ::prost::alloc::string::String,
256}
257#[derive(Clone, PartialEq, ::prost::Message)]
258pub struct ListResponse {
259    #[prost(message, repeated, tag = "1")]
260    pub knowledge_bases: ::prost::alloc::vec::Vec<KnowledgeBase>,
261}
262#[derive(Clone, PartialEq, ::prost::Message)]
263pub struct DeleteRequest {
264    #[prost(string, tag = "1")]
265    pub knowledge_base_rid: ::prost::alloc::string::String,
266}
267#[derive(Clone, Copy, PartialEq, ::prost::Message)]
268pub struct DeleteResponse {
269    #[prost(bool, tag = "1")]
270    pub success: bool,
271}
272#[derive(Clone, PartialEq, ::prost::Message)]
273pub struct GetBatchRequest {
274    #[prost(string, repeated, tag = "1")]
275    pub knowledge_base_rids: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
276}
277#[derive(Clone, PartialEq, ::prost::Message)]
278pub struct GetBatchResponse {
279    #[prost(message, repeated, tag = "1")]
280    pub knowledge_bases: ::prost::alloc::vec::Vec<KnowledgeBase>,
281}
282/// generate summary description is intentionally going to return the generated description to the frontend
283/// rather than storing it in the knowledge base directly because the description needs to be accepted by the user
284#[derive(Clone, PartialEq, ::prost::Message)]
285pub struct GenerateSummaryDescriptionRequest {
286    #[prost(string, tag = "1")]
287    pub attachment_rid: ::prost::alloc::string::String,
288}
289#[derive(Clone, PartialEq, ::prost::Message)]
290pub struct GenerateSummaryDescriptionResponse {
291    #[prost(string, tag = "1")]
292    pub summary_description: ::prost::alloc::string::String,
293}
294/// KnowledgeBaseType defines the types of knowledge base
295#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, PartialOrd, Ord, ::prost::Enumeration)]
296#[repr(i32)]
297pub enum KnowledgeBaseType {
298    /// defaults to PROMPT
299    Unspecified = 0,
300    /// knowledge base gets added directly to prompt (needs to be small enough!)
301    Prompt = 1,
302    /// knowledge base gets used via vector search on embeddings
303    Embedding = 2,
304}
305impl KnowledgeBaseType {
306    /// String value of the enum field names used in the ProtoBuf definition.
307    ///
308    /// The values are not transformed in any way and thus are considered stable
309    /// (if the ProtoBuf definition does not change) and safe for programmatic use.
310    pub fn as_str_name(&self) -> &'static str {
311        match self {
312            Self::Unspecified => "KNOWLEDGE_BASE_TYPE_UNSPECIFIED",
313            Self::Prompt => "KNOWLEDGE_BASE_TYPE_PROMPT",
314            Self::Embedding => "KNOWLEDGE_BASE_TYPE_EMBEDDING",
315        }
316    }
317    /// Creates an enum from field names used in the ProtoBuf definition.
318    pub fn from_str_name(value: &str) -> ::core::option::Option<Self> {
319        match value {
320            "KNOWLEDGE_BASE_TYPE_UNSPECIFIED" => Some(Self::Unspecified),
321            "KNOWLEDGE_BASE_TYPE_PROMPT" => Some(Self::Prompt),
322            "KNOWLEDGE_BASE_TYPE_EMBEDDING" => Some(Self::Embedding),
323            _ => None,
324        }
325    }
326}
327/// Generated client implementations.
328pub mod knowledge_base_service_client {
329    #![allow(
330        unused_variables,
331        dead_code,
332        missing_docs,
333        clippy::wildcard_imports,
334        clippy::let_unit_value,
335    )]
336    use tonic::codegen::*;
337    use tonic::codegen::http::Uri;
338    /// KnowledgeBaseService provides AI-powered knowledge base management
339    #[derive(Debug, Clone)]
340    pub struct KnowledgeBaseServiceClient<T> {
341        inner: tonic::client::Grpc<T>,
342    }
343    impl KnowledgeBaseServiceClient<tonic::transport::Channel> {
344        /// Attempt to create a new client by connecting to a given endpoint.
345        pub async fn connect<D>(dst: D) -> Result<Self, tonic::transport::Error>
346        where
347            D: TryInto<tonic::transport::Endpoint>,
348            D::Error: Into<StdError>,
349        {
350            let conn = tonic::transport::Endpoint::new(dst)?.connect().await?;
351            Ok(Self::new(conn))
352        }
353    }
354    impl<T> KnowledgeBaseServiceClient<T>
355    where
356        T: tonic::client::GrpcService<tonic::body::Body>,
357        T::Error: Into<StdError>,
358        T::ResponseBody: Body<Data = Bytes> + std::marker::Send + 'static,
359        <T::ResponseBody as Body>::Error: Into<StdError> + std::marker::Send,
360    {
361        pub fn new(inner: T) -> Self {
362            let inner = tonic::client::Grpc::new(inner);
363            Self { inner }
364        }
365        pub fn with_origin(inner: T, origin: Uri) -> Self {
366            let inner = tonic::client::Grpc::with_origin(inner, origin);
367            Self { inner }
368        }
369        pub fn with_interceptor<F>(
370            inner: T,
371            interceptor: F,
372        ) -> KnowledgeBaseServiceClient<InterceptedService<T, F>>
373        where
374            F: tonic::service::Interceptor,
375            T::ResponseBody: Default,
376            T: tonic::codegen::Service<
377                http::Request<tonic::body::Body>,
378                Response = http::Response<
379                    <T as tonic::client::GrpcService<tonic::body::Body>>::ResponseBody,
380                >,
381            >,
382            <T as tonic::codegen::Service<
383                http::Request<tonic::body::Body>,
384            >>::Error: Into<StdError> + std::marker::Send + std::marker::Sync,
385        {
386            KnowledgeBaseServiceClient::new(InterceptedService::new(inner, interceptor))
387        }
388        /// Compress requests with the given encoding.
389        ///
390        /// This requires the server to support it otherwise it might respond with an
391        /// error.
392        #[must_use]
393        pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
394            self.inner = self.inner.send_compressed(encoding);
395            self
396        }
397        /// Enable decompressing responses.
398        #[must_use]
399        pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
400            self.inner = self.inner.accept_compressed(encoding);
401            self
402        }
403        /// Limits the maximum size of a decoded message.
404        ///
405        /// Default: `4MB`
406        #[must_use]
407        pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
408            self.inner = self.inner.max_decoding_message_size(limit);
409            self
410        }
411        /// Limits the maximum size of an encoded message.
412        ///
413        /// Default: `usize::MAX`
414        #[must_use]
415        pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
416            self.inner = self.inner.max_encoding_message_size(limit);
417            self
418        }
419        /// CreateOrUpdateKnowledgeBase creates a knowledge base in the workspace
420        pub async fn create_or_update_knowledge_base(
421            &mut self,
422            request: impl tonic::IntoRequest<super::CreateOrUpdateKnowledgeBaseRequest>,
423        ) -> std::result::Result<
424            tonic::Response<super::CreateOrUpdateKnowledgeBaseResponse>,
425            tonic::Status,
426        > {
427            self.inner
428                .ready()
429                .await
430                .map_err(|e| {
431                    tonic::Status::unknown(
432                        format!("Service was not ready: {}", e.into()),
433                    )
434                })?;
435            let codec = tonic::codec::ProstCodec::default();
436            let path = http::uri::PathAndQuery::from_static(
437                "/nominal.ai.v1.KnowledgeBaseService/CreateOrUpdateKnowledgeBase",
438            );
439            let mut req = request.into_request();
440            req.extensions_mut()
441                .insert(
442                    GrpcMethod::new(
443                        "nominal.ai.v1.KnowledgeBaseService",
444                        "CreateOrUpdateKnowledgeBase",
445                    ),
446                );
447            self.inner.unary(req, path, codec).await
448        }
449        /// List returns all knowledge bases in the specified workspace
450        pub async fn list(
451            &mut self,
452            request: impl tonic::IntoRequest<super::ListRequest>,
453        ) -> std::result::Result<tonic::Response<super::ListResponse>, tonic::Status> {
454            self.inner
455                .ready()
456                .await
457                .map_err(|e| {
458                    tonic::Status::unknown(
459                        format!("Service was not ready: {}", e.into()),
460                    )
461                })?;
462            let codec = tonic::codec::ProstCodec::default();
463            let path = http::uri::PathAndQuery::from_static(
464                "/nominal.ai.v1.KnowledgeBaseService/List",
465            );
466            let mut req = request.into_request();
467            req.extensions_mut()
468                .insert(GrpcMethod::new("nominal.ai.v1.KnowledgeBaseService", "List"));
469            self.inner.unary(req, path, codec).await
470        }
471        /// Delete removes a knowledge base by its RID
472        pub async fn delete(
473            &mut self,
474            request: impl tonic::IntoRequest<super::DeleteRequest>,
475        ) -> std::result::Result<tonic::Response<super::DeleteResponse>, tonic::Status> {
476            self.inner
477                .ready()
478                .await
479                .map_err(|e| {
480                    tonic::Status::unknown(
481                        format!("Service was not ready: {}", e.into()),
482                    )
483                })?;
484            let codec = tonic::codec::ProstCodec::default();
485            let path = http::uri::PathAndQuery::from_static(
486                "/nominal.ai.v1.KnowledgeBaseService/Delete",
487            );
488            let mut req = request.into_request();
489            req.extensions_mut()
490                .insert(GrpcMethod::new("nominal.ai.v1.KnowledgeBaseService", "Delete"));
491            self.inner.unary(req, path, codec).await
492        }
493        /// GetBatch retrieves multiple knowledge bases by their RIDs
494        pub async fn get_batch(
495            &mut self,
496            request: impl tonic::IntoRequest<super::GetBatchRequest>,
497        ) -> std::result::Result<
498            tonic::Response<super::GetBatchResponse>,
499            tonic::Status,
500        > {
501            self.inner
502                .ready()
503                .await
504                .map_err(|e| {
505                    tonic::Status::unknown(
506                        format!("Service was not ready: {}", e.into()),
507                    )
508                })?;
509            let codec = tonic::codec::ProstCodec::default();
510            let path = http::uri::PathAndQuery::from_static(
511                "/nominal.ai.v1.KnowledgeBaseService/GetBatch",
512            );
513            let mut req = request.into_request();
514            req.extensions_mut()
515                .insert(
516                    GrpcMethod::new("nominal.ai.v1.KnowledgeBaseService", "GetBatch"),
517                );
518            self.inner.unary(req, path, codec).await
519        }
520        /// GenerateSummaryDescription generates a summary description for an attachment rid
521        pub async fn generate_summary_description(
522            &mut self,
523            request: impl tonic::IntoRequest<super::GenerateSummaryDescriptionRequest>,
524        ) -> std::result::Result<
525            tonic::Response<super::GenerateSummaryDescriptionResponse>,
526            tonic::Status,
527        > {
528            self.inner
529                .ready()
530                .await
531                .map_err(|e| {
532                    tonic::Status::unknown(
533                        format!("Service was not ready: {}", e.into()),
534                    )
535                })?;
536            let codec = tonic::codec::ProstCodec::default();
537            let path = http::uri::PathAndQuery::from_static(
538                "/nominal.ai.v1.KnowledgeBaseService/GenerateSummaryDescription",
539            );
540            let mut req = request.into_request();
541            req.extensions_mut()
542                .insert(
543                    GrpcMethod::new(
544                        "nominal.ai.v1.KnowledgeBaseService",
545                        "GenerateSummaryDescription",
546                    ),
547                );
548            self.inner.unary(req, path, codec).await
549        }
550    }
551}
552#[derive(Clone, PartialEq, ::prost::Message)]
553pub struct GetSnapshotRidByUserMessageIdRequest {
554    #[prost(string, tag = "1")]
555    pub conversation_rid: ::prost::alloc::string::String,
556    #[prost(string, tag = "2")]
557    pub message_id: ::prost::alloc::string::String,
558}
559/// Will return an empty response body in the case where the message id exists, but there is no associated snapshot
560/// This occurs in the instance where a message was sent in a non-workbook context
561#[derive(Clone, PartialEq, ::prost::Message)]
562pub struct GetSnapshotRidByUserMessageIdResponse {
563    #[prost(string, optional, tag = "1")]
564    pub snapshot_rid: ::core::option::Option<::prost::alloc::string::String>,
565}
566/// ReadOnlyMode configures read-only mode where edit tools are shadowed/disabled
567#[derive(Clone, Copy, PartialEq, ::prost::Message)]
568pub struct ReadOnlyMode {}
569/// EditMode configures edit mode where all tools are available
570#[derive(Clone, Copy, PartialEq, ::prost::Message)]
571pub struct EditMode {
572    /// when set to true, we auto accept edits for any tools typically requiring approval
573    #[prost(bool, optional, tag = "1")]
574    pub auto_accept: ::core::option::Option<bool>,
575}
576/// ConversationMode specifies the mode of the conversation
577#[derive(Clone, Copy, PartialEq, ::prost::Message)]
578pub struct ConversationMode {
579    #[prost(oneof = "conversation_mode::Mode", tags = "1, 2")]
580    pub mode: ::core::option::Option<conversation_mode::Mode>,
581}
582/// Nested message and enum types in `ConversationMode`.
583pub mod conversation_mode {
584    #[derive(Clone, Copy, PartialEq, ::prost::Oneof)]
585    pub enum Mode {
586        #[prost(message, tag = "1")]
587        ReadOnly(super::ReadOnlyMode),
588        #[prost(message, tag = "2")]
589        Edit(super::EditMode),
590    }
591}
592/// When the agent makes a request to use a tool, the user responds
593/// with one of these for every request - mapping a tool id to its approval/denial
594#[derive(Clone, PartialEq, ::prost::Message)]
595pub struct ToolApprovalResult {
596    /// identifies the tool call
597    #[prost(string, tag = "1")]
598    pub tool_call_id: ::prost::alloc::string::String,
599    #[prost(oneof = "tool_approval_result::Response", tags = "2, 3")]
600    pub response: ::core::option::Option<tool_approval_result::Response>,
601}
602/// Nested message and enum types in `ToolApprovalResult`.
603pub mod tool_approval_result {
604    #[derive(Clone, PartialEq, ::prost::Oneof)]
605    pub enum Response {
606        #[prost(message, tag = "2")]
607        Approved(super::ToolApprovedResponse),
608        #[prost(message, tag = "3")]
609        Denied(super::ToolDeniedResponse),
610    }
611}
612#[derive(Clone, PartialEq, ::prost::Message)]
613pub struct ToolApprovedResponse {
614    /// json string representation of the override argument if the user
615    /// needs it to be changed in some way
616    #[prost(string, optional, tag = "1")]
617    pub override_args: ::core::option::Option<::prost::alloc::string::String>,
618}
619#[derive(Clone, PartialEq, ::prost::Message)]
620pub struct ToolDeniedResponse {
621    #[prost(string, tag = "2")]
622    pub denial_reason: ::prost::alloc::string::String,
623}
624/// StreamChatRequest is a request to stream chat messages for AI agent.
625/// Either `message` (user prompt), `tool_approval` (tool approval), or neither (retry of a conversation
626/// that ended in a tool call request) can be present, but not both `message` and `tool_approval`.
627#[derive(Clone, PartialEq, ::prost::Message)]
628pub struct StreamChatRequest {
629    /// The conversation ID
630    #[prost(string, tag = "1")]
631    pub conversation_rid: ::prost::alloc::string::String,
632    /// The user message to append to the conversation (mutually exclusive with tool_approval)
633    #[prost(message, optional, tag = "2")]
634    pub message: ::core::option::Option<UserModelMessage>,
635    /// Optional: image files to provide to the agent
636    #[prost(message, repeated, tag = "3")]
637    pub images: ::prost::alloc::vec::Vec<ImagePart>,
638    /// Optional: tool approval result for when the LLM needs HITL for a tool call (mutually exclusive with message)
639    #[prost(message, optional, tag = "6")]
640    pub tool_approval: ::core::option::Option<ToolApprovalResult>,
641    /// Context-specific fields based on the oneofKind.
642    #[prost(oneof = "stream_chat_request::Context", tags = "4, 5")]
643    pub context: ::core::option::Option<stream_chat_request::Context>,
644}
645/// Nested message and enum types in `StreamChatRequest`.
646pub mod stream_chat_request {
647    /// Context-specific fields based on the oneofKind.
648    #[derive(Clone, PartialEq, ::prost::Oneof)]
649    pub enum Context {
650        #[prost(message, tag = "4")]
651        Workbook(super::WorkbookContext),
652        #[prost(message, tag = "5")]
653        Global(super::GlobalContext),
654    }
655}
656/// WorkbookContext contains workbook-specific context fields
657#[derive(Clone, PartialEq, ::prost::Message)]
658pub struct WorkbookContext {
659    /// RID of the workbook to use for context
660    #[prost(string, tag = "1")]
661    pub workbook_rid: ::prost::alloc::string::String,
662    /// The user's presence in the workbook
663    #[prost(message, optional, tag = "2")]
664    pub user_presence: ::core::option::Option<WorkbookUserPresence>,
665}
666/// DefaultContext (no context)
667#[derive(Clone, Copy, PartialEq, ::prost::Message)]
668pub struct GlobalContext {}
669/// WorkbookUserPresence contains the user's presence in the workbook
670/// which is used to describe what the user is viewing at the time of the message.
671#[derive(Clone, Copy, PartialEq, ::prost::Message)]
672pub struct WorkbookUserPresence {
673    #[prost(int32, tag = "1")]
674    pub tab_index: i32,
675    #[prost(message, optional, tag = "2")]
676    pub range: ::core::option::Option<TimeRange>,
677}
678/// CreateConversation request will create a new conversation thread
679/// if old conversation id is not set, a brand new, clear chat is created
680/// If old conversation id is set without a previous message id, the full conversation thread will be copied
681/// if previous message id is set with a previous message id, the conversation thread up until that message will be copied
682/// the above case is useful for branching a conversation into a new thread
683#[derive(Clone, PartialEq, ::prost::Message)]
684pub struct CreateConversationRequest {
685    #[prost(string, tag = "1")]
686    pub title: ::prost::alloc::string::String,
687    #[prost(string, tag = "2")]
688    pub workspace_rid: ::prost::alloc::string::String,
689    #[prost(string, optional, tag = "3")]
690    pub old_conversation_rid: ::core::option::Option<::prost::alloc::string::String>,
691    #[prost(string, optional, tag = "4")]
692    pub previous_message_id: ::core::option::Option<::prost::alloc::string::String>,
693    #[prost(message, optional, tag = "5")]
694    pub conversation_mode: ::core::option::Option<ConversationMode>,
695}
696/// CreateConversationResponse will return the conversation id for the new conversation
697#[derive(Clone, PartialEq, ::prost::Message)]
698pub struct CreateConversationResponse {
699    #[prost(string, tag = "1")]
700    pub new_conversation_rid: ::prost::alloc::string::String,
701}
702/// Updates the fields if specified (optional means no change for that field)
703#[derive(Clone, PartialEq, ::prost::Message)]
704pub struct UpdateConversationMetadataRequest {
705    #[prost(string, optional, tag = "1")]
706    pub title: ::core::option::Option<::prost::alloc::string::String>,
707    #[prost(string, tag = "2")]
708    pub conversation_rid: ::prost::alloc::string::String,
709    #[prost(message, optional, tag = "3")]
710    pub conversation_mode: ::core::option::Option<ConversationMode>,
711}
712#[derive(Clone, Copy, PartialEq, ::prost::Message)]
713pub struct UpdateConversationMetadataResponse {}
714#[derive(Clone, PartialEq, ::prost::Message)]
715pub struct DeleteConversationRequest {
716    #[prost(string, tag = "1")]
717    pub conversation_rid: ::prost::alloc::string::String,
718}
719#[derive(Clone, Copy, PartialEq, ::prost::Message)]
720pub struct DeleteConversationResponse {}
721/// a GetConversationRequest allows you to retrieve a subset of messages from a conversation thread represented
722/// by provided rid. To start from a particular message - you can also provide a message id.
723#[derive(Clone, PartialEq, ::prost::Message)]
724pub struct GetConversationRequest {
725    #[prost(string, tag = "1")]
726    pub conversation_rid: ::prost::alloc::string::String,
727    #[prost(string, optional, tag = "2")]
728    pub page_start_message_id: ::core::option::Option<::prost::alloc::string::String>,
729    #[prost(int32, optional, tag = "3")]
730    pub max_message_count: ::core::option::Option<i32>,
731}
732/// Model message with id allows you to identify the message ID of a given message
733#[derive(Clone, PartialEq, ::prost::Message)]
734pub struct ModelMessageWithId {
735    #[prost(string, tag = "3")]
736    pub message_id: ::prost::alloc::string::String,
737    /// WB agent user messages can have snapshot rids associated with them
738    #[prost(string, optional, tag = "4")]
739    pub snapshot_rid: ::core::option::Option<::prost::alloc::string::String>,
740    #[prost(message, optional, tag = "5")]
741    pub tool_approval_request: ::core::option::Option<ToolCallDescription>,
742    #[prost(oneof = "model_message_with_id::Content", tags = "1, 2")]
743    pub content: ::core::option::Option<model_message_with_id::Content>,
744}
745/// Nested message and enum types in `ModelMessageWithId`.
746pub mod model_message_with_id {
747    #[derive(Clone, PartialEq, ::prost::Oneof)]
748    pub enum Content {
749        #[prost(message, tag = "1")]
750        Message(super::ModelMessage),
751        #[prost(message, tag = "2")]
752        ToolAction(super::ToolAction),
753    }
754}
755#[derive(Clone, PartialEq, ::prost::Message)]
756pub struct GetConversationResponse {
757    #[prost(message, repeated, tag = "1")]
758    pub ordered_messages: ::prost::alloc::vec::Vec<ModelMessageWithId>,
759    #[prost(message, optional, tag = "2")]
760    pub conversation_metadata: ::core::option::Option<ConversationMetadata>,
761}
762/// Will generate all conversation threads that this user has in this workspace
763#[derive(Clone, PartialEq, ::prost::Message)]
764pub struct ListConversationsRequest {
765    #[prost(string, tag = "1")]
766    pub workspace_rid: ::prost::alloc::string::String,
767    #[prost(string, optional, tag = "2")]
768    pub next_page_token: ::core::option::Option<::prost::alloc::string::String>,
769    #[prost(int32, optional, tag = "3")]
770    pub page_size: ::core::option::Option<i32>,
771}
772#[derive(Clone, PartialEq, ::prost::Message)]
773pub struct ConversationMetadata {
774    #[prost(string, tag = "1")]
775    pub conversation_rid: ::prost::alloc::string::String,
776    #[prost(string, tag = "2")]
777    pub title: ::prost::alloc::string::String,
778    #[prost(message, optional, tag = "3")]
779    pub created_at: ::core::option::Option<
780        super::super::super::google::protobuf::Timestamp,
781    >,
782    #[prost(message, optional, tag = "4")]
783    pub last_updated_at: ::core::option::Option<
784        super::super::super::google::protobuf::Timestamp,
785    >,
786    #[prost(message, optional, tag = "5")]
787    pub mode: ::core::option::Option<ConversationMode>,
788}
789/// ListConversationsResponse is a list of conversations that can be used in a call to GetConversationRequest
790/// to get a full conversation from storage. These are ordered by creation time.
791#[derive(Clone, PartialEq, ::prost::Message)]
792pub struct ListConversationsResponse {
793    #[prost(message, repeated, tag = "1")]
794    pub conversations: ::prost::alloc::vec::Vec<ConversationMetadata>,
795    #[prost(string, optional, tag = "2")]
796    pub next_page_token: ::core::option::Option<::prost::alloc::string::String>,
797}
798#[derive(Clone, Copy, PartialEq, ::prost::Message)]
799pub struct TimeRange {
800    #[prost(message, optional, tag = "1")]
801    pub range_start: ::core::option::Option<Timestamp>,
802    #[prost(message, optional, tag = "2")]
803    pub range_end: ::core::option::Option<Timestamp>,
804}
805#[derive(Clone, Copy, PartialEq, ::prost::Message)]
806pub struct Timestamp {
807    #[prost(int32, tag = "1")]
808    pub seconds: i32,
809    #[prost(int32, tag = "2")]
810    pub nanoseconds: i32,
811}
812/// ModelMessage is a discriminated union of system, user, assistant, and tool messages.
813/// Each message type has its own structure and content.
814#[derive(Clone, PartialEq, ::prost::Message)]
815pub struct ModelMessage {
816    #[prost(oneof = "model_message::Kind", tags = "1, 2")]
817    pub kind: ::core::option::Option<model_message::Kind>,
818}
819/// Nested message and enum types in `ModelMessage`.
820pub mod model_message {
821    #[derive(Clone, PartialEq, ::prost::Oneof)]
822    pub enum Kind {
823        #[prost(message, tag = "1")]
824        User(super::UserModelMessage),
825        #[prost(message, tag = "2")]
826        Assistant(super::AssistantModelMessage),
827    }
828}
829/// A user message containing text
830#[derive(Clone, PartialEq, ::prost::Message)]
831pub struct UserModelMessage {
832    #[prost(message, repeated, tag = "1")]
833    pub text: ::prost::alloc::vec::Vec<UserContentPart>,
834}
835/// An assistant message containing text
836#[derive(Clone, PartialEq, ::prost::Message)]
837pub struct AssistantModelMessage {
838    #[prost(message, repeated, tag = "1")]
839    pub content_parts: ::prost::alloc::vec::Vec<AssistantContentPart>,
840}
841#[derive(Clone, PartialEq, ::prost::Message)]
842pub struct UserContentPart {
843    #[prost(oneof = "user_content_part::Part", tags = "1")]
844    pub part: ::core::option::Option<user_content_part::Part>,
845}
846/// Nested message and enum types in `UserContentPart`.
847pub mod user_content_part {
848    #[derive(Clone, PartialEq, ::prost::Oneof)]
849    pub enum Part {
850        #[prost(message, tag = "1")]
851        Text(super::TextPart),
852    }
853}
854/// Content part for assistant messages: can be text, reasoning, or mutation.
855#[derive(Clone, PartialEq, ::prost::Message)]
856pub struct AssistantContentPart {
857    #[prost(oneof = "assistant_content_part::Part", tags = "1, 2")]
858    pub part: ::core::option::Option<assistant_content_part::Part>,
859}
860/// Nested message and enum types in `AssistantContentPart`.
861pub mod assistant_content_part {
862    #[derive(Clone, PartialEq, ::prost::Oneof)]
863    pub enum Part {
864        #[prost(message, tag = "1")]
865        Text(super::TextPart),
866        #[prost(message, tag = "2")]
867        Reasoning(super::ReasoningPart),
868    }
869}
870/// Text part for user or assistant messages.
871#[derive(Clone, PartialEq, ::prost::Message)]
872pub struct TextPart {
873    #[prost(string, tag = "1")]
874    pub text: ::prost::alloc::string::String,
875}
876/// User-supplied image part.
877#[derive(Clone, PartialEq, ::prost::Message)]
878pub struct ImagePart {
879    /// The base64-encoded image data
880    #[prost(bytes = "vec", tag = "1")]
881    pub data: ::prost::alloc::vec::Vec<u8>,
882    /// The media type of the image (e.g. "image/png", "image/jpeg")
883    #[prost(string, optional, tag = "2")]
884    pub media_type: ::core::option::Option<::prost::alloc::string::String>,
885    /// Optional: the filename of the image
886    #[prost(string, optional, tag = "3")]
887    pub filename: ::core::option::Option<::prost::alloc::string::String>,
888}
889/// Reasoning part for assistant messages.
890#[derive(Clone, PartialEq, ::prost::Message)]
891pub struct ReasoningPart {
892    #[prost(string, tag = "1")]
893    pub reasoning: ::prost::alloc::string::String,
894}
895/// StreamChatResponse is a discriminated union response to a StreamChatRequest
896#[derive(Clone, PartialEq, ::prost::Message)]
897pub struct StreamChatResponse {
898    #[prost(
899        oneof = "stream_chat_response::Response",
900        tags = "1, 2, 3, 4, 5, 6, 7, 8, 10"
901    )]
902    pub response: ::core::option::Option<stream_chat_response::Response>,
903}
904/// Nested message and enum types in `StreamChatResponse`.
905pub mod stream_chat_response {
906    #[derive(Clone, PartialEq, ::prost::Oneof)]
907    pub enum Response {
908        #[prost(message, tag = "1")]
909        Finish(super::Finish),
910        #[prost(message, tag = "2")]
911        Error(super::Error),
912        #[prost(message, tag = "3")]
913        TextStart(super::TextStart),
914        #[prost(message, tag = "4")]
915        TextDelta(super::TextDelta),
916        #[prost(message, tag = "5")]
917        TextEnd(super::TextEnd),
918        #[prost(message, tag = "6")]
919        ReasoningStart(super::ReasoningStart),
920        #[prost(message, tag = "7")]
921        ReasoningDelta(super::ReasoningDelta),
922        #[prost(message, tag = "8")]
923        ReasoningEnd(super::ReasoningEnd),
924        #[prost(message, tag = "10")]
925        ToolAction(super::ToolAction),
926    }
927}
928#[derive(Clone, PartialEq, ::prost::Message)]
929pub struct ToolCallDescription {
930    #[prost(string, tag = "1")]
931    pub tool_call_id: ::prost::alloc::string::String,
932    #[prost(string, tag = "2")]
933    pub tool_name: ::prost::alloc::string::String,
934    /// string representation of the proposed tool args for display
935    #[prost(string, tag = "3")]
936    pub tool_args_json_string: ::prost::alloc::string::String,
937    /// used to conditionally render an approval button based the outcome of the tool call
938    #[prost(enumeration = "ToolCallStatus", tag = "4")]
939    pub status: i32,
940}
941/// Indicates the end of a chat session
942#[derive(Clone, PartialEq, ::prost::Message)]
943pub struct Finish {
944    /// The message ids in order of all generated messages for this agent run
945    /// These ids can be used to branch a message from that specific message
946    #[prost(string, repeated, tag = "1")]
947    pub ordered_message_ids: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
948    /// In the case that this is the first agent run in a conversation thread, we also
949    /// return the new conversation title generated
950    #[prost(string, optional, tag = "2")]
951    pub new_title: ::core::option::Option<::prost::alloc::string::String>,
952    /// In the case that the agent tries to use a tool that requires approval, it ends
953    /// its run with a ToolApprovalRequest - which must be followed with a ToolApprovalResponse
954    #[prost(message, optional, tag = "3")]
955    pub tool_approval_request: ::core::option::Option<ToolCallDescription>,
956}
957/// An error that occurred during the chat session
958#[derive(Clone, PartialEq, ::prost::Message)]
959pub struct Error {
960    #[prost(string, tag = "1")]
961    pub message: ::prost::alloc::string::String,
962}
963/// Indicates the start of a text message from the agent
964#[derive(Clone, PartialEq, ::prost::Message)]
965pub struct TextStart {
966    /// uniquely identifies the text message (e.g. uuid) so that the client can
967    /// merge parallel message streams (if it happens).
968    #[prost(string, tag = "1")]
969    pub id: ::prost::alloc::string::String,
970}
971/// A delta (continuation) of a text message from the agent
972#[derive(Clone, PartialEq, ::prost::Message)]
973pub struct TextDelta {
974    #[prost(string, tag = "1")]
975    pub id: ::prost::alloc::string::String,
976    /// The next chunk of text
977    #[prost(string, tag = "2")]
978    pub delta: ::prost::alloc::string::String,
979}
980/// Indicates the end of a text message from the agent
981#[derive(Clone, PartialEq, ::prost::Message)]
982pub struct TextEnd {
983    #[prost(string, tag = "1")]
984    pub id: ::prost::alloc::string::String,
985}
986/// Indicates the start of a reasoning message from the agent
987#[derive(Clone, PartialEq, ::prost::Message)]
988pub struct ReasoningStart {
989    #[prost(string, tag = "1")]
990    pub id: ::prost::alloc::string::String,
991}
992/// A delta (continuation) of a reasoning message from the agent
993#[derive(Clone, PartialEq, ::prost::Message)]
994pub struct ReasoningDelta {
995    #[prost(string, tag = "1")]
996    pub id: ::prost::alloc::string::String,
997    /// The next chunk of reasoning
998    #[prost(string, tag = "2")]
999    pub delta: ::prost::alloc::string::String,
1000}
1001/// Indicates the end of a reasoning message from the agent
1002#[derive(Clone, PartialEq, ::prost::Message)]
1003pub struct ReasoningEnd {
1004    #[prost(string, tag = "1")]
1005    pub id: ::prost::alloc::string::String,
1006}
1007/// this is a concise description of a tool call that the agent is making internally
1008/// without revealing too much detail about the tool call, it informs the user what the agent is doing
1009/// at a high level. the format is: `{tool_action_verb} {tool_target}` for example:
1010/// "Search channels for My Datasource"
1011#[derive(Clone, PartialEq, ::prost::Message)]
1012pub struct ToolAction {
1013    #[prost(string, tag = "1")]
1014    pub id: ::prost::alloc::string::String,
1015    /// "Thought", "Read", "Find", "Look-up", etc.
1016    #[prost(string, tag = "2")]
1017    pub tool_action_verb: ::prost::alloc::string::String,
1018    /// "workbook", "channel", "variable", "panel", etc.
1019    #[prost(string, optional, tag = "3")]
1020    pub tool_target: ::core::option::Option<::prost::alloc::string::String>,
1021}
1022#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, PartialOrd, Ord, ::prost::Enumeration)]
1023#[repr(i32)]
1024pub enum ToolCallStatus {
1025    Unspecified = 0,
1026    Approved = 1,
1027    Denied = 2,
1028    AwaitingApproval = 3,
1029}
1030impl ToolCallStatus {
1031    /// String value of the enum field names used in the ProtoBuf definition.
1032    ///
1033    /// The values are not transformed in any way and thus are considered stable
1034    /// (if the ProtoBuf definition does not change) and safe for programmatic use.
1035    pub fn as_str_name(&self) -> &'static str {
1036        match self {
1037            Self::Unspecified => "TOOL_CALL_STATUS_UNSPECIFIED",
1038            Self::Approved => "TOOL_CALL_STATUS_APPROVED",
1039            Self::Denied => "TOOL_CALL_STATUS_DENIED",
1040            Self::AwaitingApproval => "TOOL_CALL_STATUS_AWAITING_APPROVAL",
1041        }
1042    }
1043    /// Creates an enum from field names used in the ProtoBuf definition.
1044    pub fn from_str_name(value: &str) -> ::core::option::Option<Self> {
1045        match value {
1046            "TOOL_CALL_STATUS_UNSPECIFIED" => Some(Self::Unspecified),
1047            "TOOL_CALL_STATUS_APPROVED" => Some(Self::Approved),
1048            "TOOL_CALL_STATUS_DENIED" => Some(Self::Denied),
1049            "TOOL_CALL_STATUS_AWAITING_APPROVAL" => Some(Self::AwaitingApproval),
1050            _ => None,
1051        }
1052    }
1053}
1054/// Generated client implementations.
1055pub mod ai_agent_service_client {
1056    #![allow(
1057        unused_variables,
1058        dead_code,
1059        missing_docs,
1060        clippy::wildcard_imports,
1061        clippy::let_unit_value,
1062    )]
1063    use tonic::codegen::*;
1064    use tonic::codegen::http::Uri;
1065    /// AIAgentService provides AI-powered assistance for general operations
1066    #[derive(Debug, Clone)]
1067    pub struct AiAgentServiceClient<T> {
1068        inner: tonic::client::Grpc<T>,
1069    }
1070    impl AiAgentServiceClient<tonic::transport::Channel> {
1071        /// Attempt to create a new client by connecting to a given endpoint.
1072        pub async fn connect<D>(dst: D) -> Result<Self, tonic::transport::Error>
1073        where
1074            D: TryInto<tonic::transport::Endpoint>,
1075            D::Error: Into<StdError>,
1076        {
1077            let conn = tonic::transport::Endpoint::new(dst)?.connect().await?;
1078            Ok(Self::new(conn))
1079        }
1080    }
1081    impl<T> AiAgentServiceClient<T>
1082    where
1083        T: tonic::client::GrpcService<tonic::body::Body>,
1084        T::Error: Into<StdError>,
1085        T::ResponseBody: Body<Data = Bytes> + std::marker::Send + 'static,
1086        <T::ResponseBody as Body>::Error: Into<StdError> + std::marker::Send,
1087    {
1088        pub fn new(inner: T) -> Self {
1089            let inner = tonic::client::Grpc::new(inner);
1090            Self { inner }
1091        }
1092        pub fn with_origin(inner: T, origin: Uri) -> Self {
1093            let inner = tonic::client::Grpc::with_origin(inner, origin);
1094            Self { inner }
1095        }
1096        pub fn with_interceptor<F>(
1097            inner: T,
1098            interceptor: F,
1099        ) -> AiAgentServiceClient<InterceptedService<T, F>>
1100        where
1101            F: tonic::service::Interceptor,
1102            T::ResponseBody: Default,
1103            T: tonic::codegen::Service<
1104                http::Request<tonic::body::Body>,
1105                Response = http::Response<
1106                    <T as tonic::client::GrpcService<tonic::body::Body>>::ResponseBody,
1107                >,
1108            >,
1109            <T as tonic::codegen::Service<
1110                http::Request<tonic::body::Body>,
1111            >>::Error: Into<StdError> + std::marker::Send + std::marker::Sync,
1112        {
1113            AiAgentServiceClient::new(InterceptedService::new(inner, interceptor))
1114        }
1115        /// Compress requests with the given encoding.
1116        ///
1117        /// This requires the server to support it otherwise it might respond with an
1118        /// error.
1119        #[must_use]
1120        pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
1121            self.inner = self.inner.send_compressed(encoding);
1122            self
1123        }
1124        /// Enable decompressing responses.
1125        #[must_use]
1126        pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
1127            self.inner = self.inner.accept_compressed(encoding);
1128            self
1129        }
1130        /// Limits the maximum size of a decoded message.
1131        ///
1132        /// Default: `4MB`
1133        #[must_use]
1134        pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
1135            self.inner = self.inner.max_decoding_message_size(limit);
1136            self
1137        }
1138        /// Limits the maximum size of an encoded message.
1139        ///
1140        /// Default: `usize::MAX`
1141        #[must_use]
1142        pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
1143            self.inner = self.inner.max_encoding_message_size(limit);
1144            self
1145        }
1146        /// StreamChat handles bidirectional streaming chat for AI agent
1147        pub async fn stream_chat(
1148            &mut self,
1149            request: impl tonic::IntoRequest<super::StreamChatRequest>,
1150        ) -> std::result::Result<
1151            tonic::Response<tonic::codec::Streaming<super::StreamChatResponse>>,
1152            tonic::Status,
1153        > {
1154            self.inner
1155                .ready()
1156                .await
1157                .map_err(|e| {
1158                    tonic::Status::unknown(
1159                        format!("Service was not ready: {}", e.into()),
1160                    )
1161                })?;
1162            let codec = tonic::codec::ProstCodec::default();
1163            let path = http::uri::PathAndQuery::from_static(
1164                "/nominal.ai.v1.AIAgentService/StreamChat",
1165            );
1166            let mut req = request.into_request();
1167            req.extensions_mut()
1168                .insert(GrpcMethod::new("nominal.ai.v1.AIAgentService", "StreamChat"));
1169            self.inner.server_streaming(req, path, codec).await
1170        }
1171        /// GetConversation handles getting a complete conversation list, with an optional limit on number of messages returned
1172        pub async fn get_conversation(
1173            &mut self,
1174            request: impl tonic::IntoRequest<super::GetConversationRequest>,
1175        ) -> std::result::Result<
1176            tonic::Response<super::GetConversationResponse>,
1177            tonic::Status,
1178        > {
1179            self.inner
1180                .ready()
1181                .await
1182                .map_err(|e| {
1183                    tonic::Status::unknown(
1184                        format!("Service was not ready: {}", e.into()),
1185                    )
1186                })?;
1187            let codec = tonic::codec::ProstCodec::default();
1188            let path = http::uri::PathAndQuery::from_static(
1189                "/nominal.ai.v1.AIAgentService/GetConversation",
1190            );
1191            let mut req = request.into_request();
1192            req.extensions_mut()
1193                .insert(
1194                    GrpcMethod::new("nominal.ai.v1.AIAgentService", "GetConversation"),
1195                );
1196            self.inner.unary(req, path, codec).await
1197        }
1198        /// ListConversations handles getting the list of conversation ids ordered by most recently updated
1199        pub async fn list_conversations(
1200            &mut self,
1201            request: impl tonic::IntoRequest<super::ListConversationsRequest>,
1202        ) -> std::result::Result<
1203            tonic::Response<super::ListConversationsResponse>,
1204            tonic::Status,
1205        > {
1206            self.inner
1207                .ready()
1208                .await
1209                .map_err(|e| {
1210                    tonic::Status::unknown(
1211                        format!("Service was not ready: {}", e.into()),
1212                    )
1213                })?;
1214            let codec = tonic::codec::ProstCodec::default();
1215            let path = http::uri::PathAndQuery::from_static(
1216                "/nominal.ai.v1.AIAgentService/ListConversations",
1217            );
1218            let mut req = request.into_request();
1219            req.extensions_mut()
1220                .insert(
1221                    GrpcMethod::new("nominal.ai.v1.AIAgentService", "ListConversations"),
1222                );
1223            self.inner.unary(req, path, codec).await
1224        }
1225        /// CreateConversation handles creating a conversation and assigning it a conversation rid
1226        pub async fn create_conversation(
1227            &mut self,
1228            request: impl tonic::IntoRequest<super::CreateConversationRequest>,
1229        ) -> std::result::Result<
1230            tonic::Response<super::CreateConversationResponse>,
1231            tonic::Status,
1232        > {
1233            self.inner
1234                .ready()
1235                .await
1236                .map_err(|e| {
1237                    tonic::Status::unknown(
1238                        format!("Service was not ready: {}", e.into()),
1239                    )
1240                })?;
1241            let codec = tonic::codec::ProstCodec::default();
1242            let path = http::uri::PathAndQuery::from_static(
1243                "/nominal.ai.v1.AIAgentService/CreateConversation",
1244            );
1245            let mut req = request.into_request();
1246            req.extensions_mut()
1247                .insert(
1248                    GrpcMethod::new("nominal.ai.v1.AIAgentService", "CreateConversation"),
1249                );
1250            self.inner.unary(req, path, codec).await
1251        }
1252        /// UpdateConversationMetadata handles updating any metadata associated with a conversation
1253        pub async fn update_conversation_metadata(
1254            &mut self,
1255            request: impl tonic::IntoRequest<super::UpdateConversationMetadataRequest>,
1256        ) -> std::result::Result<
1257            tonic::Response<super::UpdateConversationMetadataResponse>,
1258            tonic::Status,
1259        > {
1260            self.inner
1261                .ready()
1262                .await
1263                .map_err(|e| {
1264                    tonic::Status::unknown(
1265                        format!("Service was not ready: {}", e.into()),
1266                    )
1267                })?;
1268            let codec = tonic::codec::ProstCodec::default();
1269            let path = http::uri::PathAndQuery::from_static(
1270                "/nominal.ai.v1.AIAgentService/UpdateConversationMetadata",
1271            );
1272            let mut req = request.into_request();
1273            req.extensions_mut()
1274                .insert(
1275                    GrpcMethod::new(
1276                        "nominal.ai.v1.AIAgentService",
1277                        "UpdateConversationMetadata",
1278                    ),
1279                );
1280            self.inner.unary(req, path, codec).await
1281        }
1282        /// DeleteConversation handles deleting a specific conversation by conversation rid
1283        pub async fn delete_conversation(
1284            &mut self,
1285            request: impl tonic::IntoRequest<super::DeleteConversationRequest>,
1286        ) -> std::result::Result<
1287            tonic::Response<super::DeleteConversationResponse>,
1288            tonic::Status,
1289        > {
1290            self.inner
1291                .ready()
1292                .await
1293                .map_err(|e| {
1294                    tonic::Status::unknown(
1295                        format!("Service was not ready: {}", e.into()),
1296                    )
1297                })?;
1298            let codec = tonic::codec::ProstCodec::default();
1299            let path = http::uri::PathAndQuery::from_static(
1300                "/nominal.ai.v1.AIAgentService/DeleteConversation",
1301            );
1302            let mut req = request.into_request();
1303            req.extensions_mut()
1304                .insert(
1305                    GrpcMethod::new("nominal.ai.v1.AIAgentService", "DeleteConversation"),
1306                );
1307            self.inner.unary(req, path, codec).await
1308        }
1309        /// GetSnapshotRidByUserMessageId handles resolving the snapshot rid of the workbook at the time the message is sent
1310        pub async fn get_snapshot_rid_by_user_message_id(
1311            &mut self,
1312            request: impl tonic::IntoRequest<super::GetSnapshotRidByUserMessageIdRequest>,
1313        ) -> std::result::Result<
1314            tonic::Response<super::GetSnapshotRidByUserMessageIdResponse>,
1315            tonic::Status,
1316        > {
1317            self.inner
1318                .ready()
1319                .await
1320                .map_err(|e| {
1321                    tonic::Status::unknown(
1322                        format!("Service was not ready: {}", e.into()),
1323                    )
1324                })?;
1325            let codec = tonic::codec::ProstCodec::default();
1326            let path = http::uri::PathAndQuery::from_static(
1327                "/nominal.ai.v1.AIAgentService/GetSnapshotRidByUserMessageId",
1328            );
1329            let mut req = request.into_request();
1330            req.extensions_mut()
1331                .insert(
1332                    GrpcMethod::new(
1333                        "nominal.ai.v1.AIAgentService",
1334                        "GetSnapshotRidByUserMessageId",
1335                    ),
1336                );
1337            self.inner.unary(req, path, codec).await
1338        }
1339    }
1340}
nominal_api/proto/nominal.ai.v1.rs

nominal_api/proto/
nominal.ai.v1.rs
