99from ..core .jsonable_encoder import jsonable_encoder
1010from ..core .request_options import RequestOptions
1111from ..core .unchecked_base_model import construct_type
12- from .types .agent_think_request_on_listening_action import AgentThinkRequestOnListeningAction
13- from .types .agent_think_request_on_speaking_action import AgentThinkRequestOnSpeakingAction
14- from .types .agent_think_request_on_thinking_action import AgentThinkRequestOnThinkingAction
15- from .types .agent_think_response import AgentThinkResponse
12+ from .types .agent_think_agent_management_request_on_listening_action import (
13+ AgentThinkAgentManagementRequestOnListeningAction ,
14+ )
15+ from .types .agent_think_agent_management_request_on_speaking_action import (
16+ AgentThinkAgentManagementRequestOnSpeakingAction ,
17+ )
18+ from .types .agent_think_agent_management_request_on_thinking_action import (
19+ AgentThinkAgentManagementRequestOnThinkingAction ,
20+ )
21+ from .types .agent_think_agent_management_response import AgentThinkAgentManagementResponse
1622
1723# this is used as the default value for optional parameters
1824OMIT = typing .cast (typing .Any , ...)
@@ -28,13 +34,13 @@ def agent_think(
2834 agent_id : str ,
2935 * ,
3036 text : str ,
31- on_listening_action : typing .Optional [AgentThinkRequestOnListeningAction ] = OMIT ,
32- on_thinking_action : typing .Optional [AgentThinkRequestOnThinkingAction ] = OMIT ,
33- on_speaking_action : typing .Optional [AgentThinkRequestOnSpeakingAction ] = OMIT ,
37+ on_listening_action : typing .Optional [AgentThinkAgentManagementRequestOnListeningAction ] = OMIT ,
38+ on_thinking_action : typing .Optional [AgentThinkAgentManagementRequestOnThinkingAction ] = OMIT ,
39+ on_speaking_action : typing .Optional [AgentThinkAgentManagementRequestOnSpeakingAction ] = OMIT ,
3440 interruptable : typing .Optional [bool ] = OMIT ,
3541 metadata : typing .Optional [typing .Dict [str , str ]] = OMIT ,
3642 request_options : typing .Optional [RequestOptions ] = None ,
37- ) -> HttpResponse [AgentThinkResponse ]:
43+ ) -> HttpResponse [AgentThinkAgentManagementResponse ]:
3844 """
3945 Send a custom text instruction to the specified conversational AI agent instance.
4046
@@ -56,17 +62,17 @@ def agent_think(
5662 text : str
5763 The custom instruction text to inject into the current conversation pipeline. The system processes this as user input.
5864
59- on_listening_action : typing.Optional[AgentThinkRequestOnListeningAction ]
65+ on_listening_action : typing.Optional[AgentThinkAgentManagementRequestOnListeningAction ]
6066 The action to take when the agent is in a listening state:
6167 - `inject`: Inject the custom text instruction into the current turn without interrupting it.
6268 - `ignore`: Ignore the request.
6369
64- on_thinking_action : typing.Optional[AgentThinkRequestOnThinkingAction ]
70+ on_thinking_action : typing.Optional[AgentThinkAgentManagementRequestOnThinkingAction ]
6571 The action to take when the agent is in a thinking state:
6672 - `interrupt`: Interrupt the current state and start a new conversation turn.
6773 - `ignore`: Ignore the request.
6874
69- on_speaking_action : typing.Optional[AgentThinkRequestOnSpeakingAction ]
75+ on_speaking_action : typing.Optional[AgentThinkAgentManagementRequestOnSpeakingAction ]
7076 The action to take when the agent is in a speaking state:
7177 - `interrupt`: Interrupt the current state and start a new conversation turn.
7278 - `ignore`: Ignore the request.
@@ -84,7 +90,7 @@ def agent_think(
8490
8591 Returns
8692 -------
87- HttpResponse[AgentThinkResponse ]
93+ HttpResponse[AgentThinkAgentManagementResponse ]
8894 Request was successful. The response body contains the result of the request.
8995 """
9096 _response = self ._client_wrapper .httpx_client .request (
@@ -107,9 +113,9 @@ def agent_think(
107113 try :
108114 if 200 <= _response .status_code < 300 :
109115 _data = typing .cast (
110- AgentThinkResponse ,
116+ AgentThinkAgentManagementResponse ,
111117 construct_type (
112- type_ = AgentThinkResponse , # type: ignore
118+ type_ = AgentThinkAgentManagementResponse , # type: ignore
113119 object_ = _response .json (),
114120 ),
115121 )
@@ -130,13 +136,13 @@ async def agent_think(
130136 agent_id : str ,
131137 * ,
132138 text : str ,
133- on_listening_action : typing .Optional [AgentThinkRequestOnListeningAction ] = OMIT ,
134- on_thinking_action : typing .Optional [AgentThinkRequestOnThinkingAction ] = OMIT ,
135- on_speaking_action : typing .Optional [AgentThinkRequestOnSpeakingAction ] = OMIT ,
139+ on_listening_action : typing .Optional [AgentThinkAgentManagementRequestOnListeningAction ] = OMIT ,
140+ on_thinking_action : typing .Optional [AgentThinkAgentManagementRequestOnThinkingAction ] = OMIT ,
141+ on_speaking_action : typing .Optional [AgentThinkAgentManagementRequestOnSpeakingAction ] = OMIT ,
136142 interruptable : typing .Optional [bool ] = OMIT ,
137143 metadata : typing .Optional [typing .Dict [str , str ]] = OMIT ,
138144 request_options : typing .Optional [RequestOptions ] = None ,
139- ) -> AsyncHttpResponse [AgentThinkResponse ]:
145+ ) -> AsyncHttpResponse [AgentThinkAgentManagementResponse ]:
140146 """
141147 Send a custom text instruction to the specified conversational AI agent instance.
142148
@@ -158,17 +164,17 @@ async def agent_think(
158164 text : str
159165 The custom instruction text to inject into the current conversation pipeline. The system processes this as user input.
160166
161- on_listening_action : typing.Optional[AgentThinkRequestOnListeningAction ]
167+ on_listening_action : typing.Optional[AgentThinkAgentManagementRequestOnListeningAction ]
162168 The action to take when the agent is in a listening state:
163169 - `inject`: Inject the custom text instruction into the current turn without interrupting it.
164170 - `ignore`: Ignore the request.
165171
166- on_thinking_action : typing.Optional[AgentThinkRequestOnThinkingAction ]
172+ on_thinking_action : typing.Optional[AgentThinkAgentManagementRequestOnThinkingAction ]
167173 The action to take when the agent is in a thinking state:
168174 - `interrupt`: Interrupt the current state and start a new conversation turn.
169175 - `ignore`: Ignore the request.
170176
171- on_speaking_action : typing.Optional[AgentThinkRequestOnSpeakingAction ]
177+ on_speaking_action : typing.Optional[AgentThinkAgentManagementRequestOnSpeakingAction ]
172178 The action to take when the agent is in a speaking state:
173179 - `interrupt`: Interrupt the current state and start a new conversation turn.
174180 - `ignore`: Ignore the request.
@@ -186,7 +192,7 @@ async def agent_think(
186192
187193 Returns
188194 -------
189- AsyncHttpResponse[AgentThinkResponse ]
195+ AsyncHttpResponse[AgentThinkAgentManagementResponse ]
190196 Request was successful. The response body contains the result of the request.
191197 """
192198 _response = await self ._client_wrapper .httpx_client .request (
@@ -209,9 +215,9 @@ async def agent_think(
209215 try :
210216 if 200 <= _response .status_code < 300 :
211217 _data = typing .cast (
212- AgentThinkResponse ,
218+ AgentThinkAgentManagementResponse ,
213219 construct_type (
214- type_ = AgentThinkResponse , # type: ignore
220+ type_ = AgentThinkAgentManagementResponse , # type: ignore
215221 object_ = _response .json (),
216222 ),
217223 )
0 commit comments