beamlit 0.0.34rc63__py3-none-any.whl → 0.0.34rc65__py3-none-any.whl

beamlit/agents/chat.py

@@ -6,6 +6,7 @@ from langchain_core.language_models import BaseChatModel
 from beamlit.authentication import get_authentication_headers, new_client
 from beamlit.common.settings import get_settings
 from beamlit.models import Model
+from beamlit.api.models import get_model
 
 logger = getLogger(__name__)
 
@@ -46,6 +47,12 @@ def get_chat_model(name: str, agent_model: Union[Model, None] = None) -> Tuple[B
     settings = get_settings()
     client = new_client()
 
+    if agent_model is None:
+        try:
+            agent_model = get_model.sync(name, client=client, environment=settings.environment)
+        except Exception as e:
+            logger.warning(f"Model {name} not found, defaulting to gpt-4o-mini")
+
     environment = (agent_model and agent_model.metadata and agent_model.metadata.environment) or settings.environment
     headers = get_authentication_headers(settings)
     headers["X-Beamlit-Environment"] = environment

beamlit/agents/decorator.py

@@ -1,5 +1,6 @@
 # Import necessary modules
 import functools
+import inspect
 from logging import getLogger
 
 from langgraph.checkpoint.memory import MemorySaver
@@ -34,15 +35,27 @@ def agent(
         settings = init()
 
         def wrapper(func):
+            agent_kwargs = any(
+                param.name == "agent"
+                for param in inspect.signature(func).parameters.values()
+            )
+            model_kwargs = any(
+                param.name == "model"
+                for param in inspect.signature(func).parameters.values()
+            )
+            functions_kwargs = any(
+                param.name == "functions"
+                for param in inspect.signature(func).parameters.values()
+            )
             @functools.wraps(func)
             def wrapped(*args, **kwargs):
-                return func(
-                    settings.agent.agent,
-                    settings.agent.chat_model,
-                    settings.agent.functions,
-                    *args,
-                    **kwargs,
-                )
+                if agent_kwargs:
+                    kwargs["agent"] = settings.agent.agent
+                if model_kwargs:
+                    kwargs["model"] = settings.agent.chat_model
+                if functions_kwargs:
+                    kwargs["functions"] = settings.agent.functions
+                return func(*args, **kwargs)
 
             return wrapped
 
@@ -83,21 +96,19 @@ def agent(
             remote_functions=remote_functions,
             chain=agent.spec.agent_chain,
             remote_functions_empty=not remote_functions,
+            warning=chat_model is not None,
         )
         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 "
-                f'"{settings.agent.functions_directory}". Here is a sample function you can use:\n\n'
-                "from beamlit.functions import function\n\n"
-                "@function()\n"
-                "def hello_world(query: str):\n"
-                "    return 'Hello, world!'\n"
-            )
 
         if override_agent is None and chat_model is not None:
             memory = MemorySaver()
+            if len(functions) == 0:
+                raise ValueError("You can define this function in directory "
+                    f'"{settings.agent.functions_directory}". Here is a sample function you can use:\n\n'
+                    "from beamlit.functions import function\n\n"
+                    "@function()\n"
+                    "def hello_world(query: str):\n"
+                    "    return 'Hello, world!'\n")
             _agent = create_react_agent(chat_model, functions, checkpointer=memory)
             settings.agent.agent = _agent
         else:

beamlit/functions/decorator.py

@@ -30,6 +30,7 @@ def get_functions(
     chain:Union[list[AgentChain], None]=None,
     remote_functions_empty:bool=True,
     from_decorator:str="function",
+    warning:bool=True,
 ):
     from beamlit.agents.chain import ChainToolkit
 
@@ -45,7 +46,7 @@ def get_functions(
 
     # Walk through all Python files in functions directory and subdirectories
     if not os.path.exists(dir):
-        if remote_functions_empty:
+        if remote_functions_empty and warning:
             logger.warn(f"Functions directory {dir} not found")
         return []
     for root, _, files in os.walk(dir):

{beamlit-0.0.34rc63.dist-info → beamlit-0.0.34rc65.dist-info}/METADATA

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

{beamlit-0.0.34rc63.dist-info → beamlit-0.0.34rc65.dist-info}/RECORD

@@ -6,8 +6,8 @@ beamlit/run.py,sha256=HtDYDjD7oVfQ8r3T5_t4qN5UDJOJfsQILi45Z21ArAg,1446
 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=4-MDbQ5Csrj0fjgdJo0EMf0tujkOWUYBHCEjyBqaeYc,3766
-beamlit/agents/decorator.py,sha256=YU8FHrn9bYfgKuGtt42iPzOqiVvWC9GzbFLmhTacIsU,4183
+beamlit/agents/chat.py,sha256=b7-ZkRqgS7nxFLbh_fOf9riAoPYnlb9_vHxaf2xYuvI,4057
+beamlit/agents/decorator.py,sha256=alz0ZPki6BYH4qQRfF5i4faPQV96cr5Q_VTn3DMeQ8Y,4743
 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
@@ -141,7 +141,7 @@ beamlit/deploy/deploy.py,sha256=h8ugvkSGcMKLqM-_uD5BZUzng5tJRp-Gea1Jup6QzTE,1025
 beamlit/deploy/format.py,sha256=U6UZEFAYLnGJJ7O2YmSdlUUFhnWNGAv6NZ-DW4KTgvI,2049
 beamlit/deploy/parser.py,sha256=Ga0poCZkoRnuTw082QnTcNGCBJncoRAnVsn8-1FsaJE,6907
 beamlit/functions/__init__.py,sha256=NcQPZZNfWhAJ1T1F6Xn21LFPMbZ7aMR2Sve3uZOkBCQ,170
-beamlit/functions/decorator.py,sha256=WkNF4-Y6XC9m-W9LRDJpzQEruQf98FxLpdY-Tc1VCUE,8679
+beamlit/functions/decorator.py,sha256=S9trFiDvLKlN-2f8SXccS44AxyCMX1blYhW_HJ12YXc,8714
 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
@@ -262,6 +262,6 @@ beamlit/serve/app.py,sha256=_aG2UVQ3Y85rUW3ehu9TlzLnowkfh54IIz558ftqOMw,3638
 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.34rc63.dist-info/METADATA,sha256=5yT2O9_PcJv8YSEgI8j46uT5f40N1oCx-I0fhj9zPdg,2412
-beamlit-0.0.34rc63.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.34rc63.dist-info/RECORD,,
+beamlit-0.0.34rc65.dist-info/METADATA,sha256=-Xa2kRZiGR3uVjLBH5SJHB2Xd8SNs40qezqTb2pPea8,2412
+beamlit-0.0.34rc65.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.34rc65.dist-info/RECORD,,