From 3e053bdd8df69bbb2a850ee48c37747d16a02630 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kacper=20Kali=C5=84ski?=
 <47140412+KaQuMiQ@users.noreply.github.com>
Date: Mon, 2 Sep 2024 14:47:17 +0200
Subject: [PATCH] Fix instance cache

---
 pyproject.toml            | 2 +-
 src/draive/utils/cache.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 9f93f02..eb629b1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "draive"
 description = "Framework designed to simplify and accelerate the development of LLM-based applications."
-version = "0.28.3"
+version = "0.28.4"
 readme = "README.md"
 maintainers = [
   { name = "Kacper Kaliński", email = "kacper.kalinski@miquido.com" },
diff --git a/src/draive/utils/cache.py b/src/draive/utils/cache.py
index 716373a..2c0050e 100644
--- a/src/draive/utils/cache.py
+++ b/src/draive/utils/cache.py
@@ -233,11 +233,11 @@ def next_expire_time() -> float | None:
 
     def __get__(
         self,
-        instance: object,
+        instance: object | None,
         owner: type | None = None,
         /,
     ) -> Callable[Args, Coroutine[None, None, Result]]:
-        if owner is None:
+        if owner is None or instance is None:
             return self
 
         else: