beamlit 0.0.34rc62__py3-none-any.whl → 0.0.34rc64__py3-none-any.whl

beamlit/agents/chat.py

@@ -1,11 +1,12 @@
 from logging import getLogger
-from typing import Tuple
+from typing import Tuple, Union
 
 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__)
 
@@ -42,11 +43,17 @@ def get_cohere_chat_model(**kwargs):
 
     return ChatCohere(**kwargs)
 
-def get_chat_model(name: str, agent_model: Model) -> Tuple[BaseChatModel, str, str]:
+def get_chat_model(name: str, agent_model: Union[Model, None] = None) -> Tuple[BaseChatModel, str, str]:
     settings = get_settings()
     client = new_client()
 
-    environment = (agent_model.metadata and agent_model.metadata.environment) or settings.environment
+    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
 
@@ -87,7 +94,8 @@ def get_chat_model(name: str, agent_model: Model) -> Tuple[BaseChatModel, str, s
     }
 
     provider = (
-        agent_model.spec
+        agent_model
+        and agent_model.spec
         and agent_model.spec.runtime
         and agent_model.spec.runtime.type_
     )
@@ -96,7 +104,8 @@ def get_chat_model(name: str, agent_model: Model) -> Tuple[BaseChatModel, str, s
         provider = "openai"
 
     model = (
-        agent_model.spec
+        agent_model
+        and agent_model.spec
         and agent_model.spec.runtime
         and agent_model.spec.runtime.model
     )

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
 
@@ -85,7 +98,7 @@ def agent(
             remote_functions_empty=not remote_functions,
         )
         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/functions/decorator.py

@@ -12,7 +12,6 @@ from fastapi import Request
 from langchain_core.tools import StructuredTool
 from langchain_core.tools.base import create_schema_from_function
 
-from beamlit.agents.chain import ChainToolkit
 from beamlit.authentication import new_client
 from beamlit.client import AuthenticatedClient
 from beamlit.common import slugify
@@ -32,6 +31,8 @@ def get_functions(
     remote_functions_empty:bool=True,
     from_decorator:str="function",
 ):
+    from beamlit.agents.chain import ChainToolkit
+
     settings = get_settings()
     if client is None:
         client = new_client()
@@ -116,7 +117,7 @@ def get_functions(
                                                 )
                                             )
                                         else:
-                                            
+
                                             functions.append(
                                                 StructuredTool(
                                                     name=func.__name__,

{beamlit-0.0.34rc62.dist-info → beamlit-0.0.34rc64.dist-info}/METADATA

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

{beamlit-0.0.34rc62.dist-info → beamlit-0.0.34rc64.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=eSpLhZkfL8Llf7TZeDnRT7HH70buBJO1DOwKm9UNeHs,3675
-beamlit/agents/decorator.py,sha256=YU8FHrn9bYfgKuGtt42iPzOqiVvWC9GzbFLmhTacIsU,4183
+beamlit/agents/chat.py,sha256=b7-ZkRqgS7nxFLbh_fOf9riAoPYnlb9_vHxaf2xYuvI,4057
+beamlit/agents/decorator.py,sha256=JWz74ugL-walNXMths2517CE-LBDoeSf_wQeM8mCPa4,4768
 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=6Sc-ZX5g-p8GlmdI2MmTW-XhVa8eeMawmwSQlndTLCE,8718
+beamlit/functions/decorator.py,sha256=WkNF4-Y6XC9m-W9LRDJpzQEruQf98FxLpdY-Tc1VCUE,8679
 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.34rc62.dist-info/METADATA,sha256=a5Vc8440X2LFgBPnIwx0SqImcwaKO3_OaUVLfTl7K7M,2412
-beamlit-0.0.34rc62.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.34rc62.dist-info/RECORD,,
+beamlit-0.0.34rc64.dist-info/METADATA,sha256=Nc9HziVd7JjcSU0GAaVxtvrcmNvJo42lpVAqtjSlXY0,2412
+beamlit-0.0.34rc64.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.34rc64.dist-info/RECORD,,