beamlit 0.0.24rc18__py3-none-any.whl → 0.0.24rc20__py3-none-any.whl

beamlit/agents/decorator.py

@@ -1,6 +1,7 @@
 # Import necessary modules
 import ast
 import asyncio
+import functools
 import importlib
 import os
 from logging import getLogger
@@ -94,6 +95,7 @@ def agent(
     def wrapper(func):
         settings = get_settings()
 
+        @functools.wraps(func)
         def wrapped(*args, **kwargs):
             return func(
                 settings.agent.agent,

beamlit/functions/decorator.py

@@ -1,5 +1,6 @@
 """Decorators for creating function tools with Beamlit and LangChain integration."""
 
+import json
 from collections.abc import Callable
 from logging import getLogger
 
@@ -30,17 +31,21 @@ def get_remote_function(func: Callable, bl_function: FunctionDeployment):
                 resource_name=name,
                 environment=settings.environment,
                 method="POST",
-                headers={},
-                json=kwargs,
+                headers={"Content-Type": "application/json"},
+                data=json.dumps(kwargs),
             )
+            content = response.text
             if response.status_code >= 400:
                 content = f"{response.status_code}:{response.text}"
+                logger.error(f"Error calling remote function: {content}")
                 return f"Error calling remote function: {content}"
             logger.debug(
                 f"Response from remote function: NAME={name}"
-                f" RESPONSE={response.json()} ENVIRONMENT={settings.environment}"
+                f" RESPONSE={content} ENVIRONMENT={settings.environment}"
             )
-            return response.json()
+            if response.headers.get("content-type") == "application/json":
+                return response.json()
+            return content
         except Exception as e:
             logger.error(f"Error calling function {bl_function.id}: {e}")
             raise e
@@ -83,3 +88,4 @@ def function(
         return func
 
     return wrapper
+    return wrapper

{beamlit-0.0.24rc18.dist-info → beamlit-0.0.24rc20.dist-info}/METADATA

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

{beamlit-0.0.24rc18.dist-info → beamlit-0.0.24rc20.dist-info}/RECORD

@@ -6,7 +6,7 @@ beamlit/run.py,sha256=y61iDBaR0917ihj5q-cJ_r3BFW1Rn5K_kDAISw5O6aU,1339
 beamlit/types.py,sha256=E1hhDh_zXfsSQ0NCt9-uw90_Mr5iIlsdfnfvxv5HarU,1005
 beamlit/agents/__init__.py,sha256=nf1iwQwGtCG6nDqyVhxfWoqR6dv6X3bvSpCeqkTCFaM,101
 beamlit/agents/chat.py,sha256=aI7pObyywRyg3dBpubzHAUWTbTk1nwtxvpY7iIP1RLY,2704
-beamlit/agents/decorator.py,sha256=XDrWN-Xx4czSy1sIkfQY4t9-UnsM21vMll58CP1TRLw,6468
+beamlit/agents/decorator.py,sha256=VOiCUpGyzXgf2-KRmp4IMA6Y-CcL1NZEax5kXLGt8GY,6516
 beamlit/api/__init__.py,sha256=zTSiG_ujSjAqWPyc435YXaX9XTlpMjiJWBbV-f-YtdA,45
 beamlit/api/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 beamlit/api/agents/create_agent.py,sha256=HFExosu02JZqZz7I6U6WjN81TERz6p2i8CzQCyiRYXo,4112
@@ -143,7 +143,7 @@ beamlit/common/utils.py,sha256=jouz5igBvT37Xn_e94-foCHyQczVim-UzVcoIF6RWJ4,657
 beamlit/deploy/__init__.py,sha256=GS7l7Jtm2yKs7iNLKcfjYO-rAhUzggQ3xiYSf3oxLBY,91
 beamlit/deploy/deploy.py,sha256=bWUXG_5b_XhFcG5t75vel2EbEfZPIdg_YLXtzkfgXUs,18799
 beamlit/functions/__init__.py,sha256=_RPG1Bfg54JGdIPnViAU6n9zD7E1cDNsdXi8oYGskzE,138
-beamlit/functions/decorator.py,sha256=uYZOVxD-7ZNHORTQfCIn0qdNKPIZupsr7_QdUmWEKu0,2996
+beamlit/functions/decorator.py,sha256=EnkvWvmJTFAbTzsoJZxWUemwv3j0FqyGlGmMhjrsp6c,3281
 beamlit/functions/github/__init__.py,sha256=gYnUkeegukOfbymdabuuJkScvH-_ZJygX05BoqkPn0o,49
 beamlit/functions/github/github.py,sha256=FajzLCNkpXcwfgnC0l9rOGT2eSPLCz8-qrMzK9N_ZNc,598
 beamlit/functions/github/kit/__init__.py,sha256=jBwPqZv6C23_utukohxqXZwrlicNlI7PYPUj0Den7Cw,136
@@ -298,6 +298,6 @@ beamlit/serve/app.py,sha256=Xoc_A-e9_4U9OZKoKXSc6qc_NTRPuVb6ddAZVENhd34,2617
 beamlit/serve/middlewares/__init__.py,sha256=1dVmnOmhAQWvWktqHkKSIX-YoF6fmMU8xkUQuhg_rJU,148
 beamlit/serve/middlewares/accesslog.py,sha256=wM52-hcwtO-_hdM1pnsEJzerzJf1MzEyN5m85BdDccE,609
 beamlit/serve/middlewares/processtime.py,sha256=lDAaIasZ4bwvN-HKHvZpaD9r-yrkVNZYx4abvbjbrCg,411
-beamlit-0.0.24rc18.dist-info/METADATA,sha256=0YDdLTLn8yDmAwzK9Mp4ubyUVklAvqF8O5_0T3wj_l4,2027
-beamlit-0.0.24rc18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.24rc18.dist-info/RECORD,,
+beamlit-0.0.24rc20.dist-info/METADATA,sha256=4MNBJTmEDnw74EEhTDiNOSbFwODc-BM1tarQjF32bFs,2027
+beamlit-0.0.24rc20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.24rc20.dist-info/RECORD,,