h2o-wave 1.4.0__py3-none-any.whl → 1.5.1__py3-none-any.whl

h2o_wave/core.py

@@ -43,6 +43,12 @@ BROADCAST = 'broadcast'
 
 def _get_env(key: str, value: Any):
     return os.environ.get(f'H2O_WAVE_{key}', value)
+def _get_timeout(val: str) -> Optional[int]:
+    try:
+        num = int(val)
+        return num if num >= 0 else None
+    except ValueError:
+        raise ValueError(f'Timeout must be numeric, got {val}')
 
 
 _base_url = _get_env('BASE_URL', '/')
@@ -60,6 +66,10 @@ class _Config:
         self.hub_access_key_secret: str = _get_env('ACCESS_KEY_SECRET', 'access_key_secret')
         self.app_access_key_id: str = _get_env('APP_ACCESS_KEY_ID', None) or secrets.token_urlsafe(16)
         self.app_access_key_secret: str = _get_env('APP_ACCESS_KEY_SECRET', None) or secrets.token_urlsafe(16)
+        self.app_connect_timeout: Optional[int] = _get_timeout(_get_env('APP_CONNECT_TIMEOUT', '5'))
+        self.app_read_timeout: Optional[int] = _get_timeout(_get_env('APP_READ_TIMEOUT', '5'))
+        self.app_write_timeout: Optional[int] = _get_timeout(_get_env('APP_WRITE_TIMEOUT', '5'))
+        self.app_pool_timeout: Optional[int] = _get_timeout(_get_env('APP_POOL_TIMEOUT', '5'))
 
 
 _config = _Config()
@@ -640,9 +650,16 @@ class Site:
     """
 
     def __init__(self):
+        timeout = httpx.Timeout(
+            connect=_config.app_connect_timeout,
+            read=_config.app_read_timeout,
+            write=_config.app_write_timeout,
+            pool=_config.app_pool_timeout,
+        )
         self._http = httpx.Client(
             auth=(_config.hub_access_key_id, _config.hub_access_key_secret),
             verify=False,
+            timeout=timeout,
         )
         self.cache = _ServerCache(self._http)
 
@@ -870,9 +887,16 @@ class AsyncSite:
     """
 
     def __init__(self):
+        timeout = httpx.Timeout(
+            connect=_config.app_connect_timeout,
+            read=_config.app_read_timeout,
+            write=_config.app_write_timeout,
+            pool=_config.app_pool_timeout,
+        )
         self._http = httpx.AsyncClient(
             auth=(_config.hub_access_key_id, _config.hub_access_key_secret),
             verify=False,
+            timeout=timeout,
         )
         self.cache = _AsyncServerCache(self._http)
 

h2o_wave/server.py

@@ -77,7 +77,13 @@ class Auth:
         if not self.refresh_token:
             return
 
-        async with httpx.AsyncClient(auth=(_config.hub_access_key_id, _config.hub_access_key_secret), verify=False) as http:
+        timeout = httpx.Timeout(
+            connect=_config.app_connect_timeout,
+            read=_config.app_read_timeout,
+            write=_config.app_write_timeout,
+            pool=_config.app_pool_timeout,
+        )
+        async with httpx.AsyncClient(auth=(_config.hub_access_key_id, _config.hub_access_key_secret), verify=False, timeout=timeout) as http:
             res = await http.get(_config.hub_address + '_auth/refresh', headers={'Wave-Session-ID': self._session_id})
             return self.__extract_tokens(res.headers)
 
@@ -89,7 +95,13 @@ class Auth:
         if not self.refresh_token:
             return
 
-        with httpx.Client(auth=(_config.hub_access_key_id, _config.hub_access_key_secret), verify=False) as http:
+        timeout = httpx.Timeout(
+            connect=_config.app_connect_timeout,
+            read=_config.app_read_timeout,
+            write=_config.app_write_timeout,
+            pool=_config.app_pool_timeout,
+        )
+        with httpx.Client(auth=(_config.hub_access_key_id, _config.hub_access_key_secret), verify=False, timeout=timeout) as http:
             res = http.get(_config.hub_address + '_auth/refresh', headers={'Wave-Session-ID': self._session_id})
             return self.__extract_tokens(res.headers)
 
@@ -215,9 +227,16 @@ WebAppState = Tuple[Expando, Dict[str, Expando], Dict[str, Expando]]
 
 class _Wave:
     def __init__(self):
+        timeout = httpx.Timeout(
+            connect=_config.app_connect_timeout,
+            read=_config.app_read_timeout,
+            write=_config.app_write_timeout,
+            pool=_config.app_pool_timeout,
+        )
         self._http = httpx.AsyncClient(
             auth=(_config.hub_access_key_id, _config.hub_access_key_secret),
             verify=False,
+            timeout=timeout,
         )
 
     async def call(self, method: str, **kwargs):

h2o_wave/types.py

@@ -8467,6 +8467,7 @@ class ChatbotCard:
             generating: Optional[bool] = None,
             suggestions: Optional[List[ChatSuggestion]] = None,
             disabled: Optional[bool] = None,
+            value: Optional[str] = None,
             commands: Optional[List[Command]] = None,
     ):
         _guard_scalar('ChatbotCard.box', box, (str,), False, False, False)
@@ -8476,6 +8477,7 @@ class ChatbotCard:
         _guard_scalar('ChatbotCard.generating', generating, (bool,), False, True, False)
         _guard_vector('ChatbotCard.suggestions', suggestions, (ChatSuggestion,), False, True, False)
         _guard_scalar('ChatbotCard.disabled', disabled, (bool,), False, True, False)
+        _guard_scalar('ChatbotCard.value', value, (str,), False, True, False)
         _guard_vector('ChatbotCard.commands', commands, (Command,), False, True, False)
         self.box = box
         """A string indicating how to place this component on the page."""
@@ -8493,6 +8495,8 @@ class ChatbotCard:
         """Clickable prompt suggestions shown below the last response."""
         self.disabled = disabled
         """True if the user input should be disabled."""
+        self.value = value
+        """Value of the user input."""
         self.commands = commands
         """Contextual menu commands for this component."""
 
@@ -8505,6 +8509,7 @@ class ChatbotCard:
         _guard_scalar('ChatbotCard.generating', self.generating, (bool,), False, True, False)
         _guard_vector('ChatbotCard.suggestions', self.suggestions, (ChatSuggestion,), False, True, False)
         _guard_scalar('ChatbotCard.disabled', self.disabled, (bool,), False, True, False)
+        _guard_scalar('ChatbotCard.value', self.value, (str,), False, True, False)
         _guard_vector('ChatbotCard.commands', self.commands, (Command,), False, True, False)
         return _dump(
             view='chatbot',
@@ -8516,6 +8521,7 @@ class ChatbotCard:
             generating=self.generating,
             suggestions=None if self.suggestions is None else [__e.dump() for __e in self.suggestions],
             disabled=self.disabled,
+            value=self.value,
             commands=None if self.commands is None else [__e.dump() for __e in self.commands],
         )
 
@@ -8537,6 +8543,8 @@ class ChatbotCard:
         _guard_vector('ChatbotCard.suggestions', __d_suggestions, (dict,), False, True, False)
         __d_disabled: Any = __d.get('disabled')
         _guard_scalar('ChatbotCard.disabled', __d_disabled, (bool,), False, True, False)
+        __d_value: Any = __d.get('value')
+        _guard_scalar('ChatbotCard.value', __d_value, (str,), False, True, False)
         __d_commands: Any = __d.get('commands')
         _guard_vector('ChatbotCard.commands', __d_commands, (dict,), False, True, False)
         box: str = __d_box
@@ -8547,6 +8555,7 @@ class ChatbotCard:
         generating: Optional[bool] = __d_generating
         suggestions: Optional[List[ChatSuggestion]] = None if __d_suggestions is None else [ChatSuggestion.load(__e) for __e in __d_suggestions]
         disabled: Optional[bool] = __d_disabled
+        value: Optional[str] = __d_value
         commands: Optional[List[Command]] = None if __d_commands is None else [Command.load(__e) for __e in __d_commands]
         return ChatbotCard(
             box,
@@ -8557,6 +8566,7 @@ class ChatbotCard:
             generating,
             suggestions,
             disabled,
+            value,
             commands,
         )
 

h2o_wave/ui.py

@@ -2938,6 +2938,7 @@ def chatbot_card(
         generating: Optional[bool] = None,
         suggestions: Optional[List[ChatSuggestion]] = None,
         disabled: Optional[bool] = None,
+        value: Optional[str] = None,
         commands: Optional[List[Command]] = None,
 ) -> ChatbotCard:
     """Create a chatbot card to allow getting prompts from users and providing them with LLM generated answers.
@@ -2951,6 +2952,7 @@ def chatbot_card(
         generating: True to show a button to stop the text generation. Defaults to False.
         suggestions: Clickable prompt suggestions shown below the last response.
         disabled: True if the user input should be disabled.
+        value: Value of the user input.
         commands: Contextual menu commands for this component.
     Returns:
         A `h2o_wave.types.ChatbotCard` instance.
@@ -2964,6 +2966,7 @@ def chatbot_card(
         generating,
         suggestions,
         disabled,
+        value,
         commands,
     )
 

h2o_wave/version.py

@@ -1 +1 @@
-__version__ = '1.4.0'
+__version__ = '1.5.1'

{h2o_wave-1.4.0.dist-info → h2o_wave-1.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: h2o-wave
-Version: 1.4.0
+Version: 1.5.1
 Summary: Python driver for H2O Wave Realtime Apps
 Project-URL: Homepage, https://wave.h2o.ai/
 Project-URL: Documentation, https://wave.h2o.ai/

{h2o_wave-1.4.0.dist-info → h2o_wave-1.5.1.dist-info}/RECORD

@@ -1,22 +1,22 @@
 h2o_wave/__init__.py,sha256=XIclw-HLtXgr0wNf5eQpkh6ZiqD9QAoiS-vZhp3IR5k,1674
 h2o_wave/__main__.py,sha256=MoNOW43ppIqCdY3iq0n25Q3SKLyk8Igg5fD_sSROK4c,638
 h2o_wave/cli.py,sha256=5pIWZe-lRNmE1QUc94mk4XLLaQquvnCWwSs4qMgO3p4,13681
-h2o_wave/core.py,sha256=whFb-Z2wokNXe_39I8aL3i7JOx6eczRfWOkn5wb_YwM,39087
+h2o_wave/core.py,sha256=dKUZOPDL8W-MCDLtC3A1TTuGNGO9cjAtbaQ8dfjceUA,40184
 h2o_wave/db.py,sha256=H3W_EyMfnwr4UjqPVoAsE19O2QzY1ptIYGMOqU0YUQo,7489
 h2o_wave/graphics.py,sha256=HLYrX-lwsMKbyLmy2ClG5L46DA2_hSCEPTsv0gPVoyg,25866
 h2o_wave/ide.py,sha256=wAJjfEI1ekQrL455FGRVeNi3KiH6ELgsG3qm31Q-kRs,6810
 h2o_wave/metadata.py,sha256=d8FrJ-nQhBIvntA2W9VVZ0-lxEE_uy85ShxCC_FFDEs,82
 h2o_wave/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h2o_wave/routing.py,sha256=de8GVfUAb6bwFXtsWj6NXmjMVGELknlZb03F-R4ManY,10592
-h2o_wave/server.py,sha256=nkCbmUtT3LvNe54ldN75nWprXn9y9BguH4Fqc8khx84,17974
+h2o_wave/server.py,sha256=eAVP8E_fqpeAwYtZjk9cGKSy7cLwGbh0Zj_WBrB1j5M,18706
 h2o_wave/share.py,sha256=2zgywet8540O6xM-JD3po1glyP2PBJ3lIxaWBJbQvtQ,1164
 h2o_wave/test.py,sha256=hF_fS5e25bACnzENjpDrikv7nPOs0iENh4MuXX9BaVA,2738
-h2o_wave/types.py,sha256=BJ0fQaay4OydjNtgdfhfBV7OeIVcIf-C5tVvJ7C0b8Y,659661
-h2o_wave/ui.py,sha256=oAeKO5Ec128Eo9s4Mtdv1E0QgQv2_O2pRkQ3Sb8oBe8,172281
+h2o_wave/types.py,sha256=9aW8RCgA4I8Euddm_20UxMm8JsQJVebIOs9TLcck0fM,660143
+h2o_wave/ui.py,sha256=SFGRSfR_TWSLVRnIqKBxNqqHg4gcpXcRwM2JZwmxZbw,172373
 h2o_wave/ui_ext.py,sha256=zx_2Ec2-p_ztm8brfVaVF0fTQWVDrb_YxcGfVb-wA10,2325
-h2o_wave/version.py,sha256=EyMGX1ADFzN6XVXHWbJUtKPONYKeFkvWoKIFPDDB2I8,22
-h2o_wave-1.4.0.dist-info/METADATA,sha256=31B32gFSe0sQ0ZFepDwwsLndh1C9UxVz1Iqzrd168oU,2907
-h2o_wave-1.4.0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-h2o_wave-1.4.0.dist-info/entry_points.txt,sha256=kFeXNqSZlW1_H7YcRdSOhz5V00F4vhDQ0NuDuvRwLGw,43
-h2o_wave-1.4.0.dist-info/licenses/LICENSE,sha256=hpuFayniDwysSKD0tHGELH2KJDVyhUrKS29torRIpqY,53
-h2o_wave-1.4.0.dist-info/RECORD,,
+h2o_wave/version.py,sha256=cXu5Qklt-TkM22YbYOO8jfr6COB8p2hFc2AtZc9K1Ns,22
+h2o_wave-1.5.1.dist-info/METADATA,sha256=zmJfVRPTutWjDKj2rVZfDLS4ZcWkRtSV5nJ4OOnGn5s,2907
+h2o_wave-1.5.1.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+h2o_wave-1.5.1.dist-info/entry_points.txt,sha256=kFeXNqSZlW1_H7YcRdSOhz5V00F4vhDQ0NuDuvRwLGw,43
+h2o_wave-1.5.1.dist-info/licenses/LICENSE,sha256=hpuFayniDwysSKD0tHGELH2KJDVyhUrKS29torRIpqY,53
+h2o_wave-1.5.1.dist-info/RECORD,,