beamlit 0.0.57rc115__py3-none-any.whl → 0.0.57rc117__py3-none-any.whl

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
 
 from .voice.openai import OpenAIVoiceReactAgent
 
@@ -120,6 +119,14 @@ def get_azure_marketplace_chat_model(**kwargs):
         **kwargs
     )  # It seems to use a compatible endpoint, so we can use the classic OpenAI interface
 
+def get_gemini_chat_model(**kwargs):
+    from langchain_google_genai import ChatGoogleGenerativeAI  # type: ignore
+
+    print(kwargs)
+    return ChatGoogleGenerativeAI(
+        **kwargs,
+    )
+
 def get_chat_model(name: str, agent_model: Union[Model, None] = None) -> BaseChatModel:
     """
     Gets a chat model instance for the specified model name.
@@ -215,6 +222,19 @@ def get_chat_model_full(name: str, agent_model: Union[Model, None] = None) -> Tu
             "func": get_azure_marketplace_chat_model,
             "kwargs": {},
         },
+        "gemini": {
+            "func": get_gemini_chat_model,
+            "kwargs": {
+                "api_key": "fake_api_key",
+                "client_options": {
+                    # "api_endpoint": get_base_url(name).replace("/v1", "")
+                    "api_endpoint": f"http://localhost:8787/{settings.workspace}/models/{name}",
+                },
+                "transport": "rest",
+                "additional_headers": {"X-Beamlit-Authorization": f"Bearer {jwt}"},
+            },
+            "remove_kwargs": ["api_key", "default_headers"]
+        },
     }
 
     provider = (

beamlit/agents/decorator.py

@@ -10,15 +10,14 @@ import inspect
 from logging import getLogger
 from typing import Callable
 
-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 Settings, init
 from beamlit.errors import UnexpectedStatus
 from beamlit.functions import get_functions
 from beamlit.models import Agent, AgentSpec, Metadata
+from langgraph.checkpoint.memory import MemorySaver
+from langgraph.prebuilt import create_react_agent
 
 from .chat import get_chat_model_full
 from .voice.openai import OpenAIVoiceReactAgent
@@ -116,7 +115,7 @@ async def initialize_agent(
                 _agent = create_react_agent(chat_model, functions, checkpointer=memory, state_modifier=agent.spec.prompt or "")
             except AttributeError: # special case for azure-marketplace where it uses the old OpenAI interface (no tools)
                 logger.warning("Using the old OpenAI interface for Azure Marketplace, no tools available")
-                _agent = create_react_agent(chat_model, [], checkpointer=memory, state_modifier=agent.spec.prompt or "")
+                _agent = create_react_agent(chat_model, [], checkpointer=memory, state_modifier=(agent and agent.spec and agent.spec.prompt) or "")
 
         settings.agent.agent = _agent
     else:

beamlit/agents/thread.py

@@ -20,14 +20,6 @@ def get_default_thread(request: Request) -> str:
     """
     if request.headers.get("X-Beamlit-Thread-Id"):
         return request.headers.get("X-Beamlit-Thread-Id")
-    if request.headers.get("X-Beamlit-Sub"):
-        return request.headers.get("X-Beamlit-Sub")
-    authorization = request.headers.get("Authorization", request.headers.get("X-Beamlit-Authorization"))
-    if authorization and len(authorization.split("Bearer ")) > 1:
-        try:
-            token = authorization.split(" ")[1]
-            decoded = jwt.decode(token, options={"verify_signature": False})
-            return decoded["sub"]
-        except Exception:
-            return ""
-    return ""
+    if request.headers.get("Thread-Id"):
+        return request.headers.get("Thread-Id")
+    return ""

{beamlit-0.0.57rc115.dist-info → beamlit-0.0.57rc117.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.57rc115
+Version: 0.0.57rc117
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 License-File: LICENSE

{beamlit-0.0.57rc115.dist-info → beamlit-0.0.57rc117.dist-info}/RECORD

@@ -6,9 +6,9 @@ beamlit/run.py,sha256=YMI8iTPB1M9gd_1FG958tw-Prf9EwIcca4jK4igi7MM,4448
 beamlit/types.py,sha256=E1hhDh_zXfsSQ0NCt9-uw90_Mr5iIlsdfnfvxv5HarU,1005
 beamlit/agents/__init__.py,sha256=bWsFaXUbAps3IsL3Prti89m1s714vICXodbQi77h3vY,206
 beamlit/agents/chain.py,sha256=JsinjAYBr3oaM4heouZaiaV2jMmi779LHAMtD_4P59s,4867
-beamlit/agents/chat.py,sha256=ufuydptucLNe_Jyr7lQO1WfQ5pe0I5YKh-y0smwWABM,8301
-beamlit/agents/decorator.py,sha256=yRxzrLOMhXW0S4ThRD7cm6UqUarw6gdB_ZP2jnZanvM,8311
-beamlit/agents/thread.py,sha256=fbQUsnEOPqr-E-Gx-YhmP-c26XCpjqDeVdbNbqdGpDE,1196
+beamlit/agents/chat.py,sha256=3yd5ptQ-TlHGkK1GQ4Gw69YmhtuQCxtlpuBsCMiJ2Kw,9059
+beamlit/agents/decorator.py,sha256=wgdbgJ65LfmhldGWxG8PBYU_7datnVpqKRzom6Ok3GU,8337
+beamlit/agents/thread.py,sha256=QecahZCH8_U91cvCw2X6cYn9N1liVXVl3u3lFxVDdNg,798
 beamlit/agents/voice/openai.py,sha256=-RDBwl16i4TbUhFo5-77Ci3zmI3Y8U2yf2MmvXR2haQ,9479
 beamlit/agents/voice/utils.py,sha256=tQidyM40Ewuy12wKqpvJLvfJgneQ0sZf50dqnerPGHg,836
 beamlit/api/__init__.py,sha256=zTSiG_ujSjAqWPyc435YXaX9XTlpMjiJWBbV-f-YtdA,45
@@ -254,8 +254,8 @@ beamlit/serve/app.py,sha256=5XZci-R95Zjl97wMtQd1BRtonnkJJ2AeoTVFPKGAOfA,4283
 beamlit/serve/middlewares/__init__.py,sha256=O7fyfE1DIYmajFY9WWdzxCgeAQWZzJfeUjzHGbpWaAk,309
 beamlit/serve/middlewares/accesslog.py,sha256=lcu33j4epFSHRBaeTpyt8deNb3kaM3K91-andw4fp80,1112
 beamlit/serve/middlewares/processtime.py,sha256=3x5w1yQexB0xFNKK6fgLbINxT-eLLunfZ6UDV0bIIF4,944
-beamlit-0.0.57rc115.dist-info/METADATA,sha256=-4QXAnnZZ9N1D6GkBoJnpIbzthjSFdu9TLQY_NpJraI,3547
-beamlit-0.0.57rc115.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.57rc115.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
-beamlit-0.0.57rc115.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-beamlit-0.0.57rc115.dist-info/RECORD,,
+beamlit-0.0.57rc117.dist-info/METADATA,sha256=bFmOkATbRNyO8DL3GQ9k61uWV9wnm8vOBEtMV8ndpWE,3547
+beamlit-0.0.57rc117.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.57rc117.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
+beamlit-0.0.57rc117.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+beamlit-0.0.57rc117.dist-info/RECORD,,