From 2f0b32d5b253176c943bbd82a1a2be7e9ec4a137 Mon Sep 17 00:00:00 2001 From: Damien Deville Date: Wed, 11 Jan 2023 21:19:28 -0800 Subject: [PATCH] Fix `FineTune.stream_events` We incorrectly changed the original method to be async rather than making a new async version. This PR fixes that. Fixes 173. --- openai/api_resources/fine_tune.py | 40 ++++++++++++++++++++++++++++++- 1 file changed, 39 insertions(+), 1 deletion(-) diff --git a/openai/api_resources/fine_tune.py b/openai/api_resources/fine_tune.py index 65dba48836..d9c8359538 100644 --- a/openai/api_resources/fine_tune.py +++ b/openai/api_resources/fine_tune.py @@ -128,7 +128,45 @@ def _prepare_stream_events( return requestor, url @classmethod - async def stream_events( + def stream_events( + cls, + id, + api_key=None, + api_base=None, + api_type=None, + request_id=None, + api_version=None, + organization=None, + **params, + ): + requestor, url = cls._prepare_stream_events( + id, + api_key, + api_base, + api_type, + request_id, + api_version, + organization, + **params, + ) + + response, _, api_key = requestor.request( + "get", url, params, stream=True, request_id=request_id + ) + + assert not isinstance(response, OpenAIResponse) # must be an iterator + return ( + util.convert_to_openai_object( + line, + api_key, + api_version, + organization, + ) + for line in response + ) + + @classmethod + async def astream_events( cls, id, api_key=None,