From 91400007066630c85c385aba744b7d508071f5da Mon Sep 17 00:00:00 2001
From: Nolan Tremelling <34580718+NolanTrem@users.noreply.github.com>
Date: Thu, 20 Feb 2025 14:40:04 -0800
Subject: [PATCH] Add models to Python SDK (#1994)

---
 py/r2r/__init__.py    | 1 +
 py/shared/__init__.py | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/py/r2r/__init__.py b/py/r2r/__init__.py
index 6def0bc78..95c2c7c8a 100644
--- a/py/r2r/__init__.py
+++ b/py/r2r/__init__.py
@@ -3,6 +3,7 @@
 from sdk.async_client import R2RAsyncClient
 from sdk.models import R2RException
 from sdk.sync_client import R2RClient
+from shared import *
 
 __version__ = metadata.version("r2r")
 
diff --git a/py/shared/__init__.py b/py/shared/__init__.py
index e69de29bb..ae2a3b41d 100644
--- a/py/shared/__init__.py
+++ b/py/shared/__init__.py
@@ -0,0 +1,6 @@
+from .abstractions import *
+from .abstractions import __all__ as abstractions_all
+from .api.models import *
+from .utils import *
+
+__all__ = abstractions_all