beamlit 0.0.33rc50__py3-none-any.whl → 0.0.34rc52__py3-none-any.whl

beamlit/agents/decorator.py

@@ -6,9 +6,10 @@ import importlib
 import os
 from logging import getLogger
 
-from langchain_core.tools import Tool
+from langchain_core.tools import StructuredTool
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.prebuilt import create_react_agent
+from langchain_core.tools.base import create_schema_from_function
 
 from beamlit.api.models import get_model
 from beamlit.authentication import new_client
@@ -94,19 +95,21 @@ def get_functions(client, dir="src/functions", from_decorator="function", remote
                                     else:
                                         if asyncio.iscoroutinefunction(func):
                                             functions.append(
-                                                Tool(
+                                                StructuredTool(
                                                     name=func.__name__,
                                                     description=func.__doc__,
                                                     func=func,
                                                     coroutine=func,
+                                                    args_schema=create_schema_from_function(func.__name__, func)
                                                 )
                                             )
                                         else:
                                             functions.append(
-                                                Tool(
+                                                StructuredTool(
                                                     name=func.__name__,
                                                     description=func.__doc__,
                                                     func=func,
+                                                    args_schema=create_schema_from_function(func.__name__, func)
                                                 )
                                             )
                     except Exception as e:
@@ -152,7 +155,7 @@ def agent(
             remote_functions_empty=not remote_functions,
         )
 
-        settings.agent.functions = functions
+        
 
         if agent is not None:
             metadata = AgentMetadata(**agent.get("metadata", {}))
@@ -208,6 +211,8 @@ def agent(
             toolkit.initialize()
             functions.extend(toolkit.get_tools())
 
+        settings.agent.functions = functions
+
         if override_agent is None and len(functions) == 0:
             raise ValueError(
                 "You must define at least one function, you can define this function in directory "

{beamlit-0.0.33rc50.dist-info → beamlit-0.0.34rc52.dist-info}/METADATA

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

{beamlit-0.0.33rc50.dist-info → beamlit-0.0.34rc52.dist-info}/RECORD

@@ -7,7 +7,7 @@ beamlit/types.py,sha256=E1hhDh_zXfsSQ0NCt9-uw90_Mr5iIlsdfnfvxv5HarU,1005
 beamlit/agents/__init__.py,sha256=nf1iwQwGtCG6nDqyVhxfWoqR6dv6X3bvSpCeqkTCFaM,101
 beamlit/agents/chain.py,sha256=vfCjiFHuu02uTTGicxMlFzjyICQkIjpXrBGs-7uJEsg,2826
 beamlit/agents/chat.py,sha256=gVyv4FGBdQTDhdutX8l64OUNa6Fdqaw4eCfEDRH0IPQ,3558
-beamlit/agents/decorator.py,sha256=PUw5xfG4KmmVeSbgI-Ac7KAm5n0addPhg5MysfxLpWk,10297
+beamlit/agents/decorator.py,sha256=7w2rAo0Kg280lpo67byF5vor4qqMtlhuRqziwrSA62s,10629
 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=t5Pr62My2EhQlcIY71MrI73-0_q5Djr3a_Ybt9MIiQQ,3587
@@ -256,6 +256,6 @@ beamlit/serve/app.py,sha256=DXWxQoMeuA5FYvBMyLrP94OEWQbwLf4GZk3I9fkwSPA,3523
 beamlit/serve/middlewares/__init__.py,sha256=1dVmnOmhAQWvWktqHkKSIX-YoF6fmMU8xkUQuhg_rJU,148
 beamlit/serve/middlewares/accesslog.py,sha256=Mu4T4_9OvHybjA0ApzZFpgi2C8f3X1NbUk-76v634XM,631
 beamlit/serve/middlewares/processtime.py,sha256=lDAaIasZ4bwvN-HKHvZpaD9r-yrkVNZYx4abvbjbrCg,411
-beamlit-0.0.33rc50.dist-info/METADATA,sha256=EwI_0qBXTM0HfSJ-k7w8d37h2-X4ZJ0VucVWV8iKtVU,2405
-beamlit-0.0.33rc50.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.33rc50.dist-info/RECORD,,
+beamlit-0.0.34rc52.dist-info/METADATA,sha256=5z-4coIVveGscH6kroJEgJ3SVl1C9nEOkmRV5T38zaU,2405
+beamlit-0.0.34rc52.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.34rc52.dist-info/RECORD,,