@@ -12,6 +12,8 @@ import google.protobuf.message
1212import google .protobuf .timestamp_pb2
1313import sys
1414import typing
15+ import yandex .cloud .ai .assistants .v1 .searchindex .search_index_pb2
16+ import yandex .cloud .ai .files .v1 .file_pb2
1517
1618if sys .version_info >= (3 , 10 ):
1719 import typing as typing_extensions
@@ -75,6 +77,7 @@ class Message(google.protobuf.message.Message):
7577 LABELS_FIELD_NUMBER : builtins .int
7678 CONTENT_FIELD_NUMBER : builtins .int
7779 STATUS_FIELD_NUMBER : builtins .int
80+ CITATIONS_FIELD_NUMBER : builtins .int
7881 id : builtins .str
7982 """Unique identifier of the message."""
8083 thread_id : builtins .str
@@ -99,6 +102,10 @@ class Message(google.protobuf.message.Message):
99102 def content (self ) -> global___MessageContent :
100103 """Content of the message."""
101104
105+ @property
106+ def citations (self ) -> google .protobuf .internal .containers .RepeatedCompositeFieldContainer [global___Citation ]:
107+ """List of citations used to generate the message."""
108+
102109 def __init__ (
103110 self ,
104111 * ,
@@ -110,9 +117,10 @@ class Message(google.protobuf.message.Message):
110117 labels : collections .abc .Mapping [builtins .str , builtins .str ] | None = ...,
111118 content : global___MessageContent | None = ...,
112119 status : global___Message .MessageStatus .ValueType = ...,
120+ citations : collections .abc .Iterable [global___Citation ] | None = ...,
113121 ) -> None : ...
114122 def HasField (self , field_name : typing .Literal ["author" , b"author" , "content" , b"content" , "created_at" , b"created_at" ]) -> builtins .bool : ...
115- def ClearField (self , field_name : typing .Literal ["author" , b"author" , "content" , b"content" , "created_at" , b"created_at" , "created_by" , b"created_by" , "id" , b"id" , "labels" , b"labels" , "status" , b"status" , "thread_id" , b"thread_id" ]) -> None : ...
123+ def ClearField (self , field_name : typing .Literal ["author" , b"author" , "citations" , b"citations" , " content" , b"content" , "created_at" , b"created_at" , "created_by" , b"created_by" , "id" , b"id" , "labels" , b"labels" , "status" , b"status" , "thread_id" , b"thread_id" ]) -> None : ...
116124
117125global___Message = Message
118126
@@ -248,3 +256,98 @@ class Author(google.protobuf.message.Message):
248256 def ClearField (self , field_name : typing .Literal ["id" , b"id" , "role" , b"role" ]) -> None : ...
249257
250258global___Author = Author
259+
260+ @typing .final
261+ class Citation (google .protobuf .message .Message ):
262+ """Represents a citation used for generating a message."""
263+
264+ DESCRIPTOR : google .protobuf .descriptor .Descriptor
265+
266+ SOURCES_FIELD_NUMBER : builtins .int
267+ @property
268+ def sources (self ) -> google .protobuf .internal .containers .RepeatedCompositeFieldContainer [global___Source ]:
269+ """List of sources for citation."""
270+
271+ def __init__ (
272+ self ,
273+ * ,
274+ sources : collections .abc .Iterable [global___Source ] | None = ...,
275+ ) -> None : ...
276+ def ClearField (self , field_name : typing .Literal ["sources" , b"sources" ]) -> None : ...
277+
278+ global___Citation = Citation
279+
280+ @typing .final
281+ class Source (google .protobuf .message .Message ):
282+ """Represents a source used for generating a message citation."""
283+
284+ DESCRIPTOR : google .protobuf .descriptor .Descriptor
285+
286+ CHUNK_FIELD_NUMBER : builtins .int
287+ @property
288+ def chunk (self ) -> global___FileChunk :
289+ """File chunk source."""
290+
291+ def __init__ (
292+ self ,
293+ * ,
294+ chunk : global___FileChunk | None = ...,
295+ ) -> None : ...
296+ def HasField (self , field_name : typing .Literal ["SourceType" , b"SourceType" , "chunk" , b"chunk" ]) -> builtins .bool : ...
297+ def ClearField (self , field_name : typing .Literal ["SourceType" , b"SourceType" , "chunk" , b"chunk" ]) -> None : ...
298+ def WhichOneof (self , oneof_group : typing .Literal ["SourceType" , b"SourceType" ]) -> typing .Literal ["chunk" ] | None : ...
299+
300+ global___Source = Source
301+
302+ @typing .final
303+ class FileChunk (google .protobuf .message .Message ):
304+ """FileChunk represents a chunk of a file used as a source."""
305+
306+ DESCRIPTOR : google .protobuf .descriptor .Descriptor
307+
308+ SEARCH_INDEX_FIELD_NUMBER : builtins .int
309+ SOURCE_FILE_FIELD_NUMBER : builtins .int
310+ CONTENT_FIELD_NUMBER : builtins .int
311+ @property
312+ def search_index (self ) -> yandex .cloud .ai .assistants .v1 .searchindex .search_index_pb2 .SearchIndex :
313+ """Search index associated with the file chunk."""
314+
315+ @property
316+ def source_file (self ) -> yandex .cloud .ai .files .v1 .file_pb2 .File :
317+ """The original file from which the chunk is derived."""
318+
319+ @property
320+ def content (self ) -> global___ChunkContent :
321+ """Content of the file chunk."""
322+
323+ def __init__ (
324+ self ,
325+ * ,
326+ search_index : yandex .cloud .ai .assistants .v1 .searchindex .search_index_pb2 .SearchIndex | None = ...,
327+ source_file : yandex .cloud .ai .files .v1 .file_pb2 .File | None = ...,
328+ content : global___ChunkContent | None = ...,
329+ ) -> None : ...
330+ def HasField (self , field_name : typing .Literal ["content" , b"content" , "search_index" , b"search_index" , "source_file" , b"source_file" ]) -> builtins .bool : ...
331+ def ClearField (self , field_name : typing .Literal ["content" , b"content" , "search_index" , b"search_index" , "source_file" , b"source_file" ]) -> None : ...
332+
333+ global___FileChunk = FileChunk
334+
335+ @typing .final
336+ class ChunkContent (google .protobuf .message .Message ):
337+ """Represents the content of a file chunk."""
338+
339+ DESCRIPTOR : google .protobuf .descriptor .Descriptor
340+
341+ CONTENT_FIELD_NUMBER : builtins .int
342+ @property
343+ def content (self ) -> google .protobuf .internal .containers .RepeatedCompositeFieldContainer [global___ContentPart ]:
344+ """A list of content parts that make up the chunk."""
345+
346+ def __init__ (
347+ self ,
348+ * ,
349+ content : collections .abc .Iterable [global___ContentPart ] | None = ...,
350+ ) -> None : ...
351+ def ClearField (self , field_name : typing .Literal ["content" , b"content" ]) -> None : ...
352+
353+ global___ChunkContent = ChunkContent
0 commit comments