beamlit 0.0.33rc43__tar.gz → 0.0.33rc44__tar.gz

{beamlit-0.0.33rc43 → beamlit-0.0.33rc44}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.33rc43
+Version: 0.0.33rc44
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12

{beamlit-0.0.33rc43 → beamlit-0.0.33rc44}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beamlit"
-version = "0.0.33.pre-43"
+version = "0.0.33.pre-44"
 description = "Add your description here"
 readme = "README.md"
 authors = [

{beamlit-0.0.33rc43 → beamlit-0.0.33rc44}/src/beamlit/functions/decorator.py

@@ -3,7 +3,9 @@
 import json
 from collections.abc import Callable
 from logging import getLogger
+import functools
 
+from fastapi import Request
 from langchain_core.tools import create_schema_from_function
 
 from beamlit.authentication import new_client
@@ -77,7 +79,7 @@ def function(*args, function: Function | dict = None, kit=False, **kwargs: dict)
         )
     if isinstance(function, dict):
         function = Function(**function)
-        
+
     def wrapper(func: Callable) -> Callable:
         if function and not func.__doc__ and function.spec and function.spec.description:
             func.__doc__ = function.spec.description
@@ -90,6 +92,13 @@ def function(*args, function: Function | dict = None, kit=False, **kwargs: dict)
                     parse_docstring=func.__doc__,
                 )
             return remote_func
-        return func
+        
+        @functools.wraps(func)
+        async def wrapped(*args, **kwargs):
+            if isinstance(args[0], Request):
+                body = await args[0].json()
+                args = [body.get(param) for param in func.__code__.co_varnames[:func.__code__.co_argcount]]
+            return func(*args, **kwargs)
+        return wrapped
 
     return wrapper

{beamlit-0.0.33rc43 → beamlit-0.0.33rc44}/src/beamlit/serve/app.py

@@ -71,13 +71,11 @@ async def root(request: Request):
     settings = get_settings()
     logger = getLogger(__name__)
     try:
-        body = await request.json()
-
         original_func = getattr(func, "__wrapped__", func)
         if asyncio.iscoroutinefunction(func) or asyncio.iscoroutinefunction(original_func):
-            response = await func(body)
+            response = await func(request)
         else:
-            response = func(body)
+            response = func(request)
 
         if isinstance(response, Response):
             return response