diff --git a/examples/ai_horde_client/image/async_manual_client_example.py b/examples/ai_horde_client/image/async_manual_client_example.py index 5dbf620..fe236e5 100644 --- a/examples/ai_horde_client/image/async_manual_client_example.py +++ b/examples/ai_horde_client/image/async_manual_client_example.py @@ -1,5 +1,6 @@ import argparse import asyncio +import sys import time from pathlib import Path @@ -119,4 +120,7 @@ async def main(apikey: str = ANON_API_KEY) -> None: ) args = parser.parse_args() + if sys.platform == "win32": + asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy()) + asyncio.run(main(args.apikey)) diff --git a/examples/ai_horde_client/image/async_simple_client_example.py b/examples/ai_horde_client/image/async_simple_client_example.py index aa1dd05..3cc973b 100644 --- a/examples/ai_horde_client/image/async_simple_client_example.py +++ b/examples/ai_horde_client/image/async_simple_client_example.py @@ -1,5 +1,6 @@ import argparse import asyncio +import sys from pathlib import Path import aiohttp @@ -161,5 +162,8 @@ async def async_simple_generate_example(apikey: str = ANON_API_KEY) -> None: ) args = parser.parse_args() + if sys.platform == "win32": + asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy()) + # Run the example. asyncio.run(async_simple_generate_example(args.apikey)) diff --git a/examples/ai_horde_client/text/async_text_generate.py b/examples/ai_horde_client/text/async_text_generate.py index f551bcd..f9aece7 100644 --- a/examples/ai_horde_client/text/async_text_generate.py +++ b/examples/ai_horde_client/text/async_text_generate.py @@ -1,5 +1,6 @@ import argparse import asyncio +import sys from pathlib import Path import aiofiles @@ -115,4 +116,7 @@ async def main(apikey: str) -> None: ) args = parser.parse_args() + if sys.platform == "win32": + asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy()) + asyncio.run(main(args.api_key)) diff --git a/horde_sdk/__init__.py b/horde_sdk/__init__.py index c7d0f6b..afcbdbe 100644 --- a/horde_sdk/__init__.py +++ b/horde_sdk/__init__.py @@ -60,6 +60,7 @@ def _dev_env_var_warnings() -> None: # pragma: no cover _dev_env_var_warnings() + from horde_sdk.consts import ( PAYLOAD_HTTP_METHODS, HTTPMethod, diff --git a/tests/conftest.py b/tests/conftest.py index bdec2f1..1e348b2 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,12 +1,17 @@ +import asyncio import base64 import os import pathlib +import sys import pytest from loguru import logger os.environ["TESTS_ONGOING"] = "1" +if sys.platform == "win32": + asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy()) + from horde_sdk.ai_horde_api.apimodels import ImageGenerateAsyncRequest, ImageGenerationInputPayload from horde_sdk.generic_api.consts import ANON_API_KEY