Increase completions run timeout

Author: vhaldemar

src/yandex_cloud_ml_sdk/_models/completions/model.py

@@ -176,7 +176,7 @@ async def _run(
         self,
         messages: MessageInputType,
         *,
-        timeout=60,
+        timeout=180,
     ) -> GPTModelResult[ToolCallTypeT]:
         async for result in self._run_sync_impl(
             messages=messages,
@@ -193,7 +193,7 @@ async def _run_stream(
         self,
         messages: MessageInputType,
         *,
-        timeout=60,
+        timeout=180,
     ) -> AsyncIterator[GPTModelResult[ToolCallTypeT]]:
         async for result in self._run_sync_impl(
             messages=messages,
@@ -267,7 +267,7 @@ async def run(
         self,
         messages: MessageInputType,
         *,
-        timeout=60,
+        timeout=180,
     ) -> GPTModelResult[AsyncToolCall]:
         return await self._run(
             messages=messages,
@@ -278,7 +278,7 @@ async def run_stream(
         self,
         messages: MessageInputType,
         *,
-        timeout=60,
+        timeout=180,
     ) -> AsyncIterator[GPTModelResult[AsyncToolCall]]:
         async for result in self._run_stream(
             messages=messages,
@@ -412,7 +412,7 @@ def run(
         self,
         messages: MessageInputType,
         *,
-        timeout=60,
+        timeout=180,
     ) -> GPTModelResult[ToolCall]:
         return self.__run(
             messages=messages,
@@ -423,7 +423,7 @@ def run_stream(
         self,
         messages: MessageInputType,
         *,
-        timeout=60,
+        timeout=180,
     ) -> Iterator[GPTModelResult[ToolCall]]:
         yield from self.__run_stream(
             messages=messages,