From 482143f4c3c4219f78913da657e428da59453545 Mon Sep 17 00:00:00 2001 From: Pratham Soni Date: Tue, 6 Feb 2024 17:56:06 -0500 Subject: [PATCH] =?UTF-8?q?Update=20=5F=5Finit=5F=5F.py=20to=20include=20h?= =?UTF-8?q?andling=20models=20and=20remove=20unpatch=20in=20=5F=E2=80=A6?= =?UTF-8?q?=20(#409)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- instructor/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/instructor/__init__.py b/instructor/__init__.py index 2f2b3fc..b5cef02 100644 --- a/instructor/__init__.py +++ b/instructor/__init__.py @@ -8,7 +8,7 @@ from .dsl import ( openai_moderation, ) from .function_calls import OpenAISchema, openai_schema, Mode -from .patch import apatch, patch +from .patch import apatch, patch, handle_parallel_model, handle_response_model __all__ = [ "OpenAISchema", @@ -24,5 +24,6 @@ __all__ = [ "openai_moderation", "FinetuneFormat", "Instructions", - "unpatch", + "handle_parallel_model", + "handle_response_model", ]