|
15 | 15 | import base64
|
16 | 16 | import uuid
|
17 | 17 | from io import BytesIO
|
18 |
| -from typing import Generic, Literal, Optional, Sequence, TypeVar, get_args |
| 18 | +from typing import Generic, Literal, Optional, TypeVar |
19 | 19 |
|
20 | 20 | from langchain_core.messages import AIMessage, HumanMessage
|
21 | 21 | from langchain_core.messages import BaseMessage as LangchainBaseMessage
|
@@ -143,45 +143,11 @@ class HRIConnector(Generic[T], BaseConnector[T]):
|
143 | 143 | Used for sending and receiving messages between human and robot from various sources.
|
144 | 144 | """
|
145 | 145 |
|
146 |
| - configured_targets: Sequence[str] |
147 |
| - configured_sources: Sequence[str] |
148 |
| - |
149 |
| - def __init__( |
150 |
| - self, configured_targets: Sequence[str], configured_sources: Sequence[str] |
151 |
| - ): |
152 |
| - super().__init__() |
153 |
| - self.configured_targets = configured_targets |
154 |
| - self.configured_sources = configured_sources |
155 |
| - if not hasattr(self, "__orig_bases__"): |
156 |
| - self.__orig_bases__ = {} |
157 |
| - raise HRIException( |
158 |
| - f"Error while instantiating {str(self.__class__)}: Message type T derived from HRIMessage needs to be provided e.g. Connector[MessageType]()" |
159 |
| - ) |
160 |
| - self.T_class = get_args(self.__orig_bases__[-1])[0] |
161 |
| - |
162 | 146 | def _build_message(
|
163 | 147 | self,
|
164 | 148 | message: LangchainBaseMessage | RAIMultimodalMessage,
|
165 | 149 | communication_id: Optional[str] = None,
|
166 | 150 | seq_no: int = 0,
|
167 | 151 | seq_end: bool = False,
|
168 | 152 | ) -> T:
|
169 |
| - return self.T_class.from_langchain(message, communication_id, seq_no, seq_end) |
170 |
| - |
171 |
| - def send_all_targets( |
172 |
| - self, |
173 |
| - message: LangchainBaseMessage | RAIMultimodalMessage, |
174 |
| - communication_id: Optional[str] = None, |
175 |
| - seq_no: int = 0, |
176 |
| - seq_end: bool = False, |
177 |
| - ): |
178 |
| - for target in self.configured_targets: |
179 |
| - to_send = self._build_message(message, communication_id, seq_no, seq_end) |
180 |
| - self.send_message(to_send, target) |
181 |
| - |
182 |
| - def receive_all_sources(self, timeout_sec: float = 1.0) -> dict[str, T]: |
183 |
| - ret = {} |
184 |
| - for source in self.configured_sources: |
185 |
| - received = self.receive_message(source, timeout_sec) |
186 |
| - ret[source] = received |
187 |
| - return ret |
| 153 | + return self.T_class.from_langchain(message, communication_id, seq_no, seq_end) # type: ignore |
0 commit comments