futurehouse-client 0.3.18.dev109__py3-none-any.whl → 0.3.18.dev110__py3-none-any.whl

futurehouse_client/utils/__init__.py

@@ -1,3 +1,7 @@
-from .context import UserContext
+from .context import JOB_ID_CONTEXT_KEY, USER_JWT_CONTEXT_KEY, RequestContext
 
-__all__ = ["UserContext"]
+__all__ = [
+    "JOB_ID_CONTEXT_KEY",
+    "USER_JWT_CONTEXT_KEY",
+    "RequestContext",
+]

futurehouse_client/utils/context.py

@@ -1,16 +1,45 @@
-class UserContext:
-    """A context manager for storing user information from the initial request."""
+from typing import Any, ClassVar
 
-    _user_jwt = None
+USER_JWT_CONTEXT_KEY = "user_jwt"
+JOB_ID_CONTEXT_KEY = "job_id"
+
+
+class RequestContext:
+    """A context manager for storing information from the initial request."""
+
+    _context: ClassVar[dict[str, Any]] = {}
 
     @classmethod
-    def set_user_jwt(cls, jwt: str) -> None:
-        cls._user_jwt = jwt
+    def set(cls, key: str, value: Any) -> None:
+        """Set a context variable.
+
+        Args:
+            key: The context variable name
+            value: The value to store
+        """
+        cls._context[key] = value
 
     @classmethod
-    def get_user_jwt(cls) -> str | None:
-        return cls._user_jwt
+    def get(cls, key: str) -> Any:
+        """Get a context variable.
+
+        Args:
+            key: The context variable name
+            default: Default value if key doesn't exist
+
+        Returns:
+            The stored value or default if not found
+        """
+        return cls._context.get(key, None)
 
     @classmethod
-    def clear_user_jwt(cls) -> None:
-        cls._user_jwt = None
+    def clear(cls, key: str | None = None) -> None:
+        """Clear a specific context variable or all variables.
+
+        Args:
+            key: Specific key to clear, or None to clear all
+        """
+        if key is None:
+            cls._context.clear()
+        elif key in cls._context:
+            del cls._context[key]

{futurehouse_client-0.3.18.dev109.dist-info → futurehouse_client-0.3.18.dev110.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.18.dev109
+Version: 0.3.18.dev110
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 Classifier: Operating System :: OS Independent

{futurehouse_client-0.3.18.dev109.dist-info → futurehouse_client-0.3.18.dev110.dist-info}/RECORD

@@ -6,12 +6,12 @@ futurehouse_client/models/__init__.py,sha256=ta3jFLM_LsDz1rKDmx8rja8sT7WtSKoFvMg
 futurehouse_client/models/app.py,sha256=yfZ9tyw4VATVAfYrU7aTdCNPSljLEho09_nIbh8oZDY,23174
 futurehouse_client/models/client.py,sha256=n4HD0KStKLm6Ek9nL9ylP-bkK10yzAaD1uIDF83Qp_A,1828
 futurehouse_client/models/rest.py,sha256=lgwkMIXz0af-49BYSkKeS7SRqvN3motqnAikDN4YGTc,789
-futurehouse_client/utils/__init__.py,sha256=mCp1UP3eyrWyHpFAEHgKcazKCJ5g87MxlDIZVzi-5Tk,60
-futurehouse_client/utils/context.py,sha256=MdYYldQfsqxnGJUDcMLkbRtKHb2srHfcMf6aR-oZh6Y,387
+futurehouse_client/utils/__init__.py,sha256=Tlz5A-h2h2R8SZo7HXtNGNwDY08zCMWtE33PnSw5jjo,169
+futurehouse_client/utils/context.py,sha256=QN_-Z61IQtAUOZaBDoP5F6pqvrkMxZ7y-1t7LEuF6lI,1162
 futurehouse_client/utils/general.py,sha256=A_rtTiYW30ELGEZlWCIArO7q1nEmqi8hUlmBRYkMQ_c,767
 futurehouse_client/utils/module_utils.py,sha256=aFyd-X-pDARXz9GWpn8SSViUVYdSbuy9vSkrzcVIaGI,4955
 futurehouse_client/utils/monitoring.py,sha256=UjRlufe67kI3VxRHOd5fLtJmlCbVA2Wqwpd4uZhXkQM,8728
-futurehouse_client-0.3.18.dev109.dist-info/METADATA,sha256=yeDUNer_jX0nDUAgV26q5ssYrcWmWlCpuqy6iRuSz_c,12767
-futurehouse_client-0.3.18.dev109.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-futurehouse_client-0.3.18.dev109.dist-info/top_level.txt,sha256=TRuLUCt_qBnggdFHCX4O_BoCu1j2X43lKfIZC-ElwWY,19
-futurehouse_client-0.3.18.dev109.dist-info/RECORD,,
+futurehouse_client-0.3.18.dev110.dist-info/METADATA,sha256=DQlgzMDU6QxiO2hvu29l1K3EDgjipG2ew6vtsfFdlF8,12767
+futurehouse_client-0.3.18.dev110.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+futurehouse_client-0.3.18.dev110.dist-info/top_level.txt,sha256=TRuLUCt_qBnggdFHCX4O_BoCu1j2X43lKfIZC-ElwWY,19
+futurehouse_client-0.3.18.dev110.dist-info/RECORD,,