beamlit 0.0.24rc18__tar.gz → 0.0.24rc20__tar.gz

{beamlit-0.0.24rc18 → beamlit-0.0.24rc20}/PKG-INFO

@@ -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 → beamlit-0.0.24rc20}/pyproject.toml

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

{beamlit-0.0.24rc18 → beamlit-0.0.24rc20}/src/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-0.0.24rc18 → beamlit-0.0.24rc20}/src/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