beamlit 0.0.37__py3-none-any.whl → 0.0.38__py3-none-any.whl

beamlit/agents/__init__.py

@@ -1,5 +1,5 @@
-from .chat import get_chat_model
+from .chat import get_chat_model, get_chat_model_full
 from .decorator import agent
 from .thread import get_default_thread
 
-__all__ = ["agent", "get_chat_model", "get_default_thread"]
+__all__ = ["agent", "get_chat_model", "get_chat_model_full", "get_default_thread"]

beamlit/agents/chat.py

@@ -1,12 +1,11 @@
 from logging import getLogger
 from typing import Tuple, Union
 
-from langchain_core.language_models import BaseChatModel
-
 from beamlit.api.models import get_model
 from beamlit.authentication import get_authentication_headers, new_client
 from beamlit.common.settings import get_settings
 from beamlit.models import Model
+from langchain_core.language_models import BaseChatModel
 
 logger = getLogger(__name__)
 
@@ -43,7 +42,16 @@ def get_cohere_chat_model(**kwargs):
 
     return ChatCohere(**kwargs)
 
-def get_chat_model(name: str, agent_model: Union[Model, None] = None) -> Tuple[BaseChatModel, str, str]:
+def get_deepseek_chat_model(**kwargs):
+    from langchain_deepseek import ChatDeepSeek  # type: ignore
+
+    return ChatDeepSeek(**kwargs)
+
+def get_chat_model(name: str, agent_model: Union[Model, None] = None) -> BaseChatModel:
+    [chat_model, _, __] = get_chat_model_full(name, agent_model)
+    return chat_model
+
+def get_chat_model_full(name: str, agent_model: Union[Model, None] = None) -> Tuple[BaseChatModel, str, str]:
     settings = get_settings()
     client = new_client()
 
@@ -69,7 +77,10 @@ def get_chat_model(name: str, agent_model: Union[Model, None] = None) -> Tuple[B
         },
         "anthropic": {
             "func": get_anthropic_chat_model,
-            "kwargs": {},
+            "kwargs": {
+                "base_url": get_base_url(name).replace("/v1", ""),
+            },
+            "remove_kwargs": ["default_query"]
         },
         "mistral": {
             "func": get_mistral_chat_model,
@@ -89,6 +100,13 @@ def get_chat_model(name: str, agent_model: Union[Model, None] = None) -> Tuple[B
             "func": get_cohere_chat_model,
             "kwargs": {
                 "cohere_api_key": jwt,
+                "base_url": get_base_url(name).replace("/v1", ""),
+            },
+        },
+        "deepseek": {
+            "func": get_deepseek_chat_model,
+            "kwargs": {
+                "api_key": jwt,
             },
         },
     }

beamlit/agents/decorator.py

@@ -3,23 +3,23 @@ import functools
 import inspect
 from logging import getLogger
 
-from langgraph.checkpoint.memory import MemorySaver
-from langgraph.prebuilt import create_react_agent
-
 from beamlit.api.models import get_model, list_models
 from beamlit.authentication import new_client
 from beamlit.common.settings import init
 from beamlit.errors import UnexpectedStatus
 from beamlit.functions import get_functions
 from beamlit.models import Agent, AgentMetadata, AgentSpec
+from langgraph.checkpoint.memory import MemorySaver
+from langgraph.prebuilt import create_react_agent
 
-from .chat import get_chat_model
+from .chat import get_chat_model_full
 
 
 def agent(
     agent: Agent | dict = None,
     override_model=None,
     override_agent=None,
+    override_functions=None,
     mcp_hub=None,
     remote_functions=None,
 ):
@@ -85,19 +85,23 @@ def agent(
                     raise e
 
                 if settings.agent.model:
-                    chat_model, provider, model = get_chat_model(agent.spec.model, settings.agent.model)
+                    chat_model, provider, model = get_chat_model_full(agent.spec.model, settings.agent.model)
                     settings.agent.chat_model = chat_model
                     logger.info(f"Chat model configured, using: {provider}:{model}")
 
-        functions = get_functions(
-            client=client,
-            dir=settings.agent.functions_directory,
-            mcp_hub=mcp_hub,
-            remote_functions=remote_functions,
-            chain=agent.spec.agent_chain,
+        if override_functions is not None:
+            functions = override_functions
+        else:
+            functions = get_functions(
+                client=client,
+                dir=settings.agent.functions_directory,
+                mcp_hub=mcp_hub,
+                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:
@@ -111,7 +115,7 @@ def agent(
                     models_select = f"You can select one from the your models: {models}"
                 except Exception:
                     pass
-                    
+
                 raise ValueError(f"You must provide a model.\n"
                     f"{models_select}\n"
                     f"You can create one at {settings.app_url}/{settings.workspace}/global-inference-network/models/create\n"

beamlit/common/instrumentation.py

@@ -269,11 +269,11 @@ def instrument_app(app: FastAPI):
             if instrumentor_class:
                 try:
                     instrumentor_class().instrument()
-                    log.info(f"Successfully instrumented {name}")
+                    log.debug(f"Successfully instrumented {name}")
                 except Exception as e:
-                    log.error(f"Failed to instrument {name}: {str(e)}")
+                    log.debug(f"Failed to instrument {name}: {str(e)}")
             else:
-                log.error(f"Could not load instrumentor for {name}")
+                log.debug(f"Could not load instrumentor for {name}")
         else:
             log.debug(
                 f"Skipping {name} instrumentation - required package '{required_package}' not installed"

{beamlit-0.0.37.dist-info → beamlit-0.0.38.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.37
+Version: 0.0.38
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12
@@ -8,9 +8,14 @@ Requires-Dist: asgi-correlation-id<5.0.0,>=4.3.4
 Requires-Dist: attrs>=21.3.0
 Requires-Dist: fastapi[standard]<0.116.0,>=0.115.4
 Requires-Dist: httpx<0.28.0,>=0.20.0
+Requires-Dist: langchain-anthropic>=0.3.4
+Requires-Dist: langchain-cohere>=0.4.2
 Requires-Dist: langchain-community<0.4.0,>=0.3.3
 Requires-Dist: langchain-core<0.4.0,>=0.3.13
-Requires-Dist: langchain-openai<0.3.0,>=0.2.4
+Requires-Dist: langchain-deepseek-official>=0.1.0
+Requires-Dist: langchain-mistralai>=0.2.5
+Requires-Dist: langchain-openai<0.4.0,>=0.3.0
+Requires-Dist: langchain-xai>=0.2.0
 Requires-Dist: langgraph<0.3.0,>=0.2.40
 Requires-Dist: mcp>=1.1.2
 Requires-Dist: opentelemetry-api>=1.28.2

{beamlit-0.0.37.dist-info → beamlit-0.0.38.dist-info}/RECORD

@@ -4,10 +4,10 @@ beamlit/errors.py,sha256=gO8GBmKqmSNgAg-E5oT-oOyxztvp7V_6XG7OUTT15q0,546
 beamlit/py.typed,sha256=8ZJUsxZiuOy1oJeVhsTWQhTG_6pTVHVXk5hJL79ebTk,25
 beamlit/run.py,sha256=HtDYDjD7oVfQ8r3T5_t4qN5UDJOJfsQILi45Z21ArAg,1446
 beamlit/types.py,sha256=E1hhDh_zXfsSQ0NCt9-uw90_Mr5iIlsdfnfvxv5HarU,1005
-beamlit/agents/__init__.py,sha256=4aXI1Sp8NuKJ8yVw9qUFRmFhSrWdT5b0bniqRKzZ42U,162
+beamlit/agents/__init__.py,sha256=bWsFaXUbAps3IsL3Prti89m1s714vICXodbQi77h3vY,206
 beamlit/agents/chain.py,sha256=vfCjiFHuu02uTTGicxMlFzjyICQkIjpXrBGs-7uJEsg,2826
-beamlit/agents/chat.py,sha256=6PE2f_tf6l4_-9mZ---ZHvxN52pmLkHkgh1ebPoG1BY,4052
-beamlit/agents/decorator.py,sha256=_N5uYircuULo2EZ1B0vaD1Rsc64534DtojmUVAJGUAw,5921
+beamlit/agents/chat.py,sha256=5iv6dM7Xlg2oP2cGgAlpFreJl4P8GZAiN8MM3v0YVYs,4714
+beamlit/agents/decorator.py,sha256=e9ajjak6fYrDBkQvfQhzHncYuxuDuEmpWdBbloG5ueE,6064
 beamlit/agents/thread.py,sha256=LN5Ss-uOf5_hdB0WV1dqpn-N-pDJB3C2hUvlCzdqtdk,519
 beamlit/api/__init__.py,sha256=zTSiG_ujSjAqWPyc435YXaX9XTlpMjiJWBbV-f-YtdA,45
 beamlit/api/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -132,7 +132,7 @@ beamlit/authentication/credentials.py,sha256=p_1xenabCbQuRz7BiFk7oTK4uCxAt_zoyku
 beamlit/authentication/device_mode.py,sha256=LNHjoKe3u4TWApLKO34cJjN92UsVkHSYSbXr6eKxo64,3721
 beamlit/common/__init__.py,sha256=saX5X3hRCJ9erSlXuSkZ2VGgquvpgdcofAU_9sM4bCE,354
 beamlit/common/error.py,sha256=f9oJDFxhoHK-vpjxBgEp0NwWIk0N_THPemUI7uQxVzU,270
-beamlit/common/instrumentation.py,sha256=TKBbLNSPz0TKGufxSbVuLdJ9Tunh7-Y967SxwIX4j-o,9287
+beamlit/common/instrumentation.py,sha256=L4Y6Xu7rQwIHngfp-d0LtWxReaPte6rIzykvkDaIyzE,9288
 beamlit/common/logger.py,sha256=nN_dSOl4bs13QU3Rod-w3e3jYOnlSrHx3_bs-ACY6Aw,1115
 beamlit/common/secrets.py,sha256=sid81bOe3LflkMKDHwBsBs9nIju8bp5-v9qU9gkyNMc,212
 beamlit/common/settings.py,sha256=OL1pZLB3BgN7QPCz9VR7gjYb-LoYpelilS7rU7FubCE,3962
@@ -285,6 +285,6 @@ beamlit/serve/app.py,sha256=ROS_tb9cO4GvOQKCwloyAzpYraTdIb3oG6sChXikeNw,3285
 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.37.dist-info/METADATA,sha256=qJFiMr8EeCEeqoM8COtZf9cPo_IPrLZDbAKfUK8mRlI,3502
-beamlit-0.0.37.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.37.dist-info/RECORD,,
+beamlit-0.0.38.dist-info/METADATA,sha256=of74Mv_sPBES0fa5d1vARpRk_BhCe-N7WnkzHvG7O8Y,3711
+beamlit-0.0.38.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.38.dist-info/RECORD,,