luminarycloud 0.13.0__py3-none-any.whl → 0.14.0__py3-none-any.whl

luminarycloud/_proto/assistant/assistant_pb2.pyi

@@ -6,17 +6,66 @@ import builtins
 import collections.abc
 import google.protobuf.descriptor
 import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
 import google.protobuf.message
 import luminarycloud._proto.client.simulation_pb2
 import sys
+import typing
 
-if sys.version_info >= (3, 8):
+if sys.version_info >= (3, 10):
     import typing as typing_extensions
 else:
     import typing_extensions
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+class _AssistRequestSource:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _AssistRequestSourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_AssistRequestSource.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    WEBAPP_UI: _AssistRequestSource.ValueType  # 0
+    NOTEBOOK_UI: _AssistRequestSource.ValueType  # 1
+
+class AssistRequestSource(_AssistRequestSource, metaclass=_AssistRequestSourceEnumTypeWrapper): ...
+
+WEBAPP_UI: AssistRequestSource.ValueType  # 0
+NOTEBOOK_UI: AssistRequestSource.ValueType  # 1
+global___AssistRequestSource = AssistRequestSource
+
+class _CellType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _CellTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_CellType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    CODE: _CellType.ValueType  # 0
+    MARKDOWN: _CellType.ValueType  # 1
+    RAW: _CellType.ValueType  # 2
+
+class CellType(_CellType, metaclass=_CellTypeEnumTypeWrapper): ...
+
+CODE: CellType.ValueType  # 0
+MARKDOWN: CellType.ValueType  # 1
+RAW: CellType.ValueType  # 2
+global___CellType = CellType
+
+class _StreamOutputType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _StreamOutputTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_StreamOutputType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    STDOUT: _StreamOutputType.ValueType  # 0
+    STDERR: _StreamOutputType.ValueType  # 1
+
+class StreamOutputType(_StreamOutputType, metaclass=_StreamOutputTypeEnumTypeWrapper): ...
+
+STDOUT: StreamOutputType.ValueType  # 0
+STDERR: StreamOutputType.ValueType  # 1
+global___StreamOutputType = StreamOutputType
+
 class GetChatSessionsRequest(google.protobuf.message.Message):
     """a request for getting sessions associated with the current project"""
 
@@ -84,16 +133,327 @@ class NewChatSessionResponse(google.protobuf.message.Message):
 
 global___NewChatSessionResponse = NewChatSessionResponse
 
-class Request(google.protobuf.message.Message):
-    """a request from user to assist."""
+class NotebookOutputItem(google.protobuf.message.Message):
+    """Represents different types of output for notebook cells
+    Based on Jupyter Notebook format specification
+    """
 
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
+    STREAM_FIELD_NUMBER: builtins.int
+    DISPLAY_DATA_FIELD_NUMBER: builtins.int
+    EXECUTE_RESULT_FIELD_NUMBER: builtins.int
+    ERROR_FIELD_NUMBER: builtins.int
+    @property
+    def stream(self) -> global___StreamOutput:
+        """For stream output (stdout/stderr)"""
+    @property
+    def display_data(self) -> global___MimeBundle:
+        """For display_data outputs"""
+    @property
+    def execute_result(self) -> global___ExecuteResult:
+        """For execute_result outputs"""
+    @property
+    def error(self) -> global___ErrorOutput:
+        """For error outputs"""
+    def __init__(
+        self,
+        *,
+        stream: global___StreamOutput | None = ...,
+        display_data: global___MimeBundle | None = ...,
+        execute_result: global___ExecuteResult | None = ...,
+        error: global___ErrorOutput | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["display_data", b"display_data", "error", b"error", "execute_result", b"execute_result", "output_content", b"output_content", "stream", b"stream"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["display_data", b"display_data", "error", b"error", "execute_result", b"execute_result", "output_content", b"output_content", "stream", b"stream"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["output_content", b"output_content"]) -> typing_extensions.Literal["stream", "display_data", "execute_result", "error"] | None: ...
+
+global___NotebookOutputItem = NotebookOutputItem
+
+class StreamOutput(google.protobuf.message.Message):
+    """Stream output (stdout/stderr)"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    TEXT_FIELD_NUMBER: builtins.int
+    name: global___StreamOutputType.ValueType
+    """The stream type (stdout or stderr)"""
+    text: builtins.str
+    """Content of the stream"""
+    def __init__(
+        self,
+        *,
+        name: global___StreamOutputType.ValueType = ...,
+        text: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["name", b"name", "text", b"text"]) -> None: ...
+
+global___StreamOutput = StreamOutput
+
+class ErrorOutput(google.protobuf.message.Message):
+    """Error output"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ENAME_FIELD_NUMBER: builtins.int
+    EVALUE_FIELD_NUMBER: builtins.int
+    TRACEBACK_FIELD_NUMBER: builtins.int
+    ename: builtins.str
+    """Exception name"""
+    evalue: builtins.str
+    """Exception value"""
+    @property
+    def traceback(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """List of traceback frames"""
+    def __init__(
+        self,
+        *,
+        ename: builtins.str = ...,
+        evalue: builtins.str = ...,
+        traceback: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["ename", b"ename", "evalue", b"evalue", "traceback", b"traceback"]) -> None: ...
+
+global___ErrorOutput = ErrorOutput
+
+class ExecuteResult(google.protobuf.message.Message):
+    """Execute result output"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATA_FIELD_NUMBER: builtins.int
+    @property
+    def data(self) -> global___MimeBundle:
+        """Output data"""
+    def __init__(
+        self,
+        *,
+        data: global___MimeBundle | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["data", b"data"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["data", b"data"]) -> None: ...
+
+global___ExecuteResult = ExecuteResult
+
+class MimeBundle(google.protobuf.message.Message):
+    """Mime-bundle for rich display outputs"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class DataEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.str
+        def __init__(
+            self,
+            *,
+            key: builtins.str = ...,
+            value: builtins.str = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    class MetadataEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.str
+        def __init__(
+            self,
+            *,
+            key: builtins.str = ...,
+            value: builtins.str = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    DATA_FIELD_NUMBER: builtins.int
+    METADATA_FIELD_NUMBER: builtins.int
+    @property
+    def data(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
+        """Mapped by mime-type (e.g., "text/plain", "image/png")"""
+    @property
+    def metadata(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
+        """Optional metadata for outputs"""
+    def __init__(
+        self,
+        *,
+        data: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+        metadata: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["data", b"data", "metadata", b"metadata"]) -> None: ...
+
+global___MimeBundle = MimeBundle
+
+class NotebookCell(google.protobuf.message.Message):
+    """Represents a cell in a notebook"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CELL_ID_FIELD_NUMBER: builtins.int
+    CELL_TYPE_FIELD_NUMBER: builtins.int
+    CONTENT_FIELD_NUMBER: builtins.int
+    OUTPUTS_FIELD_NUMBER: builtins.int
+    cell_id: builtins.str
+    """The id of the cell"""
+    cell_type: global___CellType.ValueType
+    """The type of the cell (code or markdown)"""
+    content: builtins.str
+    """The content of the cell ("source" in the notebook cell)"""
+    @property
+    def outputs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___NotebookOutputItem]:
+        """The output of the cell"""
+    def __init__(
+        self,
+        *,
+        cell_id: builtins.str = ...,
+        cell_type: global___CellType.ValueType = ...,
+        content: builtins.str = ...,
+        outputs: collections.abc.Iterable[global___NotebookOutputItem] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["cell_id", b"cell_id", "cell_type", b"cell_type", "content", b"content", "outputs", b"outputs"]) -> None: ...
+
+global___NotebookCell = NotebookCell
+
+class NotebookContext(google.protobuf.message.Message):
+    """Represents the context of a notebook"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ID_FIELD_NUMBER: builtins.int
+    NAME_FIELD_NUMBER: builtins.int
+    CELLS_FIELD_NUMBER: builtins.int
+    id: builtins.str
+    """The id of the notebook"""
+    name: builtins.str
+    """The notebook name"""
+    @property
+    def cells(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___NotebookCell]:
+        """The cells of the notebook"""
+    def __init__(
+        self,
+        *,
+        id: builtins.str = ...,
+        name: builtins.str = ...,
+        cells: collections.abc.Iterable[global___NotebookCell] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["cells", b"cells", "id", b"id", "name", b"name"]) -> None: ...
+
+global___NotebookContext = NotebookContext
+
+class SourceCode(google.protobuf.message.Message):
+    """Represents a source code (any language)"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    LANGUAGE_FIELD_NUMBER: builtins.int
+    PATH_FIELD_NUMBER: builtins.int
+    CONTENT_FIELD_NUMBER: builtins.int
+    language: builtins.str
+    path: builtins.str
+    """The path to the source code"""
+    content: builtins.str
+    """The source code content"""
+    def __init__(
+        self,
+        *,
+        language: builtins.str = ...,
+        path: builtins.str = ...,
+        content: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["content", b"content", "language", b"language", "path", b"path"]) -> None: ...
+
+global___SourceCode = SourceCode
+
+class Image(google.protobuf.message.Message):
+    """Represents an image in a notebook"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    IMAGE_BASE64_FIELD_NUMBER: builtins.int
+    DESCRIPTION_FIELD_NUMBER: builtins.int
+    image_base64: builtins.str
+    """The base64 encoded image"""
+    description: builtins.str
+    """The description of the image"""
+    def __init__(
+        self,
+        *,
+        image_base64: builtins.str = ...,
+        description: builtins.str | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_description", b"_description", "description", b"description"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_description", b"_description", "description", b"description", "image_base64", b"image_base64"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_description", b"_description"]) -> typing_extensions.Literal["description"] | None: ...
+
+global___Image = Image
+
+class ContextItem(google.protobuf.message.Message):
+    """The context items can be one of several types"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NOTEBOOK_CONTEXT_FIELD_NUMBER: builtins.int
+    SOURCE_CODE_FIELD_NUMBER: builtins.int
+    TEXT_FIELD_NUMBER: builtins.int
+    IMAGE_FIELD_NUMBER: builtins.int
+    @property
+    def notebook_context(self) -> global___NotebookContext:
+        """Notebook context"""
+    @property
+    def source_code(self) -> global___SourceCode:
+        """Source code"""
+    text: builtins.str
+    """Any text content"""
+    @property
+    def image(self) -> global___Image:
+        """Image content"""
+    def __init__(
+        self,
+        *,
+        notebook_context: global___NotebookContext | None = ...,
+        source_code: global___SourceCode | None = ...,
+        text: builtins.str = ...,
+        image: global___Image | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["image", b"image", "item", b"item", "notebook_context", b"notebook_context", "source_code", b"source_code", "text", b"text"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["image", b"image", "item", b"item", "notebook_context", b"notebook_context", "source_code", b"source_code", "text", b"text"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["item", b"item"]) -> typing_extensions.Literal["notebook_context", "source_code", "text", "image"] | None: ...
+
+global___ContextItem = ContextItem
+
+class AssistantContext(google.protobuf.message.Message):
+    """Represents the context of the assistant"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CONTEXT_FIELD_NUMBER: builtins.int
+    @property
+    def context(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ContextItem]:
+        """The context of the assistant"""
+    def __init__(
+        self,
+        *,
+        context: collections.abc.Iterable[global___ContextItem] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["context", b"context"]) -> None: ...
+
+global___AssistantContext = AssistantContext
+
+class Request(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     SESSION_FIELD_NUMBER: builtins.int
     REQUEST_FIELD_NUMBER: builtins.int
     CMD_FIELD_NUMBER: builtins.int
     ACCESS_FIELD_NUMBER: builtins.int
     SCOPE_FIELD_NUMBER: builtins.int
+    SOURCE_FIELD_NUMBER: builtins.int
+    CONTEXT_FIELD_NUMBER: builtins.int
     session: builtins.str
     """id of the long lived session, an opaque string set by initiator"""
     request: builtins.str
@@ -103,6 +463,11 @@ class Request(google.protobuf.message.Message):
     access: builtins.str
     """the access tokens for assistant to act on behalf of the user"""
     scope: builtins.str
+    source: global___AssistRequestSource.ValueType
+    """the source of the request"""
+    @property
+    def context(self) -> global___AssistantContext:
+        """the context sent to the assistant"""
     def __init__(
         self,
         *,
@@ -111,8 +476,12 @@ class Request(google.protobuf.message.Message):
         cmd: builtins.str = ...,
         access: builtins.str = ...,
         scope: builtins.str = ...,
+        source: global___AssistRequestSource.ValueType = ...,
+        context: global___AssistantContext | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["access", b"access", "cmd", b"cmd", "request", b"request", "scope", b"scope", "session", b"session"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_context", b"_context", "context", b"context"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_context", b"_context", "access", b"access", "cmd", b"cmd", "context", b"context", "request", b"request", "scope", b"scope", "session", b"session", "source", b"source"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_context", b"_context"]) -> typing_extensions.Literal["context"] | None: ...
 
 global___Request = Request
 
@@ -126,6 +495,7 @@ class Reply(google.protobuf.message.Message):
     LAST_FIELD_NUMBER: builtins.int
     ACTION_FIELD_NUMBER: builtins.int
     RESPONSE_FIELD_NUMBER: builtins.int
+    PLOTLY_PLOT_FIELD_NUMBER: builtins.int
     session: builtins.str
     """id of the session"""
     request: builtins.str
@@ -137,6 +507,8 @@ class Reply(google.protobuf.message.Message):
         """request to run a predefined action"""
     response: builtins.str
     """show a response to the user"""
+    plotly_plot: builtins.str
+    """show a plotly plot to the user"""
     def __init__(
         self,
         *,
@@ -145,10 +517,11 @@ class Reply(google.protobuf.message.Message):
         last: builtins.bool = ...,
         action: global___Action | None = ...,
         response: builtins.str = ...,
+        plotly_plot: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["action", b"action", "response", b"response", "step", b"step"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["action", b"action", "last", b"last", "request", b"request", "response", b"response", "session", b"session", "step", b"step"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["step", b"step"]) -> typing_extensions.Literal["action", "response"] | None: ...
+    def HasField(self, field_name: typing_extensions.Literal["action", b"action", "plotly_plot", b"plotly_plot", "response", b"response", "step", b"step"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["action", b"action", "last", b"last", "plotly_plot", b"plotly_plot", "request", b"request", "response", b"response", "session", b"session", "step", b"step"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["step", b"step"]) -> typing_extensions.Literal["action", "response", "plotly_plot"] | None: ...
 
 global___Reply = Reply