chatlas 0.13.1__py3-none-any.whl → 0.13.2__py3-none-any.whl

chatlas/_content.py

@@ -1,7 +1,9 @@
 from __future__ import annotations
 
+import inspect
+import warnings
 from pprint import pformat
-from typing import TYPE_CHECKING, Any, Literal, Optional, Union
+from typing import TYPE_CHECKING, Any, Literal, Optional, Union, cast
 
 import orjson
 from pydantic import BaseModel, ConfigDict
@@ -465,8 +467,36 @@ class ContentToolResult(Content):
 
     @staticmethod
     def _to_json(value: Any) -> object:
+        if hasattr(value, "to_pandas") and callable(value.to_pandas):
+            # Many (most?) df libs (polars, pyarrow, ...) have a .to_pandas()
+            # method, and pandas has a .to_json() method
+            value = value.to_pandas()
+
         if hasattr(value, "to_json") and callable(value.to_json):
-            return value.to_json()
+            # pandas defaults to "columns", which is not ideal for LLMs
+            # https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.to_json.html
+            sig = inspect.signature(value.to_json)
+            if "orient" in list(sig.parameters.keys()):
+                return value.to_json(orient="records")
+            else:
+                return value.to_json()
+
+        # Support for df libs (beyond those with a .to_pandas() method)
+        if hasattr(value, "__narwhals_dataframe__"):
+            try:
+                import narwhals
+
+                val = cast(narwhals.DataFrame, narwhals.from_native(value))
+                return val.to_pandas().to_json(orient="records")
+            except ImportError:
+                warnings.warn(
+                    f"Tool result object of type {type(value)} appears to be a "
+                    "narwhals-compatible DataFrame. If you run into issues with "
+                    "the LLM not understanding this value, try installing narwhals: "
+                    "`pip install narwhals`.",
+                    ImportWarning,
+                    stacklevel=2,
+                )
 
         if hasattr(value, "to_dict") and callable(value.to_dict):
             value = value.to_dict()

chatlas/_provider_databricks.py

@@ -106,7 +106,13 @@ class DatabricksProvider(OpenAIProvider):
         import httpx
         from openai import AsyncOpenAI
 
-        super().__init__(name=name, model=model)
+        super().__init__(
+            name=name,
+            model=model,
+            # The OpenAI() constructor will fail if no API key is present.
+            # However, a dummy value is fine -- WorkspaceClient() handles the auth.
+            api_key="not-used",
+        )
 
         self._seed = None
 

chatlas/_provider_openai.py

@@ -884,7 +884,13 @@ class OpenAIAzureProvider(OpenAIProvider):
         model: Optional[str] = "UnusedValue",
         kwargs: Optional["ChatAzureClientArgs"] = None,
     ):
-        super().__init__(name=name, model=deployment_id)
+        super().__init__(
+            name=name,
+            model=deployment_id,
+            # The OpenAI() constructor will fail if no API key is present.
+            # However, a dummy value is fine -- AzureOpenAI() handles the auth.
+            api_key=api_key or "not-used",
+        )
 
         self._seed = seed
 

chatlas/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.13.1'
-__version_tuple__ = version_tuple = (0, 13, 1)
+__version__ = version = '0.13.2'
+__version_tuple__ = version_tuple = (0, 13, 2)
 
 __commit_id__ = commit_id = None

chatlas/types/anthropic/_client.py

@@ -3,7 +3,7 @@
 # ---------------------------------------------------------
 
 
-from typing import Mapping, Optional, TypedDict, Union
+from typing import Mapping, Optional, TypedDict
 
 import anthropic
 import httpx
@@ -13,7 +13,7 @@ class ChatClientArgs(TypedDict, total=False):
     api_key: str | None
     auth_token: str | None
     base_url: str | httpx.URL | None
-    timeout: Union[float, anthropic.Timeout, None, anthropic.NotGiven]
+    timeout: float | anthropic.Timeout | None | anthropic.NotGiven
     max_retries: int
     default_headers: Optional[Mapping[str, str]]
     default_query: Optional[Mapping[str, object]]

chatlas/types/anthropic/_submit.py

@@ -47,26 +47,24 @@ class SubmitInputArgs(TypedDict, total=False):
         ],
         str,
     ]
-    service_tier: Union[Literal["auto", "standard_only"], anthropic.NotGiven]
-    stop_sequences: Union[Sequence[str], anthropic.NotGiven]
-    stream: Union[Literal[False], Literal[True], anthropic.NotGiven]
+    service_tier: Union[Literal["auto", "standard_only"], anthropic.Omit]
+    stop_sequences: Union[Sequence[str], anthropic.Omit]
+    stream: Union[Literal[False], Literal[True], anthropic.Omit]
     system: Union[
-        str,
-        Iterable[anthropic.types.text_block_param.TextBlockParam],
-        anthropic.NotGiven,
+        str, Iterable[anthropic.types.text_block_param.TextBlockParam], anthropic.Omit
     ]
-    temperature: float | anthropic.NotGiven
+    temperature: float | anthropic.Omit
     thinking: Union[
         anthropic.types.thinking_config_enabled_param.ThinkingConfigEnabledParam,
         anthropic.types.thinking_config_disabled_param.ThinkingConfigDisabledParam,
-        anthropic.NotGiven,
+        anthropic.Omit,
     ]
     tool_choice: Union[
         anthropic.types.tool_choice_auto_param.ToolChoiceAutoParam,
         anthropic.types.tool_choice_any_param.ToolChoiceAnyParam,
         anthropic.types.tool_choice_tool_param.ToolChoiceToolParam,
         anthropic.types.tool_choice_none_param.ToolChoiceNoneParam,
-        anthropic.NotGiven,
+        anthropic.Omit,
     ]
     tools: Union[
         Iterable[
@@ -79,10 +77,10 @@ class SubmitInputArgs(TypedDict, total=False):
                 anthropic.types.web_search_tool_20250305_param.WebSearchTool20250305Param,
             ]
         ],
-        anthropic.NotGiven,
+        anthropic.Omit,
     ]
-    top_k: int | anthropic.NotGiven
-    top_p: float | anthropic.NotGiven
+    top_k: int | anthropic.Omit
+    top_p: float | anthropic.Omit
     extra_headers: Optional[Mapping[str, Union[str, anthropic.Omit]]]
     extra_query: Optional[Mapping[str, object]]
     extra_body: object | None

chatlas/types/openai/_client.py

@@ -16,7 +16,7 @@ class ChatClientArgs(TypedDict, total=False):
     webhook_secret: str | None
     base_url: str | httpx.URL | None
     websocket_base_url: str | httpx.URL | None
-    timeout: Union[float, openai.Timeout, None, openai.NotGiven]
+    timeout: float | openai.Timeout | None | openai.NotGiven
     max_retries: int
     default_headers: Optional[Mapping[str, str]]
     default_query: Optional[Mapping[str, object]]

chatlas/types/openai/_submit.py

@@ -108,61 +108,61 @@ class SubmitInputArgs(TypedDict, total=False):
     audio: Union[
         openai.types.chat.chat_completion_audio_param.ChatCompletionAudioParam,
         None,
-        openai.NotGiven,
+        openai.Omit,
     ]
-    frequency_penalty: Union[float, None, openai.NotGiven]
+    frequency_penalty: Union[float, None, openai.Omit]
     function_call: Union[
         Literal["none", "auto"],
         openai.types.chat.chat_completion_function_call_option_param.ChatCompletionFunctionCallOptionParam,
-        openai.NotGiven,
+        openai.Omit,
     ]
     functions: Union[
-        Iterable[openai.types.chat.completion_create_params.Function], openai.NotGiven
+        Iterable[openai.types.chat.completion_create_params.Function], openai.Omit
     ]
-    logit_bias: Union[dict[str, int], None, openai.NotGiven]
-    logprobs: Union[bool, None, openai.NotGiven]
-    max_completion_tokens: Union[int, None, openai.NotGiven]
-    max_tokens: Union[int, None, openai.NotGiven]
-    metadata: Union[dict[str, str], None, openai.NotGiven]
-    modalities: Union[list[Literal["text", "audio"]], None, openai.NotGiven]
-    n: Union[int, None, openai.NotGiven]
-    parallel_tool_calls: bool | openai.NotGiven
+    logit_bias: Union[dict[str, int], None, openai.Omit]
+    logprobs: Union[bool, None, openai.Omit]
+    max_completion_tokens: Union[int, None, openai.Omit]
+    max_tokens: Union[int, None, openai.Omit]
+    metadata: Union[dict[str, str], None, openai.Omit]
+    modalities: Union[list[Literal["text", "audio"]], None, openai.Omit]
+    n: Union[int, None, openai.Omit]
+    parallel_tool_calls: bool | openai.Omit
     prediction: Union[
         openai.types.chat.chat_completion_prediction_content_param.ChatCompletionPredictionContentParam,
         None,
-        openai.NotGiven,
+        openai.Omit,
     ]
-    presence_penalty: Union[float, None, openai.NotGiven]
-    prompt_cache_key: str | openai.NotGiven
+    presence_penalty: Union[float, None, openai.Omit]
+    prompt_cache_key: str | openai.Omit
     reasoning_effort: Union[
-        Literal["minimal", "low", "medium", "high"], None, openai.NotGiven
+        Literal["minimal", "low", "medium", "high"], None, openai.Omit
     ]
     response_format: Union[
         openai.types.shared_params.response_format_text.ResponseFormatText,
         openai.types.shared_params.response_format_json_schema.ResponseFormatJSONSchema,
         openai.types.shared_params.response_format_json_object.ResponseFormatJSONObject,
-        openai.NotGiven,
+        openai.Omit,
     ]
-    safety_identifier: str | openai.NotGiven
-    seed: Union[int, None, openai.NotGiven]
+    safety_identifier: str | openai.Omit
+    seed: Union[int, None, openai.Omit]
     service_tier: Union[
-        Literal["auto", "default", "flex", "scale", "priority"], None, openai.NotGiven
+        Literal["auto", "default", "flex", "scale", "priority"], None, openai.Omit
     ]
-    stop: Union[str, None, Sequence[str], openai.NotGiven]
-    store: Union[bool, None, openai.NotGiven]
-    stream: Union[Literal[False], None, Literal[True], openai.NotGiven]
+    stop: Union[str, None, Sequence[str], openai.Omit]
+    store: Union[bool, None, openai.Omit]
+    stream: Union[Literal[False], None, Literal[True], openai.Omit]
     stream_options: Union[
         openai.types.chat.chat_completion_stream_options_param.ChatCompletionStreamOptionsParam,
         None,
-        openai.NotGiven,
+        openai.Omit,
     ]
-    temperature: Union[float, None, openai.NotGiven]
+    temperature: Union[float, None, openai.Omit]
     tool_choice: Union[
         Literal["none", "auto", "required"],
         openai.types.chat.chat_completion_allowed_tool_choice_param.ChatCompletionAllowedToolChoiceParam,
         openai.types.chat.chat_completion_named_tool_choice_param.ChatCompletionNamedToolChoiceParam,
         openai.types.chat.chat_completion_named_tool_choice_custom_param.ChatCompletionNamedToolChoiceCustomParam,
-        openai.NotGiven,
+        openai.Omit,
     ]
     tools: Union[
         Iterable[
@@ -171,14 +171,14 @@ class SubmitInputArgs(TypedDict, total=False):
                 openai.types.chat.chat_completion_custom_tool_param.ChatCompletionCustomToolParam,
             ]
         ],
-        openai.NotGiven,
+        openai.Omit,
     ]
-    top_logprobs: Union[int, None, openai.NotGiven]
-    top_p: Union[float, None, openai.NotGiven]
-    user: str | openai.NotGiven
-    verbosity: Union[Literal["low", "medium", "high"], None, openai.NotGiven]
+    top_logprobs: Union[int, None, openai.Omit]
+    top_p: Union[float, None, openai.Omit]
+    user: str | openai.Omit
+    verbosity: Union[Literal["low", "medium", "high"], None, openai.Omit]
     web_search_options: (
-        openai.types.chat.completion_create_params.WebSearchOptions | openai.NotGiven
+        openai.types.chat.completion_create_params.WebSearchOptions | openai.Omit
     )
     extra_headers: Optional[Mapping[str, Union[str, openai.Omit]]]
     extra_query: Optional[Mapping[str, object]]

{chatlas-0.13.1.dist-info → chatlas-0.13.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chatlas
-Version: 0.13.1
+Version: 0.13.2
 Summary: A simple and consistent interface for chatting with LLMs
 Project-URL: Homepage, https://posit-dev.github.io/chatlas
 Project-URL: Documentation, https://posit-dev.github.io/chatlas
@@ -38,9 +38,12 @@ Requires-Dist: databricks-sdk; extra == 'dev'
 Requires-Dist: google-genai>=1.14.0; extra == 'dev'
 Requires-Dist: htmltools; extra == 'dev'
 Requires-Dist: matplotlib; extra == 'dev'
+Requires-Dist: narwhals; extra == 'dev'
 Requires-Dist: numpy>1.24.4; extra == 'dev'
 Requires-Dist: openai; extra == 'dev'
+Requires-Dist: pandas; extra == 'dev'
 Requires-Dist: pillow; extra == 'dev'
+Requires-Dist: polars; extra == 'dev'
 Requires-Dist: python-dotenv; extra == 'dev'
 Requires-Dist: ruff>=0.6.5; extra == 'dev'
 Requires-Dist: shiny; extra == 'dev'

{chatlas-0.13.1.dist-info → chatlas-0.13.2.dist-info}/RECORD

@@ -4,7 +4,7 @@ chatlas/_batch_chat.py,sha256=1KkHENB-l7VmhCizhdvbJO5WQmRntQS6EvcSJ6VLgvM,5546
 chatlas/_batch_job.py,sha256=2__JIOo_JpcQyAAzO07r6eS4urpAxEc9m7_zsjFieQw,7359
 chatlas/_callbacks.py,sha256=3RpPaOQonTqScjXbaShgKJ1Rc-YxzWerxKRBjVssFnc,1838
 chatlas/_chat.py,sha256=oESXNVzDCJ1DpV_fBRgwK6N_fs_EoJGrlez5dJjqx5c,90664
-chatlas/_content.py,sha256=BdJQ5G5onT9Cf1tNFeXsCWWTD2zSIjWz50FYIk6_DDI,22767
+chatlas/_content.py,sha256=7uRVjGupv-XY6GFmV8455nJt-GuL5F_0YR3Fr1QLntw,24156
 chatlas/_content_image.py,sha256=EUK6wAint-JatLsiwvaPDu4D3W-NcIsDCkzABkXgfDg,8304
 chatlas/_content_pdf.py,sha256=cffeuJxzhUDukQ-Srkmpy62M8X12skYpU_FVq-Wvya4,2420
 chatlas/_display.py,sha256=wyQzSc6z1VqrJfkTLkw1wQcti9s1Pr4qT8UxFJESn4U,4664
@@ -16,7 +16,7 @@ chatlas/_merge.py,sha256=SGj_BetgA7gaOqSBKOhYmW3CYeQKTEehFrXvx3y4OYE,3924
 chatlas/_provider.py,sha256=-5Oyq8tehHJtbBWQUyFUvdqTqZNUcOq2pO5qfAw5oQo,9057
 chatlas/_provider_anthropic.py,sha256=sPPEaDObGuY7JDqU533wlUDA-HaX3sumYWaD3kdG4nE,30964
 chatlas/_provider_cloudflare.py,sha256=vFbqgQPmosopJa9qsVxTkjPn4vYC_wOlgqa6_QmwTho,5227
-chatlas/_provider_databricks.py,sha256=JIOTm0HMe0qVAt8eS0WgGKugBwBdmL80JHLFH59ongU,4850
+chatlas/_provider_databricks.py,sha256=fNLhpS6tF6Uhqyhhfo2YWLpiAxfa-3gLFytEnKVFqVg,5076
 chatlas/_provider_deepseek.py,sha256=6nPtPSo-Po6sD4i8PZJHuI5T2oATpLi5djXFGdlserk,4906
 chatlas/_provider_github.py,sha256=5h7xwgUB0UdFpqr3UmUuFcC9hfn6kJvZDab0uAURIek,5642
 chatlas/_provider_google.py,sha256=0vAFSSk8SVBEtaDmy7Tl0XlHDqR4qwxWJR8HeSgh79E,21060
@@ -24,7 +24,7 @@ chatlas/_provider_groq.py,sha256=XB2JDyuF95CcSbNkgk7JHcuy9KCW7hxTVaONDSjK8U8,367
 chatlas/_provider_huggingface.py,sha256=feJ416X0UdtyoeHZbkgolFf62D7zxNwM7i_X3NYsQQw,4669
 chatlas/_provider_mistral.py,sha256=-p4rut0KCn-PrwnOlvr6lK8-K-OXvc5H9vTX-rCzUkk,5309
 chatlas/_provider_ollama.py,sha256=jFAu4v0NLUwdG_W_nKagBHOah0VKl_auTsgcYinP9rI,4119
-chatlas/_provider_openai.py,sha256=f8ijdXMMHy17VcuA2ImMpaYvaiKjHzcnTdR8LH1xE40,30654
+chatlas/_provider_openai.py,sha256=Uo5EKOoCJHedSdzvkaDzAl0RFrwFQwqoMS29BjH-ejU,30887
 chatlas/_provider_openrouter.py,sha256=9sCXvROVIiUdwfEbkVA-15_kc6ouFUP2uV2MmUe2rFk,4385
 chatlas/_provider_perplexity.py,sha256=5q_LsUCJQ5w-jRveLDMPvZTX-GU2TVURp65mUMyDh10,4248
 chatlas/_provider_portkey.py,sha256=6wKrLZmKVxOqyO6P3HBgWqPe7y1N8une_1wp0aJq7pU,4087
@@ -35,22 +35,22 @@ chatlas/_tools.py,sha256=8rhGOsEviBJXk5Qb-a1RRb_C-DE2T3DOeN6IhblkxqI,12408
 chatlas/_turn.py,sha256=yK7alUxeP8d2iBc7amyz20BtEqcpvX6BCwWZsnlQ5R4,4515
 chatlas/_typing_extensions.py,sha256=BXmbhywjm5ssmyVLGwyP_5TWZMAobzrrgZLYkB6_etE,1076
 chatlas/_utils.py,sha256=Kku2fa1mvTYCr5D28VxE6-fwfy2e2doCi-eKQkLEg4Y,4686
-chatlas/_version.py,sha256=RBWKvLuPH5uJN_RJXTOaV5SvvBYz1PUZvU4D8m3gAvo,706
+chatlas/_version.py,sha256=XZecU3ohLLhnQNT8_JmdvW2idqqOh08RIDcm2qzc5t8,706
 chatlas/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chatlas/data/prices.json,sha256=X6qALp-dWc4nfus9lIqHoKzk3PZDPHTLoxxcN2m6fXc,62645
 chatlas/types/__init__.py,sha256=1n0xrJ7TRIKsZ2z06FLFgGqfKMFtXSIxxPvJ2j0hvPw,850
 chatlas/types/anthropic/__init__.py,sha256=OwubA-DPHYpYo0XyRyAFwftOI0mOxtHzAyhUSLcDx54,417
-chatlas/types/anthropic/_client.py,sha256=t_tnOzzsW1xWNADkNoAuZJYoE9QJ8ie7DQNnFO1pvoM,697
+chatlas/types/anthropic/_client.py,sha256=jOWBeabk_-XVrWHqj1hkiJ7o8Qj1A5QJC6vG_Mz86oc,686
 chatlas/types/anthropic/_client_bedrock.py,sha256=2J6U1QcSx1KwiiHfXs3i4YEXDXw11sp-x3iLOuESrgQ,792
-chatlas/types/anthropic/_submit.py,sha256=X7ER31k7bTZGJ9X9u8Mx-X4enJC13W0VQzt8Wz-mLeQ,3675
+chatlas/types/anthropic/_submit.py,sha256=blwmeJU9iP-YdAwZPG6mrBVkf-3Epq7uSm5VRJcBjP4,3618
 chatlas/types/google/__init__.py,sha256=ZJhi8Kwvio2zp8T1TQqmvdHqkS-Khb6BGESPjREADgo,337
 chatlas/types/google/_client.py,sha256=t7aKbxYq_xOA1Z3RnWcjewifdQFSHi7vKEj6MyKMCJk,729
 chatlas/types/google/_submit.py,sha256=19Ji4fAo1lTCbNSpR6Yi0i64RJwMGBdiZKQcnoDNRwY,1796
 chatlas/types/openai/__init__.py,sha256=Q2RAr1bSH1nHsxICK05nAmKmxdhKmhbBkWD_XHiVSrI,411
-chatlas/types/openai/_client.py,sha256=mAoQftcJIp0ssIhS8q3TIW9u6zTRNtYDmpZJO8L0mC0,849
+chatlas/types/openai/_client.py,sha256=3uKLdhRY1BnyX3dhCShv1eYVXtpABZwbpHbRb6ZxMgk,845
 chatlas/types/openai/_client_azure.py,sha256=Tf_PFRl0QAj4Nk5CD0ZNIO-SRsT39bVkEJlUTry1fb8,960
-chatlas/types/openai/_submit.py,sha256=EDtIUFcNIJ5QAt0wVyBXvUshK8FA9e86wcZDQ_HUOYs,7829
-chatlas-0.13.1.dist-info/METADATA,sha256=gG-XEXaUc9LFlqoyqHa5nNzH_Dz-QHeL6HqxECWHAgM,5635
-chatlas-0.13.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-chatlas-0.13.1.dist-info/licenses/LICENSE,sha256=zyuGzPOC7CcbOaBHsQ3UEyKYRO56KDUkor0OA4LqqDg,1081
-chatlas-0.13.1.dist-info/RECORD,,
+chatlas/types/openai/_submit.py,sha256=0e6YAg-99y0QH1VmkaprCUrvW9wFH5OZT0TrTK7w4_w,7701
+chatlas-0.13.2.dist-info/METADATA,sha256=kpIHWgbil0ST2s_YUirwiAHxQx4bWiWYsTHXGpgRMoM,5751
+chatlas-0.13.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+chatlas-0.13.2.dist-info/licenses/LICENSE,sha256=zyuGzPOC7CcbOaBHsQ3UEyKYRO56KDUkor0OA4LqqDg,1081
+chatlas-0.13.2.dist-info/RECORD,,