beamlit 0.0.37__tar.gz → 0.0.38__tar.gz

{beamlit-0.0.37 → beamlit-0.0.38}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [project]
 name = "beamlit"
-version = "0.0.37"
+version = "0.0.38"
 description = "Add your description here"
 readme = "README.md"
 authors = [
@@ -21,7 +21,7 @@ dependencies = [
     "requests>=2.32.3,<2.33.0",
     "langchain-core>=0.3.13,<0.4.0",
     "langchain-community>=0.3.3,<0.4.0",
-    "langchain-openai>=0.2.4,<0.3.0",
+    "langchain-openai>=0.3.0,<0.4.0",
     "langgraph>=0.2.40,<0.3.0",
     "fastapi[standard]>=0.115.4,<0.116.0",
     "asgi-correlation-id>=4.3.4,<5.0.0",
@@ -51,6 +51,11 @@ dependencies = [
     "opentelemetry-instrumentation-together>=0.35.0",
     "opentelemetry-instrumentation-watsonx>=0.35.0",
     "opentelemetry-instrumentation-weaviate>=0.35.0",
+    "langchain-cohere>=0.4.2",
+    "langchain-xai>=0.2.0",
+    "langchain-anthropic>=0.3.4",
+    "langchain-mistralai>=0.2.5",
+    "langchain-deepseek-official>=0.1.0",
 ]
 [dependency-groups]
 dev = [

beamlit-0.0.38/src/beamlit/agents/__init__.py

@@ -0,0 +1,5 @@
+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_chat_model_full", "get_default_thread"]

{beamlit-0.0.37 → beamlit-0.0.38}/src/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-0.0.37 → beamlit-0.0.38}/src/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-0.0.37 → beamlit-0.0.38}/src/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.38/src/integrationtest/chat.py

@@ -0,0 +1,36 @@
+import asyncio
+
+from beamlit.agents import agent, get_chat_model
+from beamlit.common import init
+from beamlit.functions import function
+from langgraph.graph.graph import CompiledGraph
+
+settings = init()
+chat = get_chat_model("gpt-4o-mini") # all good
+# chat = get_chat_model("ministral-3b-2410") # all good
+# chat = get_chat_model("cohere-command-r-plus") # all good, astream does not work
+# chat = get_chat_model("claude-3-5-sonnet") # all good
+# chat = get_chat_model("deepseek-chat")
+# chat = get_chat_model("xai-grok-beta") # all good
+
+print(chat)
+@function()
+async def get_weather(location: str):
+    """Get the weather for a given location"""
+    print("Using getWeather");
+    return "The weather in " + location + " is sunny"
+
+@agent(
+    override_model=chat,
+    override_functions=[get_weather],
+)
+async def main(request, agent: CompiledGraph):
+    config = {"configurable": {"thread_id": "thread_id"}}
+    agent_body = {"messages": [("user", "What's the weather in san francisco ?")]}
+    response = None
+    for chunk in agent.stream(agent_body, config=config):
+        response = chunk
+    print(response)
+
+
+asyncio.run(main({}))

{beamlit-0.0.37 → beamlit-0.0.38}/src/integrationtest/pyproject.toml

@@ -5,5 +5,8 @@ description = ""
 
 requires-python = ">=3.12"
 dependencies = [
-    "beamlit>=0.0.29rc21",
+    "beamlit",
 ]
+
+[tool.uv.sources]
+beamlit = { path = "../../" }