beamlit 0.0.20rc5__tar.gz → 0.0.20rc7__tar.gz

{beamlit-0.0.20rc5 → beamlit-0.0.20rc7}/PKG-INFO

@@ -1,11 +1,17 @@
 Metadata-Version: 2.3
 Name: beamlit
-Version: 0.0.20rc5
+Version: 0.0.20rc7
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12
+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-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: langgraph<0.3.0,>=0.2.40
 Requires-Dist: pydantic-settings<2.7.0,>=2.6.1
 Requires-Dist: pydantic<2.11.0,>=2.10.3
 Requires-Dist: python-dateutil>=2.8.0

beamlit-0.0.20rc7/beamlit.yaml

@@ -0,0 +1,5 @@
+agent:
+  functions_directory: src/integrationtest/functions
+server:
+  module: integrationtest.agent.main
+  port: 1338

beamlit-0.0.20rc7/pyproject.toml

@@ -0,0 +1,43 @@
+[project]
+name = "beamlit"
+version = "0.0.20.pre-7"
+description = "Add your description here"
+readme = "README.md"
+authors = [
+    { name = "cploujoux", email = "ch.ploujoux@gmail.com" }
+]
+packages = [
+    {include = "beamlit"},
+]
+include = ["CHANGELOG.md", "beamlit/py.typed"]
+requires-python = ">=3.12"
+dependencies = [
+    "httpx>=0.20.0,<0.28.0",
+    "attrs>=21.3.0",
+    "python-dateutil>=2.8.0",
+    "pyyaml>=6.0.2,<6.1.0",
+    "pydantic-settings>=2.6.1,<2.7.0",
+    "pydantic>=2.10.3,<2.11.0",
+    "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",
+    "langgraph>=0.2.40,<0.3.0",
+    "fastapi[standard]>=0.115.4,<0.116.0",
+    "asgi-correlation-id>=4.3.4,<5.0.0",
+]
+[dependency-groups]
+dev = [
+    "ruff>=0.8.2",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.ruff]
+line-length = 100
+
+[tool.ruff.lint]
+select = ["F", "I", "UP"]
+ignore = ["UP006", "UP007", "UP035"]

beamlit-0.0.20rc7/src/beamlit/agents/__init__.py

@@ -0,0 +1,4 @@
+from .chat import get_chat_model
+from .decorator import agent
+
+__all__ = ["agent", "get_chat_model"]

beamlit-0.0.20rc7/src/beamlit/agents/chat.py

@@ -0,0 +1,87 @@
+from logging import getLogger
+
+from beamlit.authentication import get_authentication_headers, new_client
+from beamlit.common.settings import get_settings
+from beamlit.models import AgentDeployment
+
+logger = getLogger(__name__)
+
+
+def get_base_url(agent_model: AgentDeployment):
+    settings = get_settings()
+    return f"{settings.run_url}/{settings.workspace}/models/{agent_model.model}/v1"
+
+
+def get_mistral_chat_model(**kwargs):
+    from langchain_mistralai.chat_models import ChatMistralAI  # type: ignore
+
+    return ChatMistralAI(**kwargs)
+
+
+def get_openai_chat_model(**kwargs):
+    from langchain_openai import ChatOpenAI  # type: ignore
+
+    return ChatOpenAI(**kwargs)
+
+
+def get_anthropic_chat_model(**kwargs):
+    from langchain_anthropic import ChatAnthropic  # type: ignore
+
+    return ChatAnthropic(**kwargs)
+
+
+def get_chat_model(agent_model: AgentDeployment):
+    settings = get_settings()
+    client = new_client()
+
+    headers = get_authentication_headers(settings)
+    headers["X-Beamlit-Environment"] = agent_model.environment
+
+    jwt = headers.pop("X-Beamlit-Authorization").replace("Bearer ", "")
+    params = {"environment": agent_model.environment}
+    chat_classes = {
+        "openai": {
+            "func": get_openai_chat_model,
+            "kwargs": {
+                "http_client": client.get_httpx_client(),
+            },
+        },
+        "anthropic": {
+            "func": get_anthropic_chat_model,
+            "kwargs": {},
+        },
+        "mistral": {
+            "func": get_mistral_chat_model,
+            "kwargs": {
+                "api_key": jwt,
+            },
+        },
+    }
+
+    if agent_model is None:
+        raise ValueError("agent_model not found in configuration")
+    if agent_model.runtime is None:
+        raise ValueError("runtime not found in agent model")
+    if agent_model.runtime.type_ is None:
+        raise ValueError("type not found in runtime")
+    if agent_model.runtime.model is None:
+        raise ValueError("model not found in runtime")
+
+    provider = agent_model.runtime.type_
+    model = agent_model.runtime.model
+
+    kwargs = {
+        "model": model,
+        "base_url": get_base_url(agent_model),
+        "default_query": params,
+        "default_headers": headers,
+        "api_key": "fake_api_key",
+        "temperature": 0,
+    }
+    chat_class = chat_classes.get(provider)
+    if not chat_class:
+        logger.warning(f"Provider {provider} not currently supported, defaulting to OpenAI")
+        chat_class = chat_classes["openai"]
+    if "kwargs" in chat_class:
+        kwargs.update(chat_class["kwargs"])
+    return chat_class["func"](**kwargs)

beamlit-0.0.20rc7/src/beamlit/agents/decorator.py

@@ -0,0 +1,147 @@
+# Import necessary modules
+import ast
+import importlib
+import os
+from logging import getLogger
+
+from langgraph.checkpoint.memory import MemorySaver
+from langgraph.prebuilt import create_react_agent
+
+from beamlit.api.models import get_model_deployment
+from beamlit.authentication import new_client
+from beamlit.common.settings import get_settings, init
+from beamlit.errors import UnexpectedStatus
+from beamlit.models import AgentDeployment
+
+from .chat import get_chat_model
+
+
+def get_functions(dir="src/functions", from_decorator="function"):
+    functions = []
+    logger = getLogger(__name__)
+
+    # Walk through all Python files in functions directory and subdirectories
+    for root, _, files in os.walk(dir):
+        for file in files:
+            if file.endswith(".py"):
+                file_path = os.path.join(root, file)
+                # Read and compile the file content
+                with open(file_path) as f:
+                    try:
+                        file_content = f.read()
+                        # Parse the file content to find decorated functions
+                        tree = ast.parse(file_content)
+
+                        # Look for function definitions with decorators
+                        for node in ast.walk(tree):
+                            if (
+                                not isinstance(node, ast.FunctionDef)
+                                or len(node.decorator_list) == 0
+                            ):
+                                continue
+                            decorator = node.decorator_list[0]
+
+                            decorator_name = ""
+                            if isinstance(decorator, ast.Call):
+                                decorator_name = decorator.func.id
+                            if isinstance(decorator, ast.Name):
+                                decorator_name = decorator.id
+
+                            if decorator_name == from_decorator:
+                                # Get the function name and decorator name
+                                func_name = node.name
+
+                                # Import the module to get the actual function
+                                spec = importlib.util.spec_from_file_location(func_name, file_path)
+                                module = importlib.util.module_from_spec(spec)
+                                spec.loader.exec_module(module)
+                                # Check if kit=True in the decorator arguments
+                                is_kit = False
+                                if isinstance(decorator, ast.Call):
+                                    for keyword in decorator.keywords:
+                                        if keyword.arg == "kit" and isinstance(
+                                            keyword.value, ast.Constant
+                                        ):
+                                            is_kit = keyword.value.value
+                                if is_kit:
+                                    kit_functions = get_functions(
+                                        dir=os.path.join(root),
+                                        from_decorator="kit",
+                                    )
+                                    functions.extend(kit_functions)
+
+                                # Get the decorated function
+                                if not is_kit and hasattr(module, func_name):
+                                    func = getattr(module, func_name)
+                                    functions.append(func)
+                    except Exception as e:
+                        logger.warning(f"Error processing {file_path}: {e!s}")
+    return functions
+
+
+def agent(
+    bl_agent: AgentDeployment = None,
+    chat_model=None,
+    agent=None,
+):
+    settings = init()
+    logger = getLogger(__name__)
+
+    def wrapper(func):
+        settings = get_settings()
+
+        def wrapped(*args, **kwargs):
+            return func(
+                settings.agent.agent,
+                settings.agent.chat_model,
+                settings.agent.functions,
+                *args,
+                **kwargs,
+            )
+
+        return wrapped
+
+    # Initialize functions array to store decorated functions
+    functions = get_functions(dir=settings.agent.functions_directory)
+    settings.agent.functions = functions
+
+    if bl_agent.model and chat_model is None:
+        client = new_client()
+        try:
+            response = get_model_deployment.sync_detailed(
+                bl_agent.model, settings.environment, client=client
+            )
+            settings.agent.model = response.parsed
+        except UnexpectedStatus as e:
+            if e.status_code == 404 and settings.environment != "production":
+                try:
+                    response = get_model_deployment.sync_detailed(
+                        bl_agent.model, "production", client=client
+                    )
+                    settings.agent.model = response.parsed
+                except UnexpectedStatus as e:
+                    if e.status_code == 404:
+                        raise ValueError(f"Model {bl_agent.model} not found")
+            else:
+                raise e
+        chat_model = get_chat_model(settings.agent.model)
+        settings.agent.chat_model = chat_model
+        runtime = settings.agent.model.runtime
+        logger.info(f"Chat model configured, using: {runtime.type_}:{runtime.model}")
+
+    if 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 agent is None and chat_model is not None:
+        memory = MemorySaver()
+        agent = create_react_agent(chat_model, functions, checkpointer=memory)
+        settings.agent.agent = agent
+
+    return wrapper

{beamlit-0.0.20rc5 → beamlit-0.0.20rc7}/src/beamlit/api/agents/delete_agent_history.py

@@ -21,7 +21,9 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[AgentHistory]:
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[AgentHistory]:
     if response.status_code == 200:
         response_200 = AgentHistory.from_dict(response.json())
 
@@ -32,7 +34,9 @@ def _parse_response(*, client: Union[AuthenticatedClient, Client], response: htt
         return None
 
 
-def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[AgentHistory]:
+def _build_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[AgentHistory]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

{beamlit-0.0.20rc5 → beamlit-0.0.20rc7}/src/beamlit/api/agents/get_agent_deployment_logs.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 import httpx
 
@@ -23,7 +23,7 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[List["ResourceDeploymentLog"]]:
+) -> Optional[list["ResourceDeploymentLog"]]:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
@@ -41,7 +41,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[List["ResourceDeploymentLog"]]:
+) -> Response[list["ResourceDeploymentLog"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -55,7 +55,7 @@ def sync_detailed(
     environment_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[List["ResourceDeploymentLog"]]:
+) -> Response[list["ResourceDeploymentLog"]]:
     """
     Args:
         agent_name (str):
@@ -66,7 +66,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['ResourceDeploymentLog']]
+        Response[list['ResourceDeploymentLog']]
     """
 
     kwargs = _get_kwargs(
@@ -86,7 +86,7 @@ def sync(
     environment_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[List["ResourceDeploymentLog"]]:
+) -> Optional[list["ResourceDeploymentLog"]]:
     """
     Args:
         agent_name (str):
@@ -97,7 +97,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['ResourceDeploymentLog']
+        list['ResourceDeploymentLog']
     """
 
     return sync_detailed(
@@ -112,7 +112,7 @@ async def asyncio_detailed(
     environment_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[List["ResourceDeploymentLog"]]:
+) -> Response[list["ResourceDeploymentLog"]]:
     """
     Args:
         agent_name (str):
@@ -123,7 +123,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['ResourceDeploymentLog']]
+        Response[list['ResourceDeploymentLog']]
     """
 
     kwargs = _get_kwargs(
@@ -141,7 +141,7 @@ async def asyncio(
     environment_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[List["ResourceDeploymentLog"]]:
+) -> Optional[list["ResourceDeploymentLog"]]:
     """
     Args:
         agent_name (str):
@@ -152,7 +152,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['ResourceDeploymentLog']
+        list['ResourceDeploymentLog']
     """
 
     return (

{beamlit-0.0.20rc5 → beamlit-0.0.20rc7}/src/beamlit/api/agents/get_agent_history.py

@@ -21,7 +21,9 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[AgentHistory]:
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[AgentHistory]:
     if response.status_code == 200:
         response_200 = AgentHistory.from_dict(response.json())
 
@@ -32,7 +34,9 @@ def _parse_response(*, client: Union[AuthenticatedClient, Client], response: htt
         return None
 
 
-def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[AgentHistory]:
+def _build_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[AgentHistory]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

{beamlit-0.0.20rc5 → beamlit-0.0.20rc7}/src/beamlit/api/agents/list_agent_deployment_history.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 import httpx
 
@@ -23,7 +23,7 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[List["AgentDeploymentHistory"]]:
+) -> Optional[list["AgentDeploymentHistory"]]:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
@@ -41,7 +41,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[List["AgentDeploymentHistory"]]:
+) -> Response[list["AgentDeploymentHistory"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -55,7 +55,7 @@ def sync_detailed(
     environment_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[List["AgentDeploymentHistory"]]:
+) -> Response[list["AgentDeploymentHistory"]]:
     """
     Args:
         agent_name (str):
@@ -66,7 +66,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['AgentDeploymentHistory']]
+        Response[list['AgentDeploymentHistory']]
     """
 
     kwargs = _get_kwargs(
@@ -86,7 +86,7 @@ def sync(
     environment_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[List["AgentDeploymentHistory"]]:
+) -> Optional[list["AgentDeploymentHistory"]]:
     """
     Args:
         agent_name (str):
@@ -97,7 +97,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['AgentDeploymentHistory']
+        list['AgentDeploymentHistory']
     """
 
     return sync_detailed(
@@ -112,7 +112,7 @@ async def asyncio_detailed(
     environment_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[List["AgentDeploymentHistory"]]:
+) -> Response[list["AgentDeploymentHistory"]]:
     """
     Args:
         agent_name (str):
@@ -123,7 +123,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['AgentDeploymentHistory']]
+        Response[list['AgentDeploymentHistory']]
     """
 
     kwargs = _get_kwargs(
@@ -141,7 +141,7 @@ async def asyncio(
     environment_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[List["AgentDeploymentHistory"]]:
+) -> Optional[list["AgentDeploymentHistory"]]:
     """
     Args:
         agent_name (str):
@@ -152,7 +152,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['AgentDeploymentHistory']
+        list['AgentDeploymentHistory']
     """
 
     return (

{beamlit-0.0.20rc5 → beamlit-0.0.20rc7}/src/beamlit/api/agents/list_agent_deployments.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 import httpx
 
@@ -22,7 +22,7 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[List["AgentDeployment"]]:
+) -> Optional[list["AgentDeployment"]]:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
@@ -40,7 +40,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[List["AgentDeployment"]]:
+) -> Response[list["AgentDeployment"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -53,7 +53,7 @@ def sync_detailed(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[List["AgentDeployment"]]:
+) -> Response[list["AgentDeployment"]]:
     """List all agent deployments
 
     Args:
@@ -64,7 +64,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['AgentDeployment']]
+        Response[list['AgentDeployment']]
     """
 
     kwargs = _get_kwargs(
@@ -82,7 +82,7 @@ def sync(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[List["AgentDeployment"]]:
+) -> Optional[list["AgentDeployment"]]:
     """List all agent deployments
 
     Args:
@@ -93,7 +93,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['AgentDeployment']
+        list['AgentDeployment']
     """
 
     return sync_detailed(
@@ -106,7 +106,7 @@ async def asyncio_detailed(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Response[List["AgentDeployment"]]:
+) -> Response[list["AgentDeployment"]]:
     """List all agent deployments
 
     Args:
@@ -117,7 +117,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['AgentDeployment']]
+        Response[list['AgentDeployment']]
     """
 
     kwargs = _get_kwargs(
@@ -133,7 +133,7 @@ async def asyncio(
     agent_name: str,
     *,
     client: AuthenticatedClient,
-) -> Optional[List["AgentDeployment"]]:
+) -> Optional[list["AgentDeployment"]]:
     """List all agent deployments
 
     Args:
@@ -144,7 +144,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        List['AgentDeployment']
+        list['AgentDeployment']
     """
 
     return (