beamlit 0.0.43__py3-none-any.whl → 0.0.44__py3-none-any.whl

beamlit/agents/chat.py

@@ -48,6 +48,20 @@ def get_deepseek_chat_model(**kwargs):
 
     return ChatDeepSeek(**kwargs)
 
+def get_azure_ai_inference_chat_model(**kwargs):
+    from langchain_openai import ChatOpenAI  # type: ignore
+
+    return ChatOpenAI(
+        **kwargs
+    )  # It uses a compatible endpoint, so we can use the ChatOpenAI interface
+
+def get_azure_marketplace_chat_model(**kwargs):
+    from langchain_openai import OpenAI  # type: ignore
+      
+    return OpenAI(
+        **kwargs
+    )  # It seems to use a compatible endpoint, so we can use the classic OpenAI interface
+
 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
@@ -110,6 +124,16 @@ def get_chat_model_full(name: str, agent_model: Union[Model, None] = None) -> Tu
                 "api_key": jwt,
             },
         },
+        "azure-ai-inference": {
+            "func": get_azure_ai_inference_chat_model,
+            "kwargs": {
+                "base_url": get_base_url(name).replace("/v1", ""),
+            },
+        },
+        "azure-marketplace": {
+            "func": get_azure_marketplace_chat_model,
+            "kwargs": {},
+        },
     }
 
     provider = (

beamlit/agents/decorator.py

@@ -136,7 +136,12 @@ def agent(
                     "@function()\n"
                     "def hello_world(query: str):\n"
                     "    return 'Hello, world!'\n")
-            _agent = create_react_agent(chat_model, functions, checkpointer=memory)
+            try:
+                _agent = create_react_agent(chat_model, functions, checkpointer=memory)
+            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)
+              
             settings.agent.agent = _agent
         else:
             settings.agent.agent = override_agent

{beamlit-0.0.43.dist-info → beamlit-0.0.44.dist-info}/METADATA

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

{beamlit-0.0.43.dist-info → beamlit-0.0.44.dist-info}/RECORD

@@ -6,8 +6,8 @@ beamlit/run.py,sha256=HtDYDjD7oVfQ8r3T5_t4qN5UDJOJfsQILi45Z21ArAg,1446
 beamlit/types.py,sha256=E1hhDh_zXfsSQ0NCt9-uw90_Mr5iIlsdfnfvxv5HarU,1005
 beamlit/agents/__init__.py,sha256=bWsFaXUbAps3IsL3Prti89m1s714vICXodbQi77h3vY,206
 beamlit/agents/chain.py,sha256=vfCjiFHuu02uTTGicxMlFzjyICQkIjpXrBGs-7uJEsg,2826
-beamlit/agents/chat.py,sha256=6iZuaP25hUvrKdth-6sbtqxkZ1wM0Fv13eRttxNZlIU,4715
-beamlit/agents/decorator.py,sha256=R5evLoproCgYjKXmH-Vq31r6G2lzZvUitXD1NoTflcA,6014
+beamlit/agents/chat.py,sha256=4R_mKeGvVXluQ5r-pnN3YaqhE3lFuetyos4zUBLnLnE,5510
+beamlit/agents/decorator.py,sha256=2FCOecAOVWBvvTMqkUmXxmtOwI2qTMVHUjVsqqCa8L0,6360
 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
@@ -278,7 +278,7 @@ 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.43.dist-info/METADATA,sha256=Ys4fQPfeQW9oMY_2LSpycE1PXbU4efFn2OEy_0QrZ4g,3733
-beamlit-0.0.43.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.43.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-beamlit-0.0.43.dist-info/RECORD,,
+beamlit-0.0.44.dist-info/METADATA,sha256=HQExKWELsmaHruDCZ8Zr-WthqrsFulBSVQhFbUeyXcQ,3733
+beamlit-0.0.44.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.44.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+beamlit-0.0.44.dist-info/RECORD,,