Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
86 changes: 53 additions & 33 deletions dapr/proto/runtime/v1/actors_pb2.py

Large diffs are not rendered by default.

261 changes: 259 additions & 2 deletions dapr/proto/runtime/v1/actors_pb2.pyi

Large diffs are not rendered by default.

92 changes: 46 additions & 46 deletions dapr/proto/runtime/v1/ai_pb2.py

Large diffs are not rendered by default.

18 changes: 16 additions & 2 deletions dapr/proto/runtime/v1/ai_pb2.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -202,6 +202,7 @@ class ConversationRequestAlpha2(_message.Message):
TOOL_CHOICE_FIELD_NUMBER: _builtins.int
RESPONSE_FORMAT_FIELD_NUMBER: _builtins.int
PROMPT_CACHE_RETENTION_FIELD_NUMBER: _builtins.int
MAX_TOKENS_FIELD_NUMBER: _builtins.int
name: _builtins.str
"""The name of Conversation component"""
context_id: _builtins.str
Expand All @@ -221,6 +222,14 @@ class ConversationRequestAlpha2(_message.Message):
ref: https://github.com/openai/openai-go/blob/main/chatcompletion.go#L1976
ref: https://python.langchain.com/docs/how_to/tool_choice/
"""
max_tokens: _builtins.int
"""Maximum number of tokens the model may generate for the completion.
Mapped to each provider's equivalent parameter (e.g. OpenAI
max_completion_tokens, Anthropic max_tokens). Overrides the component-level
maxTokens metadata default when set.
inspired by openai.ChatCompletionNewParams.MaxCompletionTokens
ref: https://github.com/openai/openai-go/blob/main/chatcompletion.go#L2131
"""
@_builtins.property
def inputs(self) -> _containers.RepeatedCompositeFieldContainer[Global___ConversationInputAlpha2]:
"""Inputs for the conversation."""
Expand Down Expand Up @@ -285,13 +294,16 @@ class ConversationRequestAlpha2(_message.Message):
tool_choice: _builtins.str | None = ...,
response_format: _struct_pb2.Struct | None = ...,
prompt_cache_retention: _duration_pb2.Duration | None = ...,
max_tokens: _builtins.int | None = ...,
) -> None: ...
_HasFieldArgType: _TypeAlias = _typing.Literal["_context_id", b"_context_id", "_prompt_cache_retention", b"_prompt_cache_retention", "_response_format", b"_response_format", "_scrub_pii", b"_scrub_pii", "_temperature", b"_temperature", "_tool_choice", b"_tool_choice", "context_id", b"context_id", "prompt_cache_retention", b"prompt_cache_retention", "response_format", b"response_format", "scrub_pii", b"scrub_pii", "temperature", b"temperature", "tool_choice", b"tool_choice"] # noqa: Y015
_HasFieldArgType: _TypeAlias = _typing.Literal["_context_id", b"_context_id", "_max_tokens", b"_max_tokens", "_prompt_cache_retention", b"_prompt_cache_retention", "_response_format", b"_response_format", "_scrub_pii", b"_scrub_pii", "_temperature", b"_temperature", "_tool_choice", b"_tool_choice", "context_id", b"context_id", "max_tokens", b"max_tokens", "prompt_cache_retention", b"prompt_cache_retention", "response_format", b"response_format", "scrub_pii", b"scrub_pii", "temperature", b"temperature", "tool_choice", b"tool_choice"] # noqa: Y015
def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
_ClearFieldArgType: _TypeAlias = _typing.Literal["_context_id", b"_context_id", "_prompt_cache_retention", b"_prompt_cache_retention", "_response_format", b"_response_format", "_scrub_pii", b"_scrub_pii", "_temperature", b"_temperature", "_tool_choice", b"_tool_choice", "context_id", b"context_id", "inputs", b"inputs", "metadata", b"metadata", "name", b"name", "parameters", b"parameters", "prompt_cache_retention", b"prompt_cache_retention", "response_format", b"response_format", "scrub_pii", b"scrub_pii", "temperature", b"temperature", "tool_choice", b"tool_choice", "tools", b"tools"] # noqa: Y015
_ClearFieldArgType: _TypeAlias = _typing.Literal["_context_id", b"_context_id", "_max_tokens", b"_max_tokens", "_prompt_cache_retention", b"_prompt_cache_retention", "_response_format", b"_response_format", "_scrub_pii", b"_scrub_pii", "_temperature", b"_temperature", "_tool_choice", b"_tool_choice", "context_id", b"context_id", "inputs", b"inputs", "max_tokens", b"max_tokens", "metadata", b"metadata", "name", b"name", "parameters", b"parameters", "prompt_cache_retention", b"prompt_cache_retention", "response_format", b"response_format", "scrub_pii", b"scrub_pii", "temperature", b"temperature", "tool_choice", b"tool_choice", "tools", b"tools"] # noqa: Y015
def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
_WhichOneofReturnType__context_id: _TypeAlias = _typing.Literal["context_id"] # noqa: Y015
_WhichOneofArgType__context_id: _TypeAlias = _typing.Literal["_context_id", b"_context_id"] # noqa: Y015
_WhichOneofReturnType__max_tokens: _TypeAlias = _typing.Literal["max_tokens"] # noqa: Y015
_WhichOneofArgType__max_tokens: _TypeAlias = _typing.Literal["_max_tokens", b"_max_tokens"] # noqa: Y015
_WhichOneofReturnType__prompt_cache_retention: _TypeAlias = _typing.Literal["prompt_cache_retention"] # noqa: Y015
_WhichOneofArgType__prompt_cache_retention: _TypeAlias = _typing.Literal["_prompt_cache_retention", b"_prompt_cache_retention"] # noqa: Y015
_WhichOneofReturnType__response_format: _TypeAlias = _typing.Literal["response_format"] # noqa: Y015
Expand All @@ -305,6 +317,8 @@ class ConversationRequestAlpha2(_message.Message):
@_typing.overload
def WhichOneof(self, oneof_group: _WhichOneofArgType__context_id) -> _WhichOneofReturnType__context_id | None: ...
@_typing.overload
def WhichOneof(self, oneof_group: _WhichOneofArgType__max_tokens) -> _WhichOneofReturnType__max_tokens | None: ...
@_typing.overload
def WhichOneof(self, oneof_group: _WhichOneofArgType__prompt_cache_retention) -> _WhichOneofReturnType__prompt_cache_retention | None: ...
@_typing.overload
def WhichOneof(self, oneof_group: _WhichOneofArgType__response_format) -> _WhichOneofReturnType__response_format | None: ...
Expand Down
4 changes: 2 additions & 2 deletions dapr/proto/runtime/v1/dapr_pb2.py

Large diffs are not rendered by default.

88 changes: 88 additions & 0 deletions dapr/proto/runtime/v1/dapr_pb2_grpc.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,6 +155,11 @@ def __init__(self, channel):
request_serializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.ListActorRemindersRequest.SerializeToString,
response_deserializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.ListActorRemindersResponse.FromString,
_registered_method=True)
self.ListActorTimers = channel.unary_unary(
'/dapr.proto.runtime.v1.Dapr/ListActorTimers',
request_serializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.ListActorTimersRequest.SerializeToString,
response_deserializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.ListActorTimersResponse.FromString,
_registered_method=True)
self.GetActorState = channel.unary_unary(
'/dapr.proto.runtime.v1.Dapr/GetActorState',
request_serializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.GetActorStateRequest.SerializeToString,
Expand All @@ -165,6 +170,11 @@ def __init__(self, channel):
request_serializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.GetActorReminderRequest.SerializeToString,
response_deserializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.GetActorReminderResponse.FromString,
_registered_method=True)
self.GetActorTimer = channel.unary_unary(
'/dapr.proto.runtime.v1.Dapr/GetActorTimer',
request_serializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.GetActorTimerRequest.SerializeToString,
response_deserializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.GetActorTimerResponse.FromString,
_registered_method=True)
self.ExecuteActorStateTransaction = channel.unary_unary(
'/dapr.proto.runtime.v1.Dapr/ExecuteActorStateTransaction',
request_serializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.ExecuteActorStateTransactionRequest.SerializeToString,
Expand Down Expand Up @@ -563,6 +573,13 @@ def ListActorReminders(self, request, context):
context.set_details('Method not implemented!')
raise NotImplementedError('Method not implemented!')

def ListActorTimers(self, request, context):
"""Lists the timers registered on this host for an actor.
"""
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
context.set_details('Method not implemented!')
raise NotImplementedError('Method not implemented!')

def GetActorState(self, request, context):
"""Gets the state for a specific actor.
"""
Expand All @@ -577,6 +594,13 @@ def GetActorReminder(self, request, context):
context.set_details('Method not implemented!')
raise NotImplementedError('Method not implemented!')

def GetActorTimer(self, request, context):
"""Gets a timer registered on this host for an actor.
"""
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
context.set_details('Method not implemented!')
raise NotImplementedError('Method not implemented!')

def ExecuteActorStateTransaction(self, request, context):
"""Executes state transactions for a specified actor
"""
Expand Down Expand Up @@ -1033,6 +1057,11 @@ def add_DaprServicer_to_server(servicer, server):
request_deserializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.ListActorRemindersRequest.FromString,
response_serializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.ListActorRemindersResponse.SerializeToString,
),
'ListActorTimers': grpc.unary_unary_rpc_method_handler(
servicer.ListActorTimers,
request_deserializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.ListActorTimersRequest.FromString,
response_serializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.ListActorTimersResponse.SerializeToString,
),
'GetActorState': grpc.unary_unary_rpc_method_handler(
servicer.GetActorState,
request_deserializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.GetActorStateRequest.FromString,
Expand All @@ -1043,6 +1072,11 @@ def add_DaprServicer_to_server(servicer, server):
request_deserializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.GetActorReminderRequest.FromString,
response_serializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.GetActorReminderResponse.SerializeToString,
),
'GetActorTimer': grpc.unary_unary_rpc_method_handler(
servicer.GetActorTimer,
request_deserializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.GetActorTimerRequest.FromString,
response_serializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.GetActorTimerResponse.SerializeToString,
),
'ExecuteActorStateTransaction': grpc.unary_unary_rpc_method_handler(
servicer.ExecuteActorStateTransaction,
request_deserializer=dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.ExecuteActorStateTransactionRequest.FromString,
Expand Down Expand Up @@ -1867,6 +1901,33 @@ def ListActorReminders(request,
metadata,
_registered_method=True)

@staticmethod
def ListActorTimers(request,
target,
options=(),
channel_credentials=None,
call_credentials=None,
insecure=False,
compression=None,
wait_for_ready=None,
timeout=None,
metadata=None):
return grpc.experimental.unary_unary(
request,
target,
'/dapr.proto.runtime.v1.Dapr/ListActorTimers',
dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.ListActorTimersRequest.SerializeToString,
dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.ListActorTimersResponse.FromString,
options,
channel_credentials,
insecure,
call_credentials,
compression,
wait_for_ready,
timeout,
metadata,
_registered_method=True)

@staticmethod
def GetActorState(request,
target,
Expand Down Expand Up @@ -1921,6 +1982,33 @@ def GetActorReminder(request,
metadata,
_registered_method=True)

@staticmethod
def GetActorTimer(request,
target,
options=(),
channel_credentials=None,
call_credentials=None,
insecure=False,
compression=None,
wait_for_ready=None,
timeout=None,
metadata=None):
return grpc.experimental.unary_unary(
request,
target,
'/dapr.proto.runtime.v1.Dapr/GetActorTimer',
dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.GetActorTimerRequest.SerializeToString,
dapr_dot_proto_dot_runtime_dot_v1_dot_actors__pb2.GetActorTimerResponse.FromString,
options,
channel_credentials,
insecure,
call_credentials,
compression,
wait_for_ready,
timeout,
metadata,
_registered_method=True)

@staticmethod
def ExecuteActorStateTransaction(request,
target,
Expand Down
Loading
Loading