nominal-api-protos 0.980.0__py3-none-any.whl → 0.990.0__py3-none-any.whl

nominal_api_protos/nominal/ai/v1/workbook_agent_pb2.pyi

@@ -1,6 +1,4 @@
-import datetime
-
-from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from nominal.ai.v1 import ai_agent_pb2 as _ai_agent_pb2
 from google.protobuf.internal import containers as _containers
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
@@ -9,7 +7,7 @@ from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
-class StreamChatRequest(_message.Message):
+class WorkbookAgentServiceStreamChatRequest(_message.Message):
     __slots__ = ("messages", "notebook_as_json", "selected_tab_index", "images", "range", "message")
     MESSAGES_FIELD_NUMBER: _ClassVar[int]
     NOTEBOOK_AS_JSON_FIELD_NUMBER: _ClassVar[int]
@@ -17,320 +15,18 @@ class StreamChatRequest(_message.Message):
     IMAGES_FIELD_NUMBER: _ClassVar[int]
     RANGE_FIELD_NUMBER: _ClassVar[int]
     MESSAGE_FIELD_NUMBER: _ClassVar[int]
-    messages: _containers.RepeatedCompositeFieldContainer[ModelMessage]
+    messages: _containers.RepeatedCompositeFieldContainer[_ai_agent_pb2.ModelMessage]
     notebook_as_json: str
     selected_tab_index: int
-    images: _containers.RepeatedCompositeFieldContainer[ImagePart]
-    range: TimeRange
+    images: _containers.RepeatedCompositeFieldContainer[_ai_agent_pb2.ImagePart]
+    range: _ai_agent_pb2.TimeRange
     message: AppendMessage
-    def __init__(self, messages: _Optional[_Iterable[_Union[ModelMessage, _Mapping]]] = ..., notebook_as_json: _Optional[str] = ..., selected_tab_index: _Optional[int] = ..., images: _Optional[_Iterable[_Union[ImagePart, _Mapping]]] = ..., range: _Optional[_Union[TimeRange, _Mapping]] = ..., message: _Optional[_Union[AppendMessage, _Mapping]] = ...) -> None: ...
+    def __init__(self, messages: _Optional[_Iterable[_Union[_ai_agent_pb2.ModelMessage, _Mapping]]] = ..., notebook_as_json: _Optional[str] = ..., selected_tab_index: _Optional[int] = ..., images: _Optional[_Iterable[_Union[_ai_agent_pb2.ImagePart, _Mapping]]] = ..., range: _Optional[_Union[_ai_agent_pb2.TimeRange, _Mapping]] = ..., message: _Optional[_Union[AppendMessage, _Mapping]] = ...) -> None: ...
 
 class AppendMessage(_message.Message):
     __slots__ = ("message", "conversation_rid")
     MESSAGE_FIELD_NUMBER: _ClassVar[int]
     CONVERSATION_RID_FIELD_NUMBER: _ClassVar[int]
-    message: UserModelMessage
+    message: _ai_agent_pb2.UserModelMessage
     conversation_rid: str
-    def __init__(self, message: _Optional[_Union[UserModelMessage, _Mapping]] = ..., conversation_rid: _Optional[str] = ...) -> None: ...
-
-class CreateConversationRequest(_message.Message):
-    __slots__ = ("title", "workspace_rid", "old_conversation_rid", "previous_message_id")
-    TITLE_FIELD_NUMBER: _ClassVar[int]
-    WORKSPACE_RID_FIELD_NUMBER: _ClassVar[int]
-    OLD_CONVERSATION_RID_FIELD_NUMBER: _ClassVar[int]
-    PREVIOUS_MESSAGE_ID_FIELD_NUMBER: _ClassVar[int]
-    title: str
-    workspace_rid: str
-    old_conversation_rid: str
-    previous_message_id: str
-    def __init__(self, title: _Optional[str] = ..., workspace_rid: _Optional[str] = ..., old_conversation_rid: _Optional[str] = ..., previous_message_id: _Optional[str] = ...) -> None: ...
-
-class CreateConversationResponse(_message.Message):
-    __slots__ = ("new_conversation_rid",)
-    NEW_CONVERSATION_RID_FIELD_NUMBER: _ClassVar[int]
-    new_conversation_rid: str
-    def __init__(self, new_conversation_rid: _Optional[str] = ...) -> None: ...
-
-class UpdateConversationMetadataRequest(_message.Message):
-    __slots__ = ("title", "conversation_rid")
-    TITLE_FIELD_NUMBER: _ClassVar[int]
-    CONVERSATION_RID_FIELD_NUMBER: _ClassVar[int]
-    title: str
-    conversation_rid: str
-    def __init__(self, title: _Optional[str] = ..., conversation_rid: _Optional[str] = ...) -> None: ...
-
-class UpdateConversationMetadataResponse(_message.Message):
-    __slots__ = ()
-    def __init__(self) -> None: ...
-
-class DeleteConversationRequest(_message.Message):
-    __slots__ = ("conversation_rid",)
-    CONVERSATION_RID_FIELD_NUMBER: _ClassVar[int]
-    conversation_rid: str
-    def __init__(self, conversation_rid: _Optional[str] = ...) -> None: ...
-
-class DeleteConversationResponse(_message.Message):
-    __slots__ = ()
-    def __init__(self) -> None: ...
-
-class GetConversationRequest(_message.Message):
-    __slots__ = ("conversation_rid", "page_start_message_id", "max_message_count")
-    CONVERSATION_RID_FIELD_NUMBER: _ClassVar[int]
-    PAGE_START_MESSAGE_ID_FIELD_NUMBER: _ClassVar[int]
-    MAX_MESSAGE_COUNT_FIELD_NUMBER: _ClassVar[int]
-    conversation_rid: str
-    page_start_message_id: str
-    max_message_count: int
-    def __init__(self, conversation_rid: _Optional[str] = ..., page_start_message_id: _Optional[str] = ..., max_message_count: _Optional[int] = ...) -> None: ...
-
-class ModelMessageWithId(_message.Message):
-    __slots__ = ("message", "tool_action", "message_id")
-    MESSAGE_FIELD_NUMBER: _ClassVar[int]
-    TOOL_ACTION_FIELD_NUMBER: _ClassVar[int]
-    MESSAGE_ID_FIELD_NUMBER: _ClassVar[int]
-    message: ModelMessage
-    tool_action: ToolAction
-    message_id: str
-    def __init__(self, message: _Optional[_Union[ModelMessage, _Mapping]] = ..., tool_action: _Optional[_Union[ToolAction, _Mapping]] = ..., message_id: _Optional[str] = ...) -> None: ...
-
-class GetConversationResponse(_message.Message):
-    __slots__ = ("ordered_messages",)
-    ORDERED_MESSAGES_FIELD_NUMBER: _ClassVar[int]
-    ordered_messages: _containers.RepeatedCompositeFieldContainer[ModelMessageWithId]
-    def __init__(self, ordered_messages: _Optional[_Iterable[_Union[ModelMessageWithId, _Mapping]]] = ...) -> None: ...
-
-class ListConversationsRequest(_message.Message):
-    __slots__ = ("workspace_rid",)
-    WORKSPACE_RID_FIELD_NUMBER: _ClassVar[int]
-    workspace_rid: str
-    def __init__(self, workspace_rid: _Optional[str] = ...) -> None: ...
-
-class ConversationMetadata(_message.Message):
-    __slots__ = ("conversation_rid", "title", "created_at", "last_updated_at")
-    CONVERSATION_RID_FIELD_NUMBER: _ClassVar[int]
-    TITLE_FIELD_NUMBER: _ClassVar[int]
-    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
-    LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
-    conversation_rid: str
-    title: str
-    created_at: _timestamp_pb2.Timestamp
-    last_updated_at: _timestamp_pb2.Timestamp
-    def __init__(self, conversation_rid: _Optional[str] = ..., title: _Optional[str] = ..., created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., last_updated_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
-
-class ListConversationsResponse(_message.Message):
-    __slots__ = ("conversations",)
-    CONVERSATIONS_FIELD_NUMBER: _ClassVar[int]
-    conversations: _containers.RepeatedCompositeFieldContainer[ConversationMetadata]
-    def __init__(self, conversations: _Optional[_Iterable[_Union[ConversationMetadata, _Mapping]]] = ...) -> None: ...
-
-class TimeRange(_message.Message):
-    __slots__ = ("range_start", "range_end")
-    RANGE_START_FIELD_NUMBER: _ClassVar[int]
-    RANGE_END_FIELD_NUMBER: _ClassVar[int]
-    range_start: Timestamp
-    range_end: Timestamp
-    def __init__(self, range_start: _Optional[_Union[Timestamp, _Mapping]] = ..., range_end: _Optional[_Union[Timestamp, _Mapping]] = ...) -> None: ...
-
-class Timestamp(_message.Message):
-    __slots__ = ("seconds", "nanoseconds")
-    SECONDS_FIELD_NUMBER: _ClassVar[int]
-    NANOSECONDS_FIELD_NUMBER: _ClassVar[int]
-    seconds: int
-    nanoseconds: int
-    def __init__(self, seconds: _Optional[int] = ..., nanoseconds: _Optional[int] = ...) -> None: ...
-
-class ModelMessage(_message.Message):
-    __slots__ = ("user", "assistant")
-    USER_FIELD_NUMBER: _ClassVar[int]
-    ASSISTANT_FIELD_NUMBER: _ClassVar[int]
-    user: UserModelMessage
-    assistant: AssistantModelMessage
-    def __init__(self, user: _Optional[_Union[UserModelMessage, _Mapping]] = ..., assistant: _Optional[_Union[AssistantModelMessage, _Mapping]] = ...) -> None: ...
-
-class UserModelMessage(_message.Message):
-    __slots__ = ("text",)
-    TEXT_FIELD_NUMBER: _ClassVar[int]
-    text: _containers.RepeatedCompositeFieldContainer[UserContentPart]
-    def __init__(self, text: _Optional[_Iterable[_Union[UserContentPart, _Mapping]]] = ...) -> None: ...
-
-class AssistantModelMessage(_message.Message):
-    __slots__ = ("content_parts",)
-    CONTENT_PARTS_FIELD_NUMBER: _ClassVar[int]
-    content_parts: _containers.RepeatedCompositeFieldContainer[AssistantContentPart]
-    def __init__(self, content_parts: _Optional[_Iterable[_Union[AssistantContentPart, _Mapping]]] = ...) -> None: ...
-
-class UserContentPart(_message.Message):
-    __slots__ = ("text",)
-    TEXT_FIELD_NUMBER: _ClassVar[int]
-    text: TextPart
-    def __init__(self, text: _Optional[_Union[TextPart, _Mapping]] = ...) -> None: ...
-
-class AssistantContentPart(_message.Message):
-    __slots__ = ("text", "reasoning")
-    TEXT_FIELD_NUMBER: _ClassVar[int]
-    REASONING_FIELD_NUMBER: _ClassVar[int]
-    text: TextPart
-    reasoning: ReasoningPart
-    def __init__(self, text: _Optional[_Union[TextPart, _Mapping]] = ..., reasoning: _Optional[_Union[ReasoningPart, _Mapping]] = ...) -> None: ...
-
-class TextPart(_message.Message):
-    __slots__ = ("text",)
-    TEXT_FIELD_NUMBER: _ClassVar[int]
-    text: str
-    def __init__(self, text: _Optional[str] = ...) -> None: ...
-
-class ImagePart(_message.Message):
-    __slots__ = ("data", "media_type", "filename")
-    DATA_FIELD_NUMBER: _ClassVar[int]
-    MEDIA_TYPE_FIELD_NUMBER: _ClassVar[int]
-    FILENAME_FIELD_NUMBER: _ClassVar[int]
-    data: bytes
-    media_type: str
-    filename: str
-    def __init__(self, data: _Optional[bytes] = ..., media_type: _Optional[str] = ..., filename: _Optional[str] = ...) -> None: ...
-
-class ReasoningPart(_message.Message):
-    __slots__ = ("reasoning",)
-    REASONING_FIELD_NUMBER: _ClassVar[int]
-    reasoning: str
-    def __init__(self, reasoning: _Optional[str] = ...) -> None: ...
-
-class StreamChatResponse(_message.Message):
-    __slots__ = ("finish", "error", "text_start", "text_delta", "text_end", "reasoning_start", "reasoning_delta", "reasoning_end", "workbook_mutation", "tool_action")
-    FINISH_FIELD_NUMBER: _ClassVar[int]
-    ERROR_FIELD_NUMBER: _ClassVar[int]
-    TEXT_START_FIELD_NUMBER: _ClassVar[int]
-    TEXT_DELTA_FIELD_NUMBER: _ClassVar[int]
-    TEXT_END_FIELD_NUMBER: _ClassVar[int]
-    REASONING_START_FIELD_NUMBER: _ClassVar[int]
-    REASONING_DELTA_FIELD_NUMBER: _ClassVar[int]
-    REASONING_END_FIELD_NUMBER: _ClassVar[int]
-    WORKBOOK_MUTATION_FIELD_NUMBER: _ClassVar[int]
-    TOOL_ACTION_FIELD_NUMBER: _ClassVar[int]
-    finish: Finish
-    error: Error
-    text_start: TextStart
-    text_delta: TextDelta
-    text_end: TextEnd
-    reasoning_start: ReasoningStart
-    reasoning_delta: ReasoningDelta
-    reasoning_end: ReasoningEnd
-    workbook_mutation: WorkbookMutation
-    tool_action: ToolAction
-    def __init__(self, finish: _Optional[_Union[Finish, _Mapping]] = ..., error: _Optional[_Union[Error, _Mapping]] = ..., text_start: _Optional[_Union[TextStart, _Mapping]] = ..., text_delta: _Optional[_Union[TextDelta, _Mapping]] = ..., text_end: _Optional[_Union[TextEnd, _Mapping]] = ..., reasoning_start: _Optional[_Union[ReasoningStart, _Mapping]] = ..., reasoning_delta: _Optional[_Union[ReasoningDelta, _Mapping]] = ..., reasoning_end: _Optional[_Union[ReasoningEnd, _Mapping]] = ..., workbook_mutation: _Optional[_Union[WorkbookMutation, _Mapping]] = ..., tool_action: _Optional[_Union[ToolAction, _Mapping]] = ...) -> None: ...
-
-class Finish(_message.Message):
-    __slots__ = ("ordered_message_ids",)
-    ORDERED_MESSAGE_IDS_FIELD_NUMBER: _ClassVar[int]
-    ordered_message_ids: _containers.RepeatedScalarFieldContainer[str]
-    def __init__(self, ordered_message_ids: _Optional[_Iterable[str]] = ...) -> None: ...
-
-class Error(_message.Message):
-    __slots__ = ("message",)
-    MESSAGE_FIELD_NUMBER: _ClassVar[int]
-    message: str
-    def __init__(self, message: _Optional[str] = ...) -> None: ...
-
-class TextStart(_message.Message):
-    __slots__ = ("id",)
-    ID_FIELD_NUMBER: _ClassVar[int]
-    id: str
-    def __init__(self, id: _Optional[str] = ...) -> None: ...
-
-class TextDelta(_message.Message):
-    __slots__ = ("id", "delta")
-    ID_FIELD_NUMBER: _ClassVar[int]
-    DELTA_FIELD_NUMBER: _ClassVar[int]
-    id: str
-    delta: str
-    def __init__(self, id: _Optional[str] = ..., delta: _Optional[str] = ...) -> None: ...
-
-class TextEnd(_message.Message):
-    __slots__ = ("id",)
-    ID_FIELD_NUMBER: _ClassVar[int]
-    id: str
-    def __init__(self, id: _Optional[str] = ...) -> None: ...
-
-class ReasoningStart(_message.Message):
-    __slots__ = ("id",)
-    ID_FIELD_NUMBER: _ClassVar[int]
-    id: str
-    def __init__(self, id: _Optional[str] = ...) -> None: ...
-
-class ReasoningDelta(_message.Message):
-    __slots__ = ("id", "delta")
-    ID_FIELD_NUMBER: _ClassVar[int]
-    DELTA_FIELD_NUMBER: _ClassVar[int]
-    id: str
-    delta: str
-    def __init__(self, id: _Optional[str] = ..., delta: _Optional[str] = ...) -> None: ...
-
-class ReasoningEnd(_message.Message):
-    __slots__ = ("id",)
-    ID_FIELD_NUMBER: _ClassVar[int]
-    id: str
-    def __init__(self, id: _Optional[str] = ...) -> None: ...
-
-class AddTabMutation(_message.Message):
-    __slots__ = ("tab_name",)
-    TAB_NAME_FIELD_NUMBER: _ClassVar[int]
-    tab_name: str
-    def __init__(self, tab_name: _Optional[str] = ...) -> None: ...
-
-class AddOrUpdatePanelMutation(_message.Message):
-    __slots__ = ("panel_as_json", "panel_id", "tab_index")
-    PANEL_AS_JSON_FIELD_NUMBER: _ClassVar[int]
-    PANEL_ID_FIELD_NUMBER: _ClassVar[int]
-    TAB_INDEX_FIELD_NUMBER: _ClassVar[int]
-    panel_as_json: str
-    panel_id: str
-    tab_index: int
-    def __init__(self, panel_as_json: _Optional[str] = ..., panel_id: _Optional[str] = ..., tab_index: _Optional[int] = ...) -> None: ...
-
-class RemovePanelsMutation(_message.Message):
-    __slots__ = ("panel_ids",)
-    PANEL_IDS_FIELD_NUMBER: _ClassVar[int]
-    panel_ids: _containers.RepeatedScalarFieldContainer[str]
-    def __init__(self, panel_ids: _Optional[_Iterable[str]] = ...) -> None: ...
-
-class AddOrReplaceVariableMutation(_message.Message):
-    __slots__ = ("compute_spec_as_json", "variable_name", "display_name")
-    COMPUTE_SPEC_AS_JSON_FIELD_NUMBER: _ClassVar[int]
-    VARIABLE_NAME_FIELD_NUMBER: _ClassVar[int]
-    DISPLAY_NAME_FIELD_NUMBER: _ClassVar[int]
-    compute_spec_as_json: str
-    variable_name: str
-    display_name: str
-    def __init__(self, compute_spec_as_json: _Optional[str] = ..., variable_name: _Optional[str] = ..., display_name: _Optional[str] = ...) -> None: ...
-
-class DeleteVariablesMutation(_message.Message):
-    __slots__ = ("variable_names",)
-    VARIABLE_NAMES_FIELD_NUMBER: _ClassVar[int]
-    variable_names: _containers.RepeatedScalarFieldContainer[str]
-    def __init__(self, variable_names: _Optional[_Iterable[str]] = ...) -> None: ...
-
-class WorkbookMutation(_message.Message):
-    __slots__ = ("id", "add_tab", "add_or_update_panel", "remove_panels", "add_or_replace_variable", "delete_variables")
-    ID_FIELD_NUMBER: _ClassVar[int]
-    ADD_TAB_FIELD_NUMBER: _ClassVar[int]
-    ADD_OR_UPDATE_PANEL_FIELD_NUMBER: _ClassVar[int]
-    REMOVE_PANELS_FIELD_NUMBER: _ClassVar[int]
-    ADD_OR_REPLACE_VARIABLE_FIELD_NUMBER: _ClassVar[int]
-    DELETE_VARIABLES_FIELD_NUMBER: _ClassVar[int]
-    id: str
-    add_tab: AddTabMutation
-    add_or_update_panel: AddOrUpdatePanelMutation
-    remove_panels: RemovePanelsMutation
-    add_or_replace_variable: AddOrReplaceVariableMutation
-    delete_variables: DeleteVariablesMutation
-    def __init__(self, id: _Optional[str] = ..., add_tab: _Optional[_Union[AddTabMutation, _Mapping]] = ..., add_or_update_panel: _Optional[_Union[AddOrUpdatePanelMutation, _Mapping]] = ..., remove_panels: _Optional[_Union[RemovePanelsMutation, _Mapping]] = ..., add_or_replace_variable: _Optional[_Union[AddOrReplaceVariableMutation, _Mapping]] = ..., delete_variables: _Optional[_Union[DeleteVariablesMutation, _Mapping]] = ...) -> None: ...
-
-class ToolAction(_message.Message):
-    __slots__ = ("id", "tool_action_verb", "tool_target")
-    ID_FIELD_NUMBER: _ClassVar[int]
-    TOOL_ACTION_VERB_FIELD_NUMBER: _ClassVar[int]
-    TOOL_TARGET_FIELD_NUMBER: _ClassVar[int]
-    id: str
-    tool_action_verb: str
-    tool_target: str
-    def __init__(self, id: _Optional[str] = ..., tool_action_verb: _Optional[str] = ..., tool_target: _Optional[str] = ...) -> None: ...
+    def __init__(self, message: _Optional[_Union[_ai_agent_pb2.UserModelMessage, _Mapping]] = ..., conversation_rid: _Optional[str] = ...) -> None: ...

nominal_api_protos/nominal/ai/v1/workbook_agent_pb2_grpc.py

@@ -3,6 +3,7 @@
 import grpc
 import warnings
 
+from nominal.ai.v1 import ai_agent_pb2 as nominal_dot_ai_dot_v1_dot_ai__agent__pb2
 from nominal.ai.v1 import workbook_agent_pb2 as nominal_dot_ai_dot_v1_dot_workbook__agent__pb2
 
 GRPC_GENERATED_VERSION = '1.76.0'
@@ -27,6 +28,7 @@ if _version_not_supported:
 
 class WorkbookAgentServiceStub(object):
     """WorkbookAgentService provides AI-powered assistance for workbook operations
+    this is deprecated in favor of the AIAgentService
     """
 
     def __init__(self, channel):
@@ -37,38 +39,39 @@ class WorkbookAgentServiceStub(object):
         """
         self.StreamChat = channel.unary_stream(
                 '/nominal.ai.v1.WorkbookAgentService/StreamChat',
-                request_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.StreamChatRequest.SerializeToString,
-                response_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.StreamChatResponse.FromString,
+                request_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.WorkbookAgentServiceStreamChatRequest.SerializeToString,
+                response_deserializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.StreamChatResponse.FromString,
                 _registered_method=True)
         self.GetConversation = channel.unary_unary(
                 '/nominal.ai.v1.WorkbookAgentService/GetConversation',
-                request_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.GetConversationRequest.SerializeToString,
-                response_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.GetConversationResponse.FromString,
+                request_serializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.GetConversationRequest.SerializeToString,
+                response_deserializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.GetConversationResponse.FromString,
                 _registered_method=True)
         self.ListConversations = channel.unary_unary(
                 '/nominal.ai.v1.WorkbookAgentService/ListConversations',
-                request_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.ListConversationsRequest.SerializeToString,
-                response_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.ListConversationsResponse.FromString,
+                request_serializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.ListConversationsRequest.SerializeToString,
+                response_deserializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.ListConversationsResponse.FromString,
                 _registered_method=True)
         self.CreateConversation = channel.unary_unary(
                 '/nominal.ai.v1.WorkbookAgentService/CreateConversation',
-                request_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.CreateConversationRequest.SerializeToString,
-                response_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.CreateConversationResponse.FromString,
+                request_serializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.CreateConversationRequest.SerializeToString,
+                response_deserializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.CreateConversationResponse.FromString,
                 _registered_method=True)
         self.UpdateConversationMetadata = channel.unary_unary(
                 '/nominal.ai.v1.WorkbookAgentService/UpdateConversationMetadata',
-                request_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.UpdateConversationMetadataRequest.SerializeToString,
-                response_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.UpdateConversationMetadataResponse.FromString,
+                request_serializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.UpdateConversationMetadataRequest.SerializeToString,
+                response_deserializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.UpdateConversationMetadataResponse.FromString,
                 _registered_method=True)
         self.DeleteConversation = channel.unary_unary(
                 '/nominal.ai.v1.WorkbookAgentService/DeleteConversation',
-                request_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.DeleteConversationRequest.SerializeToString,
-                response_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.DeleteConversationResponse.FromString,
+                request_serializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.DeleteConversationRequest.SerializeToString,
+                response_deserializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.DeleteConversationResponse.FromString,
                 _registered_method=True)
 
 
 class WorkbookAgentServiceServicer(object):
     """WorkbookAgentService provides AI-powered assistance for workbook operations
+    this is deprecated in favor of the AIAgentService
     """
 
     def StreamChat(self, request, context):
@@ -118,33 +121,33 @@ def add_WorkbookAgentServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
             'StreamChat': grpc.unary_stream_rpc_method_handler(
                     servicer.StreamChat,
-                    request_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.StreamChatRequest.FromString,
-                    response_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.StreamChatResponse.SerializeToString,
+                    request_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.WorkbookAgentServiceStreamChatRequest.FromString,
+                    response_serializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.StreamChatResponse.SerializeToString,
             ),
             'GetConversation': grpc.unary_unary_rpc_method_handler(
                     servicer.GetConversation,
-                    request_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.GetConversationRequest.FromString,
-                    response_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.GetConversationResponse.SerializeToString,
+                    request_deserializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.GetConversationRequest.FromString,
+                    response_serializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.GetConversationResponse.SerializeToString,
             ),
             'ListConversations': grpc.unary_unary_rpc_method_handler(
                     servicer.ListConversations,
-                    request_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.ListConversationsRequest.FromString,
-                    response_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.ListConversationsResponse.SerializeToString,
+                    request_deserializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.ListConversationsRequest.FromString,
+                    response_serializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.ListConversationsResponse.SerializeToString,
             ),
             'CreateConversation': grpc.unary_unary_rpc_method_handler(
                     servicer.CreateConversation,
-                    request_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.CreateConversationRequest.FromString,
-                    response_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.CreateConversationResponse.SerializeToString,
+                    request_deserializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.CreateConversationRequest.FromString,
+                    response_serializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.CreateConversationResponse.SerializeToString,
             ),
             'UpdateConversationMetadata': grpc.unary_unary_rpc_method_handler(
                     servicer.UpdateConversationMetadata,
-                    request_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.UpdateConversationMetadataRequest.FromString,
-                    response_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.UpdateConversationMetadataResponse.SerializeToString,
+                    request_deserializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.UpdateConversationMetadataRequest.FromString,
+                    response_serializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.UpdateConversationMetadataResponse.SerializeToString,
             ),
             'DeleteConversation': grpc.unary_unary_rpc_method_handler(
                     servicer.DeleteConversation,
-                    request_deserializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.DeleteConversationRequest.FromString,
-                    response_serializer=nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.DeleteConversationResponse.SerializeToString,
+                    request_deserializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.DeleteConversationRequest.FromString,
+                    response_serializer=nominal_dot_ai_dot_v1_dot_ai__agent__pb2.DeleteConversationResponse.SerializeToString,
             ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
@@ -156,6 +159,7 @@ def add_WorkbookAgentServiceServicer_to_server(servicer, server):
  # This class is part of an EXPERIMENTAL API.
 class WorkbookAgentService(object):
     """WorkbookAgentService provides AI-powered assistance for workbook operations
+    this is deprecated in favor of the AIAgentService
     """
 
     @staticmethod
@@ -173,8 +177,8 @@ class WorkbookAgentService(object):
             request,
             target,
             '/nominal.ai.v1.WorkbookAgentService/StreamChat',
-            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.StreamChatRequest.SerializeToString,
-            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.StreamChatResponse.FromString,
+            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.WorkbookAgentServiceStreamChatRequest.SerializeToString,
+            nominal_dot_ai_dot_v1_dot_ai__agent__pb2.StreamChatResponse.FromString,
             options,
             channel_credentials,
             insecure,
@@ -200,8 +204,8 @@ class WorkbookAgentService(object):
             request,
             target,
             '/nominal.ai.v1.WorkbookAgentService/GetConversation',
-            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.GetConversationRequest.SerializeToString,
-            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.GetConversationResponse.FromString,
+            nominal_dot_ai_dot_v1_dot_ai__agent__pb2.GetConversationRequest.SerializeToString,
+            nominal_dot_ai_dot_v1_dot_ai__agent__pb2.GetConversationResponse.FromString,
             options,
             channel_credentials,
             insecure,
@@ -227,8 +231,8 @@ class WorkbookAgentService(object):
             request,
             target,
             '/nominal.ai.v1.WorkbookAgentService/ListConversations',
-            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.ListConversationsRequest.SerializeToString,
-            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.ListConversationsResponse.FromString,
+            nominal_dot_ai_dot_v1_dot_ai__agent__pb2.ListConversationsRequest.SerializeToString,
+            nominal_dot_ai_dot_v1_dot_ai__agent__pb2.ListConversationsResponse.FromString,
             options,
             channel_credentials,
             insecure,
@@ -254,8 +258,8 @@ class WorkbookAgentService(object):
             request,
             target,
             '/nominal.ai.v1.WorkbookAgentService/CreateConversation',
-            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.CreateConversationRequest.SerializeToString,
-            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.CreateConversationResponse.FromString,
+            nominal_dot_ai_dot_v1_dot_ai__agent__pb2.CreateConversationRequest.SerializeToString,
+            nominal_dot_ai_dot_v1_dot_ai__agent__pb2.CreateConversationResponse.FromString,
             options,
             channel_credentials,
             insecure,
@@ -281,8 +285,8 @@ class WorkbookAgentService(object):
             request,
             target,
             '/nominal.ai.v1.WorkbookAgentService/UpdateConversationMetadata',
-            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.UpdateConversationMetadataRequest.SerializeToString,
-            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.UpdateConversationMetadataResponse.FromString,
+            nominal_dot_ai_dot_v1_dot_ai__agent__pb2.UpdateConversationMetadataRequest.SerializeToString,
+            nominal_dot_ai_dot_v1_dot_ai__agent__pb2.UpdateConversationMetadataResponse.FromString,
             options,
             channel_credentials,
             insecure,
@@ -308,8 +312,8 @@ class WorkbookAgentService(object):
             request,
             target,
             '/nominal.ai.v1.WorkbookAgentService/DeleteConversation',
-            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.DeleteConversationRequest.SerializeToString,
-            nominal_dot_ai_dot_v1_dot_workbook__agent__pb2.DeleteConversationResponse.FromString,
+            nominal_dot_ai_dot_v1_dot_ai__agent__pb2.DeleteConversationRequest.SerializeToString,
+            nominal_dot_ai_dot_v1_dot_ai__agent__pb2.DeleteConversationResponse.FromString,
             options,
             channel_credentials,
             insecure,