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#[derive(Clone, PartialEq, ::prost::Message)]
626pub struct StreamChatRequest {
627    /// The conversation ID
628    #[prost(string, tag = "1")]
629    pub conversation_rid: ::prost::alloc::string::String,
630    /// The user message to append to the conversation
631    #[prost(message, optional, tag = "2")]
632    pub message: ::core::option::Option<UserModelMessage>,
633    /// Optional: image files to provide to the agent
634    #[prost(message, repeated, tag = "3")]
635    pub images: ::prost::alloc::vec::Vec<ImagePart>,
636    /// Optional: tool approval result for when the LLM needs HITL for a tool call
637    #[prost(message, optional, tag = "6")]
638    pub tool_approval: ::core::option::Option<ToolApprovalResult>,
639    /// Context-specific fields based on the oneofKind.
640    #[prost(oneof = "stream_chat_request::Context", tags = "4, 5")]
641    pub context: ::core::option::Option<stream_chat_request::Context>,
642}
643/// Nested message and enum types in `StreamChatRequest`.
644pub mod stream_chat_request {
645    /// Context-specific fields based on the oneofKind.
646    #[derive(Clone, PartialEq, ::prost::Oneof)]
647    pub enum Context {
648        #[prost(message, tag = "4")]
649        Workbook(super::WorkbookContext),
650        #[prost(message, tag = "5")]
651        Global(super::GlobalContext),
652    }
653}
654/// WorkbookContext contains workbook-specific context fields
655#[derive(Clone, PartialEq, ::prost::Message)]
656pub struct WorkbookContext {
657    /// RID of the workbook to use for context
658    #[prost(string, tag = "1")]
659    pub workbook_rid: ::prost::alloc::string::String,
660    /// The user's presence in the workbook
661    #[prost(message, optional, tag = "2")]
662    pub user_presence: ::core::option::Option<WorkbookUserPresence>,
663}
664/// DefaultContext (no context)
665#[derive(Clone, Copy, PartialEq, ::prost::Message)]
666pub struct GlobalContext {}
667/// WorkbookUserPresence contains the user's presence in the workbook
668/// which is used to describe what the user is viewing at the time of the message.
669#[derive(Clone, Copy, PartialEq, ::prost::Message)]
670pub struct WorkbookUserPresence {
671    #[prost(int32, tag = "1")]
672    pub tab_index: i32,
673    #[prost(message, optional, tag = "2")]
674    pub range: ::core::option::Option<TimeRange>,
675}
676/// CreateConversation request will create a new conversation thread
677/// if old conversation id is not set, a brand new, clear chat is created
678/// If old conversation id is set without a previous message id, the full conversation thread will be copied
679/// if previous message id is set with a previous message id, the conversation thread up until that message will be copied
680/// the above case is useful for branching a conversation into a new thread
681#[derive(Clone, PartialEq, ::prost::Message)]
682pub struct CreateConversationRequest {
683    #[prost(string, tag = "1")]
684    pub title: ::prost::alloc::string::String,
685    #[prost(string, tag = "2")]
686    pub workspace_rid: ::prost::alloc::string::String,
687    #[prost(string, optional, tag = "3")]
688    pub old_conversation_rid: ::core::option::Option<::prost::alloc::string::String>,
689    #[prost(string, optional, tag = "4")]
690    pub previous_message_id: ::core::option::Option<::prost::alloc::string::String>,
691    #[prost(message, optional, tag = "5")]
692    pub conversation_mode: ::core::option::Option<ConversationMode>,
693}
694/// CreateConversationResponse will return the conversation id for the new conversation
695#[derive(Clone, PartialEq, ::prost::Message)]
696pub struct CreateConversationResponse {
697    #[prost(string, tag = "1")]
698    pub new_conversation_rid: ::prost::alloc::string::String,
699}
700/// Updates the fields if specified (optional means no change for that field)
701#[derive(Clone, PartialEq, ::prost::Message)]
702pub struct UpdateConversationMetadataRequest {
703    #[prost(string, optional, tag = "1")]
704    pub title: ::core::option::Option<::prost::alloc::string::String>,
705    #[prost(string, tag = "2")]
706    pub conversation_rid: ::prost::alloc::string::String,
707    #[prost(message, optional, tag = "3")]
708    pub conversation_mode: ::core::option::Option<ConversationMode>,
709}
710#[derive(Clone, Copy, PartialEq, ::prost::Message)]
711pub struct UpdateConversationMetadataResponse {}
712#[derive(Clone, PartialEq, ::prost::Message)]
713pub struct DeleteConversationRequest {
714    #[prost(string, tag = "1")]
715    pub conversation_rid: ::prost::alloc::string::String,
716}
717#[derive(Clone, Copy, PartialEq, ::prost::Message)]
718pub struct DeleteConversationResponse {}
719/// a GetConversationRequest allows you to retrieve a subset of messages from a conversation thread represented
720/// by provided rid. To start from a particular message - you can also provide a message id.
721#[derive(Clone, PartialEq, ::prost::Message)]
722pub struct GetConversationRequest {
723    #[prost(string, tag = "1")]
724    pub conversation_rid: ::prost::alloc::string::String,
725    #[prost(string, optional, tag = "2")]
726    pub page_start_message_id: ::core::option::Option<::prost::alloc::string::String>,
727    #[prost(int32, optional, tag = "3")]
728    pub max_message_count: ::core::option::Option<i32>,
729}
730/// Model message with id allows you to identify the message ID of a given message
731#[derive(Clone, PartialEq, ::prost::Message)]
732pub struct ModelMessageWithId {
733    #[prost(string, tag = "3")]
734    pub message_id: ::prost::alloc::string::String,
735    /// WB agent user messages can have snapshot rids associated with them
736    #[prost(string, optional, tag = "4")]
737    pub snapshot_rid: ::core::option::Option<::prost::alloc::string::String>,
738    #[prost(message, optional, tag = "5")]
739    pub tool_approval_request: ::core::option::Option<ToolCallDescription>,
740    #[prost(oneof = "model_message_with_id::Content", tags = "1, 2")]
741    pub content: ::core::option::Option<model_message_with_id::Content>,
742}
743/// Nested message and enum types in `ModelMessageWithId`.
744pub mod model_message_with_id {
745    #[derive(Clone, PartialEq, ::prost::Oneof)]
746    pub enum Content {
747        #[prost(message, tag = "1")]
748        Message(super::ModelMessage),
749        #[prost(message, tag = "2")]
750        ToolAction(super::ToolAction),
751    }
752}
753#[derive(Clone, PartialEq, ::prost::Message)]
754pub struct GetConversationResponse {
755    #[prost(message, repeated, tag = "1")]
756    pub ordered_messages: ::prost::alloc::vec::Vec<ModelMessageWithId>,
757    #[prost(message, optional, tag = "2")]
758    pub conversation_metadata: ::core::option::Option<ConversationMetadata>,
759}
760/// Will generate all conversation threads that this user has in this workspace
761#[derive(Clone, PartialEq, ::prost::Message)]
762pub struct ListConversationsRequest {
763    #[prost(string, tag = "1")]
764    pub workspace_rid: ::prost::alloc::string::String,
765    #[prost(string, optional, tag = "2")]
766    pub next_page_token: ::core::option::Option<::prost::alloc::string::String>,
767    #[prost(int32, optional, tag = "3")]
768    pub page_size: ::core::option::Option<i32>,
769}
770#[derive(Clone, PartialEq, ::prost::Message)]
771pub struct ConversationMetadata {
772    #[prost(string, tag = "1")]
773    pub conversation_rid: ::prost::alloc::string::String,
774    #[prost(string, tag = "2")]
775    pub title: ::prost::alloc::string::String,
776    #[prost(message, optional, tag = "3")]
777    pub created_at: ::core::option::Option<
778        super::super::super::google::protobuf::Timestamp,
779    >,
780    #[prost(message, optional, tag = "4")]
781    pub last_updated_at: ::core::option::Option<
782        super::super::super::google::protobuf::Timestamp,
783    >,
784    #[prost(message, optional, tag = "5")]
785    pub mode: ::core::option::Option<ConversationMode>,
786}
787/// ListConversationsResponse is a list of conversations that can be used in a call to GetConversationRequest
788/// to get a full conversation from storage. These are ordered by creation time.
789#[derive(Clone, PartialEq, ::prost::Message)]
790pub struct ListConversationsResponse {
791    #[prost(message, repeated, tag = "1")]
792    pub conversations: ::prost::alloc::vec::Vec<ConversationMetadata>,
793    #[prost(string, optional, tag = "2")]
794    pub next_page_token: ::core::option::Option<::prost::alloc::string::String>,
795}
796#[derive(Clone, Copy, PartialEq, ::prost::Message)]
797pub struct TimeRange {
798    #[prost(message, optional, tag = "1")]
799    pub range_start: ::core::option::Option<Timestamp>,
800    #[prost(message, optional, tag = "2")]
801    pub range_end: ::core::option::Option<Timestamp>,
802}
803#[derive(Clone, Copy, PartialEq, ::prost::Message)]
804pub struct Timestamp {
805    #[prost(int32, tag = "1")]
806    pub seconds: i32,
807    #[prost(int32, tag = "2")]
808    pub nanoseconds: i32,
809}
810/// ModelMessage is a discriminated union of system, user, assistant, and tool messages.
811/// Each message type has its own structure and content.
812#[derive(Clone, PartialEq, ::prost::Message)]
813pub struct ModelMessage {
814    #[prost(oneof = "model_message::Kind", tags = "1, 2")]
815    pub kind: ::core::option::Option<model_message::Kind>,
816}
817/// Nested message and enum types in `ModelMessage`.
818pub mod model_message {
819    #[derive(Clone, PartialEq, ::prost::Oneof)]
820    pub enum Kind {
821        #[prost(message, tag = "1")]
822        User(super::UserModelMessage),
823        #[prost(message, tag = "2")]
824        Assistant(super::AssistantModelMessage),
825    }
826}
827/// A user message containing text
828#[derive(Clone, PartialEq, ::prost::Message)]
829pub struct UserModelMessage {
830    #[prost(message, repeated, tag = "1")]
831    pub text: ::prost::alloc::vec::Vec<UserContentPart>,
832}
833/// An assistant message containing text
834#[derive(Clone, PartialEq, ::prost::Message)]
835pub struct AssistantModelMessage {
836    #[prost(message, repeated, tag = "1")]
837    pub content_parts: ::prost::alloc::vec::Vec<AssistantContentPart>,
838}
839#[derive(Clone, PartialEq, ::prost::Message)]
840pub struct UserContentPart {
841    #[prost(oneof = "user_content_part::Part", tags = "1")]
842    pub part: ::core::option::Option<user_content_part::Part>,
843}
844/// Nested message and enum types in `UserContentPart`.
845pub mod user_content_part {
846    #[derive(Clone, PartialEq, ::prost::Oneof)]
847    pub enum Part {
848        #[prost(message, tag = "1")]
849        Text(super::TextPart),
850    }
851}
852/// Content part for assistant messages: can be text, reasoning, or mutation.
853#[derive(Clone, PartialEq, ::prost::Message)]
854pub struct AssistantContentPart {
855    #[prost(oneof = "assistant_content_part::Part", tags = "1, 2")]
856    pub part: ::core::option::Option<assistant_content_part::Part>,
857}
858/// Nested message and enum types in `AssistantContentPart`.
859pub mod assistant_content_part {
860    #[derive(Clone, PartialEq, ::prost::Oneof)]
861    pub enum Part {
862        #[prost(message, tag = "1")]
863        Text(super::TextPart),
864        #[prost(message, tag = "2")]
865        Reasoning(super::ReasoningPart),
866    }
867}
868/// Text part for user or assistant messages.
869#[derive(Clone, PartialEq, ::prost::Message)]
870pub struct TextPart {
871    #[prost(string, tag = "1")]
872    pub text: ::prost::alloc::string::String,
873}
874/// User-supplied image part.
875#[derive(Clone, PartialEq, ::prost::Message)]
876pub struct ImagePart {
877    /// The base64-encoded image data
878    #[prost(bytes = "vec", tag = "1")]
879    pub data: ::prost::alloc::vec::Vec<u8>,
880    /// The media type of the image (e.g. "image/png", "image/jpeg")
881    #[prost(string, optional, tag = "2")]
882    pub media_type: ::core::option::Option<::prost::alloc::string::String>,
883    /// Optional: the filename of the image
884    #[prost(string, optional, tag = "3")]
885    pub filename: ::core::option::Option<::prost::alloc::string::String>,
886}
887/// Reasoning part for assistant messages.
888#[derive(Clone, PartialEq, ::prost::Message)]
889pub struct ReasoningPart {
890    #[prost(string, tag = "1")]
891    pub reasoning: ::prost::alloc::string::String,
892}
893/// StreamChatResponse is a discriminated union response to a StreamChatRequest
894#[derive(Clone, PartialEq, ::prost::Message)]
895pub struct StreamChatResponse {
896    #[prost(
897        oneof = "stream_chat_response::Response",
898        tags = "1, 2, 3, 4, 5, 6, 7, 8, 10"
899    )]
900    pub response: ::core::option::Option<stream_chat_response::Response>,
901}
902/// Nested message and enum types in `StreamChatResponse`.
903pub mod stream_chat_response {
904    #[derive(Clone, PartialEq, ::prost::Oneof)]
905    pub enum Response {
906        #[prost(message, tag = "1")]
907        Finish(super::Finish),
908        #[prost(message, tag = "2")]
909        Error(super::Error),
910        #[prost(message, tag = "3")]
911        TextStart(super::TextStart),
912        #[prost(message, tag = "4")]
913        TextDelta(super::TextDelta),
914        #[prost(message, tag = "5")]
915        TextEnd(super::TextEnd),
916        #[prost(message, tag = "6")]
917        ReasoningStart(super::ReasoningStart),
918        #[prost(message, tag = "7")]
919        ReasoningDelta(super::ReasoningDelta),
920        #[prost(message, tag = "8")]
921        ReasoningEnd(super::ReasoningEnd),
922        #[prost(message, tag = "10")]
923        ToolAction(super::ToolAction),
924    }
925}
926#[derive(Clone, PartialEq, ::prost::Message)]
927pub struct ToolCallDescription {
928    #[prost(string, tag = "1")]
929    pub tool_call_id: ::prost::alloc::string::String,
930    #[prost(string, tag = "2")]
931    pub tool_name: ::prost::alloc::string::String,
932    /// string representation of the proposed tool args for display
933    #[prost(string, tag = "3")]
934    pub tool_args_json_string: ::prost::alloc::string::String,
935    /// used to conditionally render an approval button based the outcome of the tool call
936    #[prost(enumeration = "ToolCallStatus", tag = "4")]
937    pub status: i32,
938}
939/// Indicates the end of a chat session
940#[derive(Clone, PartialEq, ::prost::Message)]
941pub struct Finish {
942    /// The message ids in order of all generated messages for this agent run
943    /// These ids can be used to branch a message from that specific message
944    #[prost(string, repeated, tag = "1")]
945    pub ordered_message_ids: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
946    /// In the case that this is the first agent run in a conversation thread, we also
947    /// return the new conversation title generated
948    #[prost(string, optional, tag = "2")]
949    pub new_title: ::core::option::Option<::prost::alloc::string::String>,
950    /// In the case that the agent tries to use a tool that requires approval, it ends
951    /// its run with a ToolApprovalRequest - which must be followed with a ToolApprovalResponse
952    #[prost(message, optional, tag = "3")]
953    pub tool_approval_request: ::core::option::Option<ToolCallDescription>,
954}
955/// An error that occurred during the chat session
956#[derive(Clone, PartialEq, ::prost::Message)]
957pub struct Error {
958    #[prost(string, tag = "1")]
959    pub message: ::prost::alloc::string::String,
960}
961/// Indicates the start of a text message from the agent
962#[derive(Clone, PartialEq, ::prost::Message)]
963pub struct TextStart {
964    /// uniquely identifies the text message (e.g. uuid) so that the client can
965    /// merge parallel message streams (if it happens).
966    #[prost(string, tag = "1")]
967    pub id: ::prost::alloc::string::String,
968}
969/// A delta (continuation) of a text message from the agent
970#[derive(Clone, PartialEq, ::prost::Message)]
971pub struct TextDelta {
972    #[prost(string, tag = "1")]
973    pub id: ::prost::alloc::string::String,
974    /// The next chunk of text
975    #[prost(string, tag = "2")]
976    pub delta: ::prost::alloc::string::String,
977}
978/// Indicates the end of a text message from the agent
979#[derive(Clone, PartialEq, ::prost::Message)]
980pub struct TextEnd {
981    #[prost(string, tag = "1")]
982    pub id: ::prost::alloc::string::String,
983}
984/// Indicates the start of a reasoning message from the agent
985#[derive(Clone, PartialEq, ::prost::Message)]
986pub struct ReasoningStart {
987    #[prost(string, tag = "1")]
988    pub id: ::prost::alloc::string::String,
989}
990/// A delta (continuation) of a reasoning message from the agent
991#[derive(Clone, PartialEq, ::prost::Message)]
992pub struct ReasoningDelta {
993    #[prost(string, tag = "1")]
994    pub id: ::prost::alloc::string::String,
995    /// The next chunk of reasoning
996    #[prost(string, tag = "2")]
997    pub delta: ::prost::alloc::string::String,
998}
999/// Indicates the end of a reasoning message from the agent
1000#[derive(Clone, PartialEq, ::prost::Message)]
1001pub struct ReasoningEnd {
1002    #[prost(string, tag = "1")]
1003    pub id: ::prost::alloc::string::String,
1004}
1005/// this is a concise description of a tool call that the agent is making internally
1006/// without revealing too much detail about the tool call, it informs the user what the agent is doing
1007/// at a high level. the format is: `{tool_action_verb} {tool_target}` for example:
1008/// "Search channels for My Datasource"
1009#[derive(Clone, PartialEq, ::prost::Message)]
1010pub struct ToolAction {
1011    #[prost(string, tag = "1")]
1012    pub id: ::prost::alloc::string::String,
1013    /// "Thought", "Read", "Find", "Look-up", etc.
1014    #[prost(string, tag = "2")]
1015    pub tool_action_verb: ::prost::alloc::string::String,
1016    /// "workbook", "channel", "variable", "panel", etc.
1017    #[prost(string, optional, tag = "3")]
1018    pub tool_target: ::core::option::Option<::prost::alloc::string::String>,
1019}
1020#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, PartialOrd, Ord, ::prost::Enumeration)]
1021#[repr(i32)]
1022pub enum ToolCallStatus {
1023    Unspecified = 0,
1024    Approved = 1,
1025    Denied = 2,
1026    AwaitingApproval = 3,
1027}
1028impl ToolCallStatus {
1029    /// String value of the enum field names used in the ProtoBuf definition.
1030    ///
1031    /// The values are not transformed in any way and thus are considered stable
1032    /// (if the ProtoBuf definition does not change) and safe for programmatic use.
1033    pub fn as_str_name(&self) -> &'static str {
1034        match self {
1035            Self::Unspecified => "TOOL_CALL_STATUS_UNSPECIFIED",
1036            Self::Approved => "TOOL_CALL_STATUS_APPROVED",
1037            Self::Denied => "TOOL_CALL_STATUS_DENIED",
1038            Self::AwaitingApproval => "TOOL_CALL_STATUS_AWAITING_APPROVAL",
1039        }
1040    }
1041    /// Creates an enum from field names used in the ProtoBuf definition.
1042    pub fn from_str_name(value: &str) -> ::core::option::Option<Self> {
1043        match value {
1044            "TOOL_CALL_STATUS_UNSPECIFIED" => Some(Self::Unspecified),
1045            "TOOL_CALL_STATUS_APPROVED" => Some(Self::Approved),
1046            "TOOL_CALL_STATUS_DENIED" => Some(Self::Denied),
1047            "TOOL_CALL_STATUS_AWAITING_APPROVAL" => Some(Self::AwaitingApproval),
1048            _ => None,
1049        }
1050    }
1051}
1052/// Generated client implementations.
1053pub mod ai_agent_service_client {
1054    #![allow(
1055        unused_variables,
1056        dead_code,
1057        missing_docs,
1058        clippy::wildcard_imports,
1059        clippy::let_unit_value,
1060    )]
1061    use tonic::codegen::*;
1062    use tonic::codegen::http::Uri;
1063    /// AIAgentService provides AI-powered assistance for general operations
1064    #[derive(Debug, Clone)]
1065    pub struct AiAgentServiceClient<T> {
1066        inner: tonic::client::Grpc<T>,
1067    }
1068    impl AiAgentServiceClient<tonic::transport::Channel> {
1069        /// Attempt to create a new client by connecting to a given endpoint.
1070        pub async fn connect<D>(dst: D) -> Result<Self, tonic::transport::Error>
1071        where
1072            D: TryInto<tonic::transport::Endpoint>,
1073            D::Error: Into<StdError>,
1074        {
1075            let conn = tonic::transport::Endpoint::new(dst)?.connect().await?;
1076            Ok(Self::new(conn))
1077        }
1078    }
1079    impl<T> AiAgentServiceClient<T>
1080    where
1081        T: tonic::client::GrpcService<tonic::body::Body>,
1082        T::Error: Into<StdError>,
1083        T::ResponseBody: Body<Data = Bytes> + std::marker::Send + 'static,
1084        <T::ResponseBody as Body>::Error: Into<StdError> + std::marker::Send,
1085    {
1086        pub fn new(inner: T) -> Self {
1087            let inner = tonic::client::Grpc::new(inner);
1088            Self { inner }
1089        }
1090        pub fn with_origin(inner: T, origin: Uri) -> Self {
1091            let inner = tonic::client::Grpc::with_origin(inner, origin);
1092            Self { inner }
1093        }
1094        pub fn with_interceptor<F>(
1095            inner: T,
1096            interceptor: F,
1097        ) -> AiAgentServiceClient<InterceptedService<T, F>>
1098        where
1099            F: tonic::service::Interceptor,
1100            T::ResponseBody: Default,
1101            T: tonic::codegen::Service<
1102                http::Request<tonic::body::Body>,
1103                Response = http::Response<
1104                    <T as tonic::client::GrpcService<tonic::body::Body>>::ResponseBody,
1105                >,
1106            >,
1107            <T as tonic::codegen::Service<
1108                http::Request<tonic::body::Body>,
1109            >>::Error: Into<StdError> + std::marker::Send + std::marker::Sync,
1110        {
1111            AiAgentServiceClient::new(InterceptedService::new(inner, interceptor))
1112        }
1113        /// Compress requests with the given encoding.
1114        ///
1115        /// This requires the server to support it otherwise it might respond with an
1116        /// error.
1117        #[must_use]
1118        pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
1119            self.inner = self.inner.send_compressed(encoding);
1120            self
1121        }
1122        /// Enable decompressing responses.
1123        #[must_use]
1124        pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
1125            self.inner = self.inner.accept_compressed(encoding);
1126            self
1127        }
1128        /// Limits the maximum size of a decoded message.
1129        ///
1130        /// Default: `4MB`
1131        #[must_use]
1132        pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
1133            self.inner = self.inner.max_decoding_message_size(limit);
1134            self
1135        }
1136        /// Limits the maximum size of an encoded message.
1137        ///
1138        /// Default: `usize::MAX`
1139        #[must_use]
1140        pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
1141            self.inner = self.inner.max_encoding_message_size(limit);
1142            self
1143        }
1144        /// StreamChat handles bidirectional streaming chat for AI agent
1145        pub async fn stream_chat(
1146            &mut self,
1147            request: impl tonic::IntoRequest<super::StreamChatRequest>,
1148        ) -> std::result::Result<
1149            tonic::Response<tonic::codec::Streaming<super::StreamChatResponse>>,
1150            tonic::Status,
1151        > {
1152            self.inner
1153                .ready()
1154                .await
1155                .map_err(|e| {
1156                    tonic::Status::unknown(
1157                        format!("Service was not ready: {}", e.into()),
1158                    )
1159                })?;
1160            let codec = tonic::codec::ProstCodec::default();
1161            let path = http::uri::PathAndQuery::from_static(
1162                "/nominal.ai.v1.AIAgentService/StreamChat",
1163            );
1164            let mut req = request.into_request();
1165            req.extensions_mut()
1166                .insert(GrpcMethod::new("nominal.ai.v1.AIAgentService", "StreamChat"));
1167            self.inner.server_streaming(req, path, codec).await
1168        }
1169        /// GetConversation handles getting a complete conversation list, with an optional limit on number of messages returned
1170        pub async fn get_conversation(
1171            &mut self,
1172            request: impl tonic::IntoRequest<super::GetConversationRequest>,
1173        ) -> std::result::Result<
1174            tonic::Response<super::GetConversationResponse>,
1175            tonic::Status,
1176        > {
1177            self.inner
1178                .ready()
1179                .await
1180                .map_err(|e| {
1181                    tonic::Status::unknown(
1182                        format!("Service was not ready: {}", e.into()),
1183                    )
1184                })?;
1185            let codec = tonic::codec::ProstCodec::default();
1186            let path = http::uri::PathAndQuery::from_static(
1187                "/nominal.ai.v1.AIAgentService/GetConversation",
1188            );
1189            let mut req = request.into_request();
1190            req.extensions_mut()
1191                .insert(
1192                    GrpcMethod::new("nominal.ai.v1.AIAgentService", "GetConversation"),
1193                );
1194            self.inner.unary(req, path, codec).await
1195        }
1196        /// ListConversations handles getting the list of conversation ids ordered by most recently updated
1197        pub async fn list_conversations(
1198            &mut self,
1199            request: impl tonic::IntoRequest<super::ListConversationsRequest>,
1200        ) -> std::result::Result<
1201            tonic::Response<super::ListConversationsResponse>,
1202            tonic::Status,
1203        > {
1204            self.inner
1205                .ready()
1206                .await
1207                .map_err(|e| {
1208                    tonic::Status::unknown(
1209                        format!("Service was not ready: {}", e.into()),
1210                    )
1211                })?;
1212            let codec = tonic::codec::ProstCodec::default();
1213            let path = http::uri::PathAndQuery::from_static(
1214                "/nominal.ai.v1.AIAgentService/ListConversations",
1215            );
1216            let mut req = request.into_request();
1217            req.extensions_mut()
1218                .insert(
1219                    GrpcMethod::new("nominal.ai.v1.AIAgentService", "ListConversations"),
1220                );
1221            self.inner.unary(req, path, codec).await
1222        }
1223        /// CreateConversation handles creating a conversation and assigning it a conversation rid
1224        pub async fn create_conversation(
1225            &mut self,
1226            request: impl tonic::IntoRequest<super::CreateConversationRequest>,
1227        ) -> std::result::Result<
1228            tonic::Response<super::CreateConversationResponse>,
1229            tonic::Status,
1230        > {
1231            self.inner
1232                .ready()
1233                .await
1234                .map_err(|e| {
1235                    tonic::Status::unknown(
1236                        format!("Service was not ready: {}", e.into()),
1237                    )
1238                })?;
1239            let codec = tonic::codec::ProstCodec::default();
1240            let path = http::uri::PathAndQuery::from_static(
1241                "/nominal.ai.v1.AIAgentService/CreateConversation",
1242            );
1243            let mut req = request.into_request();
1244            req.extensions_mut()
1245                .insert(
1246                    GrpcMethod::new("nominal.ai.v1.AIAgentService", "CreateConversation"),
1247                );
1248            self.inner.unary(req, path, codec).await
1249        }
1250        /// UpdateConversationMetadata handles updating any metadata associated with a conversation
1251        pub async fn update_conversation_metadata(
1252            &mut self,
1253            request: impl tonic::IntoRequest<super::UpdateConversationMetadataRequest>,
1254        ) -> std::result::Result<
1255            tonic::Response<super::UpdateConversationMetadataResponse>,
1256            tonic::Status,
1257        > {
1258            self.inner
1259                .ready()
1260                .await
1261                .map_err(|e| {
1262                    tonic::Status::unknown(
1263                        format!("Service was not ready: {}", e.into()),
1264                    )
1265                })?;
1266            let codec = tonic::codec::ProstCodec::default();
1267            let path = http::uri::PathAndQuery::from_static(
1268                "/nominal.ai.v1.AIAgentService/UpdateConversationMetadata",
1269            );
1270            let mut req = request.into_request();
1271            req.extensions_mut()
1272                .insert(
1273                    GrpcMethod::new(
1274                        "nominal.ai.v1.AIAgentService",
1275                        "UpdateConversationMetadata",
1276                    ),
1277                );
1278            self.inner.unary(req, path, codec).await
1279        }
1280        /// DeleteConversation handles deleting a specific conversation by conversation rid
1281        pub async fn delete_conversation(
1282            &mut self,
1283            request: impl tonic::IntoRequest<super::DeleteConversationRequest>,
1284        ) -> std::result::Result<
1285            tonic::Response<super::DeleteConversationResponse>,
1286            tonic::Status,
1287        > {
1288            self.inner
1289                .ready()
1290                .await
1291                .map_err(|e| {
1292                    tonic::Status::unknown(
1293                        format!("Service was not ready: {}", e.into()),
1294                    )
1295                })?;
1296            let codec = tonic::codec::ProstCodec::default();
1297            let path = http::uri::PathAndQuery::from_static(
1298                "/nominal.ai.v1.AIAgentService/DeleteConversation",
1299            );
1300            let mut req = request.into_request();
1301            req.extensions_mut()
1302                .insert(
1303                    GrpcMethod::new("nominal.ai.v1.AIAgentService", "DeleteConversation"),
1304                );
1305            self.inner.unary(req, path, codec).await
1306        }
1307        /// GetSnapshotRidByUserMessageId handles resolving the snapshot rid of the workbook at the time the message is sent
1308        pub async fn get_snapshot_rid_by_user_message_id(
1309            &mut self,
1310            request: impl tonic::IntoRequest<super::GetSnapshotRidByUserMessageIdRequest>,
1311        ) -> std::result::Result<
1312            tonic::Response<super::GetSnapshotRidByUserMessageIdResponse>,
1313            tonic::Status,
1314        > {
1315            self.inner
1316                .ready()
1317                .await
1318                .map_err(|e| {
1319                    tonic::Status::unknown(
1320                        format!("Service was not ready: {}", e.into()),
1321                    )
1322                })?;
1323            let codec = tonic::codec::ProstCodec::default();
1324            let path = http::uri::PathAndQuery::from_static(
1325                "/nominal.ai.v1.AIAgentService/GetSnapshotRidByUserMessageId",
1326            );
1327            let mut req = request.into_request();
1328            req.extensions_mut()
1329                .insert(
1330                    GrpcMethod::new(
1331                        "nominal.ai.v1.AIAgentService",
1332                        "GetSnapshotRidByUserMessageId",
1333                    ),
1334                );
1335            self.inner.unary(req, path, codec).await
1336        }
1337    }
1338}
nominal_api/proto/nominal.ai.v1.rs

nominal_api/proto/
nominal.ai.v1.rs
