Skip to content

Commit 85dc97a

Browse files
committed
rename wait_for_workflow_completion as public method
1 parent 78e7337 commit 85dc97a

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

src/mistralai/client/workflows.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -109,13 +109,13 @@ def execute_workflow_and_wait(
109109
)
110110

111111
# Wait for completion
112-
final_execution = self._wait_for_workflow_completion(
112+
final_execution = self.wait_for_workflow_completion(
113113
execution.execution_id, polling_interval, max_attempts
114114
)
115115

116116
return final_execution.result
117117

118-
def _wait_for_workflow_completion(
118+
def wait_for_workflow_completion(
119119
self,
120120
execution_id: str,
121121
polling_interval: int = 5,
@@ -207,13 +207,13 @@ async def execute_workflow_and_wait_async(
207207
)
208208

209209
# Wait for completion
210-
final_execution = await self._wait_for_workflow_completion_async(
210+
final_execution = await self.wait_for_workflow_completion_async(
211211
execution.execution_id, polling_interval, max_attempts
212212
)
213213

214214
return final_execution.result
215215

216-
async def _wait_for_workflow_completion_async(
216+
async def wait_for_workflow_completion_async(
217217
self,
218218
execution_id: str,
219219
polling_interval: int = 5,

0 commit comments

Comments
 (0)