Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
rysweet committed Nov 16, 2024
1 parent 6b833ff commit ec923b4
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 8 deletions.
1 change: 1 addition & 0 deletions python/packages/autogen-core/samples/xlang/user_input.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@

input_types = Union[ConversationClosed, Input, Output]


class UserProxy(RoutedAgent):
"""An agent that allows the user to play the role of an agent in the conversation via input."""

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,14 +28,14 @@
)

import grpc
from google.protobuf import any_pb2
from grpc.aio import StreamStreamCall
from opentelemetry.trace import TracerProvider
from typing_extensions import Self, deprecated

from autogen_core.base import JSON_DATA_CONTENT_TYPE
from autogen_core.base._serialization import MessageSerializer, SerializationRegistry
from autogen_core.base._type_helpers import ChannelArgumentType
from google.protobuf import any_pb2

from ..base import (
Agent,
Expand All @@ -53,7 +53,11 @@
)
from ..components import TypeSubscription
from ._helpers import SubscriptionManager, get_impl
from .protos import agent_worker_pb2, agent_worker_pb2_grpc, cloudevent_pb2, cloudevent_pb2_grpc, agent_events_pb2, agent_events_pb2_grpc
from .protos import (
agent_worker_pb2,
agent_worker_pb2_grpc,
cloudevent_pb2,
)
from .telemetry import MessageRuntimeTracingConfig, TraceHelper, get_telemetry_grpc_metadata

if TYPE_CHECKING:
Expand Down Expand Up @@ -381,14 +385,9 @@ async def publish_message(
with self._trace_helper.trace_block(
"create", topic_id, parent=None, extraAttributes={"message_type": message_type}
):
serialized_message = self._serialization_registry.serialize(
message, type_name=message_type, data_content_type=JSON_DATA_CONTENT_TYPE
)
telemetry_metadata = get_telemetry_grpc_metadata()
proto_data = any_pb2.Any()
proto_data.Pack(
msg=message
)
proto_data.Pack(msg=message)
if sender is not None:
source = str(AgentId(type=sender.type, key=sender.key))
else:
Expand Down

0 comments on commit ec923b4

Please sign in to comment.