🐛 Keep a reference to app in client

vortico · Sep 20, 2023 · 3e12f31 · 3e12f31
1 parent 880e6b2
commit 3e12f31
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/flama/client.py b/flama/client.py
@@ -108,6 +108,7 @@ def __init__(
             self.models = {m[0]: m[1] for m in models or {}}
 
         self.lifespan = LifespanContextManager(app) if app else None
+        self.app = app
 
         kwargs["app"] = app
         kwargs.setdefault("base_url", "http://localapp")

diff --git a/tests/test_client.py b/tests/test_client.py
@@ -155,6 +155,7 @@ def test_init_models(self, app):
         ):
             client = _BaseClient(models=[("foo", "/foo/", "model_foo.flm"), ("bar", "/bar/", "model_bar.flm")])
 
+        assert client.app == app
         assert client.lifespan
         assert client.lifespan.app == app
         assert client.models == {"foo": "/foo/", "bar": "/bar/"}