|
108 | 108 |
|
109 | 109 |
|
110 | 110 | class CNAgent(Agent): |
111 | | - def with_stt(self, vendor: CNSTT) -> "CNAgent": |
112 | | - return typing.cast("CNAgent", super().with_stt(typing.cast(BaseSTT, vendor))) |
| 111 | + def with_stt(self, vendor: BaseSTT) -> "CNAgent": |
| 112 | + return typing.cast("CNAgent", super().with_stt(vendor)) |
113 | 113 |
|
114 | | - def with_llm(self, vendor: CNLLM) -> "CNAgent": |
115 | | - return typing.cast("CNAgent", super().with_llm(typing.cast(BaseLLM, vendor))) |
| 114 | + def with_llm(self, vendor: BaseLLM) -> "CNAgent": |
| 115 | + return typing.cast("CNAgent", super().with_llm(vendor)) |
116 | 116 |
|
117 | | - def with_tts(self, vendor: CNTTS) -> "CNAgent": |
118 | | - return typing.cast("CNAgent", super().with_tts(typing.cast(BaseTTS, vendor))) |
| 117 | + def with_tts(self, vendor: BaseTTS) -> "CNAgent": |
| 118 | + return typing.cast("CNAgent", super().with_tts(vendor)) |
119 | 119 |
|
120 | | - def with_avatar(self, vendor: CNAvatar) -> "CNAgent": |
121 | | - return typing.cast("CNAgent", super().with_avatar(typing.cast(BaseAvatar, vendor))) |
| 120 | + def with_avatar(self, vendor: BaseAvatar) -> "CNAgent": |
| 121 | + return typing.cast("CNAgent", super().with_avatar(vendor)) |
122 | 122 |
|
123 | 123 |
|
124 | 124 | class GlobalAgent(Agent): |
125 | | - def with_stt(self, vendor: GlobalSTT) -> "GlobalAgent": |
126 | | - return typing.cast("GlobalAgent", super().with_stt(typing.cast(BaseSTT, vendor))) |
| 125 | + def with_stt(self, vendor: BaseSTT) -> "GlobalAgent": |
| 126 | + return typing.cast("GlobalAgent", super().with_stt(vendor)) |
127 | 127 |
|
128 | | - def with_llm(self, vendor: GlobalLLM) -> "GlobalAgent": |
129 | | - return typing.cast("GlobalAgent", super().with_llm(typing.cast(BaseLLM, vendor))) |
| 128 | + def with_llm(self, vendor: BaseLLM) -> "GlobalAgent": |
| 129 | + return typing.cast("GlobalAgent", super().with_llm(vendor)) |
130 | 130 |
|
131 | | - def with_tts(self, vendor: GlobalTTS) -> "GlobalAgent": |
132 | | - return typing.cast("GlobalAgent", super().with_tts(typing.cast(BaseTTS, vendor))) |
| 131 | + def with_tts(self, vendor: BaseTTS) -> "GlobalAgent": |
| 132 | + return typing.cast("GlobalAgent", super().with_tts(vendor)) |
133 | 133 |
|
134 | | - def with_avatar(self, vendor: GlobalAvatar) -> "GlobalAgent": |
135 | | - return typing.cast("GlobalAgent", super().with_avatar(typing.cast(BaseAvatar, vendor))) |
| 134 | + def with_avatar(self, vendor: BaseAvatar) -> "GlobalAgent": |
| 135 | + return typing.cast("GlobalAgent", super().with_avatar(vendor)) |
136 | 136 |
|
137 | 137 |
|
138 | 138 | RegionalAgent = typing.Union[CNAgent, GlobalAgent] |
0 commit comments