mirascope 1.21.2__py3-none-any.whl → 1.21.3__py3-none-any.whl

mirascope/core/bedrock/_utils/_setup_call.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import os
 from collections.abc import AsyncGenerator, Awaitable, Callable, Coroutine, Generator
 from functools import wraps
 from typing import Any, ParamSpec, cast, overload
@@ -223,12 +224,23 @@ def setup_call(
 
     call_kwargs |= cast(BedrockCallKwargs, {"modelId": model, "messages": messages})
 
+    env_vars = {}
+    if access_key_id := os.getenv("AWS_ACCESS_KEY_ID"):
+        env_vars["aws_access_key_id"] = access_key_id
+    if secret_access_key := os.getenv("AWS_SECRET_ACCESS_KEY"):
+        env_vars["aws_secret_access_key"] = secret_access_key
+    if session_token := os.getenv("AWS_SESSION_TOKEN"):
+        env_vars["aws_session_token"] = session_token
+    if region_name := os.getenv("AWS_REGION_NAME"):
+        env_vars["region_name"] = region_name
+    if profile_name := os.getenv("AWS_PROFILE"):
+        env_vars["profile_name"] = profile_name
     if client is None:
         if fn_is_async(fn):
-            session = get_session()
+            session = get_session(env_vars=env_vars)
             _client = _AsyncBedrockRuntimeWrappedClient(session, model)
         else:
-            session = Session()
+            session = Session(**env_vars)
             _client = session.client("bedrock-runtime")
     else:
         _client = client

{mirascope-1.21.2.dist-info → mirascope-1.21.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mirascope
-Version: 1.21.2
+Version: 1.21.3
 Summary: LLM abstractions that aren't obstructions
 Project-URL: Homepage, https://mirascope.com
 Project-URL: Documentation, https://mirascope.com/WELCOME

{mirascope-1.21.2.dist-info → mirascope-1.21.3.dist-info}/RECORD

@@ -158,7 +158,7 @@ mirascope/core/bedrock/_utils/_convert_message_params.py,sha256=ZPFj34ed0-4bmMld
 mirascope/core/bedrock/_utils/_get_json_output.py,sha256=hW-IBBQ5YW85VljjFJHDDtu66zsaF2ydTbFxgCX_j6A,1159
 mirascope/core/bedrock/_utils/_handle_stream.py,sha256=s8KNMNDKzvSIkFROtaZgbEJry78X_qCzTvGmHcL7UW0,3776
 mirascope/core/bedrock/_utils/_message_param_converter.py,sha256=T45kksn78idbqD9NZ3Omx1nS_IoYmTfA5y-bAHlX2fM,6846
-mirascope/core/bedrock/_utils/_setup_call.py,sha256=md-1FilnhgPpNqJpkOZKlrVpt6kRQneEeCU8GvK-p8o,8572
+mirascope/core/bedrock/_utils/_setup_call.py,sha256=XQs-JlviE0uhbBxEpjXP8812NbiObLYx5VkAwJJAF84,9168
 mirascope/core/cohere/__init__.py,sha256=vk73WFGBOEmMFEiqWMRnPfxsCBDlDcq8SaLB2A6RKeo,830
 mirascope/core/cohere/_call.py,sha256=y0nB_7h7FWCNxHRPywtAVCYXyeYX3uzTyYBPWnuLwUE,2261
 mirascope/core/cohere/_call_kwargs.py,sha256=YmHwiofs0QADGp0wXUtOr_Z5Pt849zaCtIZmVyjw2OM,292
@@ -368,7 +368,7 @@ mirascope/v0/base/ops_utils.py,sha256=1Qq-VIwgHBaYutiZsS2MUQ4OgPC3APyywI5bTiTAmA
 mirascope/v0/base/prompts.py,sha256=FM2Yz98cSnDceYogiwPrp4BALf3_F3d4fIOCGAkd-SE,1298
 mirascope/v0/base/types.py,sha256=ZfatJoX0Yl0e3jhv0D_MhiSVHLYUeJsdN3um3iE10zY,352
 mirascope/v0/base/utils.py,sha256=XREPENRQTu8gpMhHU8RC8qH_am3FfGUvY-dJ6x8i-mw,681
-mirascope-1.21.2.dist-info/METADATA,sha256=1dvAilky26Nim8aNFd5hQEypMwSS9MfUs0grmn__v9o,8730
-mirascope-1.21.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mirascope-1.21.2.dist-info/licenses/LICENSE,sha256=LAs5Q8mdawTsVdONpDGukwsoc4KEUBmmonDEL39b23Y,1072
-mirascope-1.21.2.dist-info/RECORD,,
+mirascope-1.21.3.dist-info/METADATA,sha256=C4EyG1JBa6VMuS-WqgsCGClL3eLVtBN0dPzL1nJ6Rsg,8730
+mirascope-1.21.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mirascope-1.21.3.dist-info/licenses/LICENSE,sha256=LAs5Q8mdawTsVdONpDGukwsoc4KEUBmmonDEL39b23Y,1072
+mirascope-1.21.3.dist-info/RECORD,,