beamlit 0.0.41rc85__tar.gz → 0.0.41rc86__tar.gz

{beamlit-0.0.41rc85 → beamlit-0.0.41rc86}/PKG-INFO

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

{beamlit-0.0.41rc85 → beamlit-0.0.41rc86}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beamlit"
-version = "0.0.41.pre-85"
+version = "0.0.41.pre-86"
 description = "Add your description here"
 readme = "README.md"
 authors = [

{beamlit-0.0.41rc85 → beamlit-0.0.41rc86}/src/beamlit/agents/chat.py

@@ -1,11 +1,12 @@
 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__)
 

{beamlit-0.0.41rc85 → beamlit-0.0.41rc86}/src/beamlit/common/settings.py

@@ -4,8 +4,12 @@ from typing import Tuple, Type, Union
 from langchain_core.language_models.chat_models import BaseChatModel
 from langgraph.graph.graph import CompiledGraph
 from pydantic import Field
-from pydantic_settings import (BaseSettings, PydanticBaseSettingsSource,
-                               SettingsConfigDict, YamlConfigSettingsSource)
+from pydantic_settings import (
+    BaseSettings,
+    PydanticBaseSettingsSource,
+    SettingsConfigDict,
+    YamlConfigSettingsSource,
+)
 
 from beamlit.common.logger import init as init_logger
 from beamlit.models import Agent, Function, Model

{beamlit-0.0.41rc85 → beamlit-0.0.41rc86}/src/beamlit/functions/__init__.py

@@ -1,5 +1,6 @@
 """Functions package providing function decorators and utilities."""
 
-from .decorator import function, get_functions, kit
+from .common import get_functions
+from .decorator import function, kit
 
 __all__ = ["function", "kit", "get_functions"]

beamlit-0.0.41rc85/src/beamlit/functions/decorator.py → beamlit-0.0.41rc86/src/beamlit/functions/common.py

@@ -1,14 +1,11 @@
 """Decorators for creating function tools with Beamlit and LangChain integration."""
 import ast
 import asyncio
-import functools
 import importlib.util
 import os
-from collections.abc import Callable
 from logging import getLogger
 from typing import Union
 
-from fastapi import Request
 from langchain_core.tools import StructuredTool
 from langchain_core.tools.base import create_schema_from_function
 
@@ -17,7 +14,7 @@ from beamlit.client import AuthenticatedClient
 from beamlit.common import slugify
 from beamlit.common.settings import get_settings
 from beamlit.functions.remote.remote import RemoteToolkit
-from beamlit.models import AgentChain, Function, FunctionKit
+from beamlit.models import AgentChain
 
 logger = getLogger(__name__)
 
@@ -144,38 +141,3 @@ def get_functions(
 
     return functions
 
-
-
-def kit(bl_kit: FunctionKit = None, **kwargs: dict) -> Callable:
-    """Create function tools with Beamlit and LangChain integration."""
-
-    def wrapper(func: Callable) -> Callable:
-        if bl_kit and not func.__doc__ and bl_kit.description:
-            func.__doc__ = bl_kit.description
-        return func
-
-    return wrapper
-
-
-def function(*args, function: Function | dict = None, kit=False, **kwargs: dict) -> Callable:
-    """Create function tools with Beamlit and LangChain integration."""
-    if function is not None and not isinstance(function, dict):
-        raise Exception(
-            'function must be a dictionary, example: @function(function={"metadata": {"name": "my_function"}})'
-        )
-    if isinstance(function, dict):
-        function = Function(**function)
-
-    def wrapper(func: Callable) -> Callable:
-        if function and not func.__doc__ and function.spec and function.spec.description:
-            func.__doc__ = function.spec.description
-
-        @functools.wraps(func)
-        async def wrapped(*args, **kwargs):
-            if len(args) > 0 and isinstance(args[0], Request):
-                body = await args[0].json()
-                args = [body.get(param) for param in func.__code__.co_varnames[:func.__code__.co_argcount]]
-            return await func(*args, **kwargs) if asyncio.iscoroutinefunction(func) else func(*args, **kwargs)
-        return wrapped
-
-    return wrapper

beamlit-0.0.41rc86/src/beamlit/functions/decorator.py

@@ -0,0 +1,45 @@
+"""Decorators for creating function tools with Beamlit and LangChain integration."""
+import asyncio
+import functools
+from collections.abc import Callable
+from logging import getLogger
+
+from fastapi import Request
+
+from beamlit.models import Function, FunctionKit
+
+logger = getLogger(__name__)
+
+def kit(bl_kit: FunctionKit = None, **kwargs: dict) -> Callable:
+    """Create function tools with Beamlit and LangChain integration."""
+
+    def wrapper(func: Callable) -> Callable:
+        if bl_kit and not func.__doc__ and bl_kit.description:
+            func.__doc__ = bl_kit.description
+        return func
+
+    return wrapper
+
+
+def function(*args, function: Function | dict = None, kit=False, **kwargs: dict) -> Callable:
+    """Create function tools with Beamlit and LangChain integration."""
+    if function is not None and not isinstance(function, dict):
+        raise Exception(
+            'function must be a dictionary, example: @function(function={"metadata": {"name": "my_function"}})'
+        )
+    if isinstance(function, dict):
+        function = Function(**function)
+
+    def wrapper(func: Callable) -> Callable:
+        if function and not func.__doc__ and function.spec and function.spec.description:
+            func.__doc__ = function.spec.description
+
+        @functools.wraps(func)
+        async def wrapped(*args, **kwargs):
+            if len(args) > 0 and isinstance(args[0], Request):
+                body = await args[0].json()
+                args = [body.get(param) for param in func.__code__.co_varnames[:func.__code__.co_argcount]]
+            return await func(*args, **kwargs) if asyncio.iscoroutinefunction(func) else func(*args, **kwargs)
+        return wrapped
+
+    return wrapper

{beamlit-0.0.41rc85 → beamlit-0.0.41rc86}/src/beamlit/functions/mcp/mcp.py

@@ -1,5 +1,4 @@
 import asyncio
-import urllib.parse
 import warnings
 from typing import Any, Callable
 

{beamlit-0.0.41rc85 → beamlit-0.0.41rc86}/src/beamlit/functions/remote/remote.py

@@ -5,6 +5,8 @@ from typing import Callable
 
 import pydantic
 import typing_extensions as t
+from langchain_core.tools.base import BaseTool, ToolException
+
 from beamlit.api.functions import get_function, list_functions
 from beamlit.authentication.authentication import AuthenticatedClient
 from beamlit.common.settings import get_settings
@@ -12,7 +14,6 @@ from beamlit.errors import UnexpectedStatus
 from beamlit.functions.mcp.mcp import MCPClient, MCPToolkit
 from beamlit.models import Function, StoreFunctionParameter
 from beamlit.run import RunClient
-from langchain_core.tools.base import BaseTool, ToolException
 
 
 def create_dynamic_schema(name: str, parameters: list[StoreFunctionParameter]) -> type[pydantic.BaseModel]:

{beamlit-0.0.41rc85 → beamlit-0.0.41rc86}/src/integrationtest/chat.py

@@ -1,9 +1,10 @@
 import asyncio
 
+from langgraph.graph.graph import CompiledGraph
+
 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