nominal_api/proto/
nominal.ai.v1.rs

1// This file is @generated by prost-build.
2/// StreamChatRequest is a request to stream chat messages for workbook AI agent
3#[derive(Clone, PartialEq, ::prost::Message)]
4pub struct StreamChatRequest {
5    #[prost(message, repeated, tag = "1")]
6    pub messages: ::prost::alloc::vec::Vec<ModelMessage>,
7    /// JSON-serialized representation of INotebook
8    #[prost(string, optional, tag = "2")]
9    pub notebook_as_json: ::core::option::Option<::prost::alloc::string::String>,
10    /// The current tab visible in the workbook from the user's perspective
11    #[prost(int32, optional, tag = "3")]
12    pub selected_tab_index: ::core::option::Option<i32>,
13    /// Optional: image files to provide to the agent
14    #[prost(message, repeated, tag = "4")]
15    pub images: ::prost::alloc::vec::Vec<ImagePart>,
16}
17/// ModelMessage is a discriminated union of system, user, assistant, and tool messages.
18/// Each message type has its own structure and content.
19#[derive(Clone, PartialEq, ::prost::Message)]
20pub struct ModelMessage {
21    #[prost(oneof = "model_message::Kind", tags = "1, 2")]
22    pub kind: ::core::option::Option<model_message::Kind>,
23}
24/// Nested message and enum types in `ModelMessage`.
25pub mod model_message {
26    #[derive(Clone, PartialEq, ::prost::Oneof)]
27    pub enum Kind {
28        #[prost(message, tag = "1")]
29        User(super::UserModelMessage),
30        #[prost(message, tag = "2")]
31        Assistant(super::AssistantModelMessage),
32    }
33}
34/// A user message containing text
35#[derive(Clone, PartialEq, ::prost::Message)]
36pub struct UserModelMessage {
37    #[prost(message, repeated, tag = "1")]
38    pub text: ::prost::alloc::vec::Vec<UserContentPart>,
39}
40/// An assistant message containing text
41#[derive(Clone, PartialEq, ::prost::Message)]
42pub struct AssistantModelMessage {
43    #[prost(message, repeated, tag = "1")]
44    pub content_parts: ::prost::alloc::vec::Vec<AssistantContentPart>,
45}
46#[derive(Clone, PartialEq, ::prost::Message)]
47pub struct UserContentPart {
48    #[prost(oneof = "user_content_part::Part", tags = "1")]
49    pub part: ::core::option::Option<user_content_part::Part>,
50}
51/// Nested message and enum types in `UserContentPart`.
52pub mod user_content_part {
53    #[derive(Clone, PartialEq, ::prost::Oneof)]
54    pub enum Part {
55        #[prost(message, tag = "1")]
56        Text(super::TextPart),
57    }
58}
59/// Content part for assistant messages: can be text, reasoning, or mutation.
60#[derive(Clone, PartialEq, ::prost::Message)]
61pub struct AssistantContentPart {
62    #[prost(oneof = "assistant_content_part::Part", tags = "1, 2")]
63    pub part: ::core::option::Option<assistant_content_part::Part>,
64}
65/// Nested message and enum types in `AssistantContentPart`.
66pub mod assistant_content_part {
67    #[derive(Clone, PartialEq, ::prost::Oneof)]
68    pub enum Part {
69        #[prost(message, tag = "1")]
70        Text(super::TextPart),
71        #[prost(message, tag = "2")]
72        Reasoning(super::ReasoningPart),
73    }
74}
75/// Text part for user or assistant messages.
76#[derive(Clone, PartialEq, ::prost::Message)]
77pub struct TextPart {
78    #[prost(string, tag = "1")]
79    pub text: ::prost::alloc::string::String,
80}
81/// User-supplied image part.
82#[derive(Clone, PartialEq, ::prost::Message)]
83pub struct ImagePart {
84    /// The base64-encoded image data
85    #[prost(bytes = "vec", tag = "1")]
86    pub data: ::prost::alloc::vec::Vec<u8>,
87    /// The media type of the image (e.g. "image/png", "image/jpeg")
88    #[prost(string, optional, tag = "2")]
89    pub media_type: ::core::option::Option<::prost::alloc::string::String>,
90    /// Optional: the filename of the image
91    #[prost(string, optional, tag = "3")]
92    pub filename: ::core::option::Option<::prost::alloc::string::String>,
93}
94/// Reasoning part for assistant messages.
95#[derive(Clone, PartialEq, ::prost::Message)]
96pub struct ReasoningPart {
97    #[prost(string, tag = "1")]
98    pub reasoning: ::prost::alloc::string::String,
99}
100/// StreamChatResponse is a discriminated union response to a StreamChatRequest
101#[derive(Clone, PartialEq, ::prost::Message)]
102pub struct StreamChatResponse {
103    #[prost(
104        oneof = "stream_chat_response::Response",
105        tags = "1, 2, 3, 4, 5, 6, 7, 8, 9, 10"
106    )]
107    pub response: ::core::option::Option<stream_chat_response::Response>,
108}
109/// Nested message and enum types in `StreamChatResponse`.
110pub mod stream_chat_response {
111    #[derive(Clone, PartialEq, ::prost::Oneof)]
112    pub enum Response {
113        #[prost(message, tag = "1")]
114        Finish(super::Finish),
115        #[prost(message, tag = "2")]
116        Error(super::Error),
117        #[prost(message, tag = "3")]
118        TextStart(super::TextStart),
119        #[prost(message, tag = "4")]
120        TextDelta(super::TextDelta),
121        #[prost(message, tag = "5")]
122        TextEnd(super::TextEnd),
123        #[prost(message, tag = "6")]
124        ReasoningStart(super::ReasoningStart),
125        #[prost(message, tag = "7")]
126        ReasoningDelta(super::ReasoningDelta),
127        #[prost(message, tag = "8")]
128        ReasoningEnd(super::ReasoningEnd),
129        #[prost(message, tag = "9")]
130        WorkbookMutation(super::WorkbookMutation),
131        #[prost(message, tag = "10")]
132        ToolAction(super::ToolAction),
133    }
134}
135/// Indicates the end of a chat session
136#[derive(Clone, Copy, PartialEq, ::prost::Message)]
137pub struct Finish {}
138/// An error that occurred during the chat session
139#[derive(Clone, PartialEq, ::prost::Message)]
140pub struct Error {
141    #[prost(string, tag = "1")]
142    pub message: ::prost::alloc::string::String,
143}
144/// Indicates the start of a text message from the agent
145#[derive(Clone, PartialEq, ::prost::Message)]
146pub struct TextStart {
147    /// uniquely identifies the text message (e.g. uuid) so that the client can
148    /// merge parallel message streams (if it happens).
149    #[prost(string, tag = "1")]
150    pub id: ::prost::alloc::string::String,
151}
152/// A delta (continuation) of a text message from the agent
153#[derive(Clone, PartialEq, ::prost::Message)]
154pub struct TextDelta {
155    #[prost(string, tag = "1")]
156    pub id: ::prost::alloc::string::String,
157    /// The next chunk of text
158    #[prost(string, tag = "2")]
159    pub delta: ::prost::alloc::string::String,
160}
161/// Indicates the end of a text message from the agent
162#[derive(Clone, PartialEq, ::prost::Message)]
163pub struct TextEnd {
164    #[prost(string, tag = "1")]
165    pub id: ::prost::alloc::string::String,
166}
167/// Indicates the start of a reasoning message from the agent
168#[derive(Clone, PartialEq, ::prost::Message)]
169pub struct ReasoningStart {
170    #[prost(string, tag = "1")]
171    pub id: ::prost::alloc::string::String,
172}
173/// A delta (continuation) of a reasoning message from the agent
174#[derive(Clone, PartialEq, ::prost::Message)]
175pub struct ReasoningDelta {
176    #[prost(string, tag = "1")]
177    pub id: ::prost::alloc::string::String,
178    /// The next chunk of reasoning
179    #[prost(string, tag = "2")]
180    pub delta: ::prost::alloc::string::String,
181}
182/// Indicates the end of a reasoning message from the agent
183#[derive(Clone, PartialEq, ::prost::Message)]
184pub struct ReasoningEnd {
185    #[prost(string, tag = "1")]
186    pub id: ::prost::alloc::string::String,
187}
188/// Add a new tab to the workbook
189#[derive(Clone, PartialEq, ::prost::Message)]
190pub struct AddTabMutation {
191    /// if tab_name is not provided, we'll name it "New Tab"
192    #[prost(string, optional, tag = "1")]
193    pub tab_name: ::core::option::Option<::prost::alloc::string::String>,
194}
195/// this is the "result" of the mutation
196#[derive(Clone, PartialEq, ::prost::Message)]
197pub struct AddOrUpdatePanelMutation {
198    /// JSON-serialized representation of IVizDefinition
199    #[prost(string, tag = "1")]
200    pub panel_as_json: ::prost::alloc::string::String,
201    #[prost(string, tag = "2")]
202    pub panel_id: ::prost::alloc::string::String,
203    #[prost(int32, tag = "3")]
204    pub tab_index: i32,
205}
206#[derive(Clone, PartialEq, ::prost::Message)]
207pub struct RemovePanelsMutation {
208    #[prost(string, repeated, tag = "1")]
209    pub panel_ids: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
210}
211/// AddOrReplaceVariableMutation is a mutation to add or replace a variable in the workbook
212#[derive(Clone, PartialEq, ::prost::Message)]
213pub struct AddOrReplaceVariableMutation {
214    /// scout_compute_api_ComputeSpecWithContext (this needs to be resolved)
215    /// be careful: this uses the scout_compute_api version of ComputeSpecWithContext, which stores the spec as "seriesNode"
216    /// and is NOT the same as ComputeSpecWithContext that is stored in INotebook.
217    #[prost(string, tag = "1")]
218    pub compute_spec_as_json: ::prost::alloc::string::String,
219    /// if variable_name is not provided, we'll assume it's a new variable and auto-generate a unique name
220    #[prost(string, optional, tag = "2")]
221    pub variable_name: ::core::option::Option<::prost::alloc::string::String>,
222    #[prost(string, optional, tag = "3")]
223    pub display_name: ::core::option::Option<::prost::alloc::string::String>,
224}
225/// DeleteVariablesMutation is a mutation to delete variables from the workbook
226#[derive(Clone, PartialEq, ::prost::Message)]
227pub struct DeleteVariablesMutation {
228    #[prost(string, repeated, tag = "1")]
229    pub variable_names: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
230}
231/// WorkbookMutation is a mutation to the workbook
232#[derive(Clone, PartialEq, ::prost::Message)]
233pub struct WorkbookMutation {
234    #[prost(string, tag = "1")]
235    pub id: ::prost::alloc::string::String,
236    #[prost(oneof = "workbook_mutation::Mutation", tags = "2, 3, 4, 5, 6")]
237    pub mutation: ::core::option::Option<workbook_mutation::Mutation>,
238}
239/// Nested message and enum types in `WorkbookMutation`.
240pub mod workbook_mutation {
241    #[derive(Clone, PartialEq, ::prost::Oneof)]
242    pub enum Mutation {
243        #[prost(message, tag = "2")]
244        AddTab(super::AddTabMutation),
245        #[prost(message, tag = "3")]
246        AddOrUpdatePanel(super::AddOrUpdatePanelMutation),
247        #[prost(message, tag = "4")]
248        RemovePanels(super::RemovePanelsMutation),
249        #[prost(message, tag = "5")]
250        AddOrReplaceVariable(super::AddOrReplaceVariableMutation),
251        #[prost(message, tag = "6")]
252        DeleteVariables(super::DeleteVariablesMutation),
253    }
254}
255/// this is a concise descirption of a tool call that the agent is making internally
256/// without revealing too much detail about the tool call, it informs the user what the agent is doing
257/// at a high level. the format is: `{tool_action_verb} {tool_target}` for example:
258/// "Search channels for My Datasource"
259#[derive(Clone, PartialEq, ::prost::Message)]
260pub struct ToolAction {
261    #[prost(string, tag = "1")]
262    pub id: ::prost::alloc::string::String,
263    /// "Thought", "Read", "Find", "Look-up", etc.
264    #[prost(string, tag = "2")]
265    pub tool_action_verb: ::prost::alloc::string::String,
266    /// "workbook", "channel", "variable", "panel", etc.
267    #[prost(string, optional, tag = "3")]
268    pub tool_target: ::core::option::Option<::prost::alloc::string::String>,
269}
270/// Generated client implementations.
271pub mod workbook_agent_service_client {
272    #![allow(
273        unused_variables,
274        dead_code,
275        missing_docs,
276        clippy::wildcard_imports,
277        clippy::let_unit_value,
278    )]
279    use tonic::codegen::*;
280    use tonic::codegen::http::Uri;
281    /// WorkbookAgentService provides AI-powered assistance for workbook operations
282    #[derive(Debug, Clone)]
283    pub struct WorkbookAgentServiceClient<T> {
284        inner: tonic::client::Grpc<T>,
285    }
286    impl WorkbookAgentServiceClient<tonic::transport::Channel> {
287        /// Attempt to create a new client by connecting to a given endpoint.
288        pub async fn connect<D>(dst: D) -> Result<Self, tonic::transport::Error>
289        where
290            D: TryInto<tonic::transport::Endpoint>,
291            D::Error: Into<StdError>,
292        {
293            let conn = tonic::transport::Endpoint::new(dst)?.connect().await?;
294            Ok(Self::new(conn))
295        }
296    }
297    impl<T> WorkbookAgentServiceClient<T>
298    where
299        T: tonic::client::GrpcService<tonic::body::Body>,
300        T::Error: Into<StdError>,
301        T::ResponseBody: Body<Data = Bytes> + std::marker::Send + 'static,
302        <T::ResponseBody as Body>::Error: Into<StdError> + std::marker::Send,
303    {
304        pub fn new(inner: T) -> Self {
305            let inner = tonic::client::Grpc::new(inner);
306            Self { inner }
307        }
308        pub fn with_origin(inner: T, origin: Uri) -> Self {
309            let inner = tonic::client::Grpc::with_origin(inner, origin);
310            Self { inner }
311        }
312        pub fn with_interceptor<F>(
313            inner: T,
314            interceptor: F,
315        ) -> WorkbookAgentServiceClient<InterceptedService<T, F>>
316        where
317            F: tonic::service::Interceptor,
318            T::ResponseBody: Default,
319            T: tonic::codegen::Service<
320                http::Request<tonic::body::Body>,
321                Response = http::Response<
322                    <T as tonic::client::GrpcService<tonic::body::Body>>::ResponseBody,
323                >,
324            >,
325            <T as tonic::codegen::Service<
326                http::Request<tonic::body::Body>,
327            >>::Error: Into<StdError> + std::marker::Send + std::marker::Sync,
328        {
329            WorkbookAgentServiceClient::new(InterceptedService::new(inner, interceptor))
330        }
331        /// Compress requests with the given encoding.
332        ///
333        /// This requires the server to support it otherwise it might respond with an
334        /// error.
335        #[must_use]
336        pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
337            self.inner = self.inner.send_compressed(encoding);
338            self
339        }
340        /// Enable decompressing responses.
341        #[must_use]
342        pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
343            self.inner = self.inner.accept_compressed(encoding);
344            self
345        }
346        /// Limits the maximum size of a decoded message.
347        ///
348        /// Default: `4MB`
349        #[must_use]
350        pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
351            self.inner = self.inner.max_decoding_message_size(limit);
352            self
353        }
354        /// Limits the maximum size of an encoded message.
355        ///
356        /// Default: `usize::MAX`
357        #[must_use]
358        pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
359            self.inner = self.inner.max_encoding_message_size(limit);
360            self
361        }
362        /// StreamChat handles bidirectional streaming chat for workbook AI agent
363        pub async fn stream_chat(
364            &mut self,
365            request: impl tonic::IntoRequest<super::StreamChatRequest>,
366        ) -> std::result::Result<
367            tonic::Response<tonic::codec::Streaming<super::StreamChatResponse>>,
368            tonic::Status,
369        > {
370            self.inner
371                .ready()
372                .await
373                .map_err(|e| {
374                    tonic::Status::unknown(
375                        format!("Service was not ready: {}", e.into()),
376                    )
377                })?;
378            let codec = tonic::codec::ProstCodec::default();
379            let path = http::uri::PathAndQuery::from_static(
380                "/nominal.ai.v1.WorkbookAgentService/StreamChat",
381            );
382            let mut req = request.into_request();
383            req.extensions_mut()
384                .insert(
385                    GrpcMethod::new("nominal.ai.v1.WorkbookAgentService", "StreamChat"),
386                );
387            self.inner.server_streaming(req, path, codec).await
388        }
389    }
390}
391/// CreateOrUpdateKnowledgeBaseRequest is a request to create a knowledge from an attachment in the attachment's
392/// workspace. Subsequent calls for the same attachment will overwrite the existing knowledge base.
393#[derive(Clone, PartialEq, ::prost::Message)]
394pub struct CreateOrUpdateKnowledgeBaseRequest {
395    #[prost(string, tag = "1")]
396    pub attachment_rid: ::prost::alloc::string::String,
397    /// summary of the knowledge base, will be used by the LLM to decide when to use it
398    #[prost(string, tag = "2")]
399    pub summary_description: ::prost::alloc::string::String,
400    #[prost(enumeration = "KnowledgeBaseType", tag = "3")]
401    pub r#type: i32,
402}
403/// CreateOrUpdateKnowledgeBaseResponse is the response to creating/updating a knowledge base
404#[derive(Clone, PartialEq, ::prost::Message)]
405pub struct CreateOrUpdateKnowledgeBaseResponse {
406    #[prost(string, tag = "1")]
407    pub knowledge_base_rid: ::prost::alloc::string::String,
408}
409/// KnowledgeBase represents a knowledge base entry
410#[derive(Clone, PartialEq, ::prost::Message)]
411pub struct KnowledgeBase {
412    #[prost(string, tag = "1")]
413    pub knowledge_base_rid: ::prost::alloc::string::String,
414    #[prost(string, tag = "2")]
415    pub attachment_rid: ::prost::alloc::string::String,
416    #[prost(string, tag = "3")]
417    pub workspace_rid: ::prost::alloc::string::String,
418    #[prost(string, tag = "4")]
419    pub summary_description: ::prost::alloc::string::String,
420    #[prost(enumeration = "KnowledgeBaseType", tag = "5")]
421    pub r#type: i32,
422    #[prost(int32, tag = "6")]
423    pub version: i32,
424}
425#[derive(Clone, PartialEq, ::prost::Message)]
426pub struct ListRequest {
427    #[prost(string, tag = "1")]
428    pub workspace_rid: ::prost::alloc::string::String,
429}
430#[derive(Clone, PartialEq, ::prost::Message)]
431pub struct ListResponse {
432    #[prost(message, repeated, tag = "1")]
433    pub knowledge_bases: ::prost::alloc::vec::Vec<KnowledgeBase>,
434}
435#[derive(Clone, PartialEq, ::prost::Message)]
436pub struct DeleteRequest {
437    #[prost(string, tag = "1")]
438    pub knowledge_base_rid: ::prost::alloc::string::String,
439}
440#[derive(Clone, Copy, PartialEq, ::prost::Message)]
441pub struct DeleteResponse {
442    #[prost(bool, tag = "1")]
443    pub success: bool,
444}
445#[derive(Clone, PartialEq, ::prost::Message)]
446pub struct GetBatchRequest {
447    #[prost(string, repeated, tag = "1")]
448    pub knowledge_base_rids: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
449}
450#[derive(Clone, PartialEq, ::prost::Message)]
451pub struct GetBatchResponse {
452    #[prost(message, repeated, tag = "1")]
453    pub knowledge_bases: ::prost::alloc::vec::Vec<KnowledgeBase>,
454}
455/// KnowledgeBaseType defines the types of knowledge base
456#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, PartialOrd, Ord, ::prost::Enumeration)]
457#[repr(i32)]
458pub enum KnowledgeBaseType {
459    /// defaults to PROMPT
460    Unspecified = 0,
461    /// knowledge base gets added directly to prompt (needs to be small enough!)
462    Prompt = 1,
463    /// knowledge base gets used via vector search on embeddings
464    Embedding = 2,
465}
466impl KnowledgeBaseType {
467    /// String value of the enum field names used in the ProtoBuf definition.
468    ///
469    /// The values are not transformed in any way and thus are considered stable
470    /// (if the ProtoBuf definition does not change) and safe for programmatic use.
471    pub fn as_str_name(&self) -> &'static str {
472        match self {
473            Self::Unspecified => "KNOWLEDGE_BASE_TYPE_UNSPECIFIED",
474            Self::Prompt => "KNOWLEDGE_BASE_TYPE_PROMPT",
475            Self::Embedding => "KNOWLEDGE_BASE_TYPE_EMBEDDING",
476        }
477    }
478    /// Creates an enum from field names used in the ProtoBuf definition.
479    pub fn from_str_name(value: &str) -> ::core::option::Option<Self> {
480        match value {
481            "KNOWLEDGE_BASE_TYPE_UNSPECIFIED" => Some(Self::Unspecified),
482            "KNOWLEDGE_BASE_TYPE_PROMPT" => Some(Self::Prompt),
483            "KNOWLEDGE_BASE_TYPE_EMBEDDING" => Some(Self::Embedding),
484            _ => None,
485        }
486    }
487}
488/// Generated client implementations.
489pub mod knowledge_base_service_client {
490    #![allow(
491        unused_variables,
492        dead_code,
493        missing_docs,
494        clippy::wildcard_imports,
495        clippy::let_unit_value,
496    )]
497    use tonic::codegen::*;
498    use tonic::codegen::http::Uri;
499    /// KnowledgeBaseService provides AI-powered knowledge base management
500    #[derive(Debug, Clone)]
501    pub struct KnowledgeBaseServiceClient<T> {
502        inner: tonic::client::Grpc<T>,
503    }
504    impl KnowledgeBaseServiceClient<tonic::transport::Channel> {
505        /// Attempt to create a new client by connecting to a given endpoint.
506        pub async fn connect<D>(dst: D) -> Result<Self, tonic::transport::Error>
507        where
508            D: TryInto<tonic::transport::Endpoint>,
509            D::Error: Into<StdError>,
510        {
511            let conn = tonic::transport::Endpoint::new(dst)?.connect().await?;
512            Ok(Self::new(conn))
513        }
514    }
515    impl<T> KnowledgeBaseServiceClient<T>
516    where
517        T: tonic::client::GrpcService<tonic::body::Body>,
518        T::Error: Into<StdError>,
519        T::ResponseBody: Body<Data = Bytes> + std::marker::Send + 'static,
520        <T::ResponseBody as Body>::Error: Into<StdError> + std::marker::Send,
521    {
522        pub fn new(inner: T) -> Self {
523            let inner = tonic::client::Grpc::new(inner);
524            Self { inner }
525        }
526        pub fn with_origin(inner: T, origin: Uri) -> Self {
527            let inner = tonic::client::Grpc::with_origin(inner, origin);
528            Self { inner }
529        }
530        pub fn with_interceptor<F>(
531            inner: T,
532            interceptor: F,
533        ) -> KnowledgeBaseServiceClient<InterceptedService<T, F>>
534        where
535            F: tonic::service::Interceptor,
536            T::ResponseBody: Default,
537            T: tonic::codegen::Service<
538                http::Request<tonic::body::Body>,
539                Response = http::Response<
540                    <T as tonic::client::GrpcService<tonic::body::Body>>::ResponseBody,
541                >,
542            >,
543            <T as tonic::codegen::Service<
544                http::Request<tonic::body::Body>,
545            >>::Error: Into<StdError> + std::marker::Send + std::marker::Sync,
546        {
547            KnowledgeBaseServiceClient::new(InterceptedService::new(inner, interceptor))
548        }
549        /// Compress requests with the given encoding.
550        ///
551        /// This requires the server to support it otherwise it might respond with an
552        /// error.
553        #[must_use]
554        pub fn send_compressed(mut self, encoding: CompressionEncoding) -> Self {
555            self.inner = self.inner.send_compressed(encoding);
556            self
557        }
558        /// Enable decompressing responses.
559        #[must_use]
560        pub fn accept_compressed(mut self, encoding: CompressionEncoding) -> Self {
561            self.inner = self.inner.accept_compressed(encoding);
562            self
563        }
564        /// Limits the maximum size of a decoded message.
565        ///
566        /// Default: `4MB`
567        #[must_use]
568        pub fn max_decoding_message_size(mut self, limit: usize) -> Self {
569            self.inner = self.inner.max_decoding_message_size(limit);
570            self
571        }
572        /// Limits the maximum size of an encoded message.
573        ///
574        /// Default: `usize::MAX`
575        #[must_use]
576        pub fn max_encoding_message_size(mut self, limit: usize) -> Self {
577            self.inner = self.inner.max_encoding_message_size(limit);
578            self
579        }
580        /// CreateOrUpdateKnowledgeBase creates a knowledge base in the workspace
581        pub async fn create_or_update_knowledge_base(
582            &mut self,
583            request: impl tonic::IntoRequest<super::CreateOrUpdateKnowledgeBaseRequest>,
584        ) -> std::result::Result<
585            tonic::Response<super::CreateOrUpdateKnowledgeBaseResponse>,
586            tonic::Status,
587        > {
588            self.inner
589                .ready()
590                .await
591                .map_err(|e| {
592                    tonic::Status::unknown(
593                        format!("Service was not ready: {}", e.into()),
594                    )
595                })?;
596            let codec = tonic::codec::ProstCodec::default();
597            let path = http::uri::PathAndQuery::from_static(
598                "/nominal.ai.v1.KnowledgeBaseService/CreateOrUpdateKnowledgeBase",
599            );
600            let mut req = request.into_request();
601            req.extensions_mut()
602                .insert(
603                    GrpcMethod::new(
604                        "nominal.ai.v1.KnowledgeBaseService",
605                        "CreateOrUpdateKnowledgeBase",
606                    ),
607                );
608            self.inner.unary(req, path, codec).await
609        }
610        /// List returns all knowledge bases in the specified workspace
611        pub async fn list(
612            &mut self,
613            request: impl tonic::IntoRequest<super::ListRequest>,
614        ) -> std::result::Result<tonic::Response<super::ListResponse>, tonic::Status> {
615            self.inner
616                .ready()
617                .await
618                .map_err(|e| {
619                    tonic::Status::unknown(
620                        format!("Service was not ready: {}", e.into()),
621                    )
622                })?;
623            let codec = tonic::codec::ProstCodec::default();
624            let path = http::uri::PathAndQuery::from_static(
625                "/nominal.ai.v1.KnowledgeBaseService/List",
626            );
627            let mut req = request.into_request();
628            req.extensions_mut()
629                .insert(GrpcMethod::new("nominal.ai.v1.KnowledgeBaseService", "List"));
630            self.inner.unary(req, path, codec).await
631        }
632        /// Delete removes a knowledge base by its RID
633        pub async fn delete(
634            &mut self,
635            request: impl tonic::IntoRequest<super::DeleteRequest>,
636        ) -> std::result::Result<tonic::Response<super::DeleteResponse>, tonic::Status> {
637            self.inner
638                .ready()
639                .await
640                .map_err(|e| {
641                    tonic::Status::unknown(
642                        format!("Service was not ready: {}", e.into()),
643                    )
644                })?;
645            let codec = tonic::codec::ProstCodec::default();
646            let path = http::uri::PathAndQuery::from_static(
647                "/nominal.ai.v1.KnowledgeBaseService/Delete",
648            );
649            let mut req = request.into_request();
650            req.extensions_mut()
651                .insert(GrpcMethod::new("nominal.ai.v1.KnowledgeBaseService", "Delete"));
652            self.inner.unary(req, path, codec).await
653        }
654        /// GetBatch retrieves multiple knowledge bases by their RIDs
655        pub async fn get_batch(
656            &mut self,
657            request: impl tonic::IntoRequest<super::GetBatchRequest>,
658        ) -> std::result::Result<
659            tonic::Response<super::GetBatchResponse>,
660            tonic::Status,
661        > {
662            self.inner
663                .ready()
664                .await
665                .map_err(|e| {
666                    tonic::Status::unknown(
667                        format!("Service was not ready: {}", e.into()),
668                    )
669                })?;
670            let codec = tonic::codec::ProstCodec::default();
671            let path = http::uri::PathAndQuery::from_static(
672                "/nominal.ai.v1.KnowledgeBaseService/GetBatch",
673            );
674            let mut req = request.into_request();
675            req.extensions_mut()
676                .insert(
677                    GrpcMethod::new("nominal.ai.v1.KnowledgeBaseService", "GetBatch"),
678                );
679            self.inner.unary(req, path, codec).await
680        }
681    }
682}
nominal_api/proto/nominal.ai.v1.rs

nominal_api/proto/
nominal.ai.v1.rs
