chatlas 0.6.1__tar.gz → 0.7.0__tar.gz

{chatlas-0.6.1 → chatlas-0.7.0}/.github/workflows/test.yml

@@ -29,6 +29,8 @@ jobs:
       ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
       GOOGLE_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
       AZURE_OPENAI_API_KEY: ${{ secrets.AZURE_OPENAI_API_KEY }}
+      DATABRICKS_HOST: ${{ secrets.DATABRICKS_HOST }}
+      DATABRICKS_TOKEN: ${{ secrets.DATABRICKS_TOKEN }}
       # Free tier of Google is rate limited, so we only test on 3.12
       TEST_GOOGLE: ${{ matrix.config.test_google }}
       # Free tier of Azure is rate limited, so we only test on 3.12

{chatlas-0.6.1 → chatlas-0.7.0}/CHANGELOG.md

@@ -7,6 +7,38 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 -->
 
+## [0.7.0] - 2025-04-22
+
+### New features
+
+* Added `ChatDatabricks()`, for chatting with Databrick's [foundation models](https://docs.databricks.com/aws/en/machine-learning/model-serving/score-foundation-models). (#82)
+* `.stream()` and `.stream_async()` gain a `content` argument. Set this to `"all"` to include `ContentToolResult`/`ContentToolRequest` objects in the stream. (#75)
+* `ContentToolResult`/`ContentToolRequest` are now exported to `chatlas` namespace. (#75)
+* `ContentToolResult`/`ContentToolRequest` gain a `.tagify()` method so they render sensibly in a Shiny app. (#75)
+* A tool can now return a `ContentToolResult`. This is useful for: 
+    * Specifying the format used for sending the tool result to the chat model (`model_format`). (#87)
+    * Custom rendering of the tool result (by overriding relevant methods in a subclass). (#75)
+* `Chat` gains a new `.current_display` property. When a `.chat()` or `.stream()` is currently active, this property returns an object with a `.echo()` method (to echo new content to the display). This is primarily useful for displaying custom content during a tool call. (#79)
+
+### Improvements
+
+* When a tool call ends in failure, a warning is now raised and the stacktrace is printed. (#79)
+* Several improvements to `ChatSnowflake()`:
+  * `.extract_data()` is now supported.
+  *  `async` methods are now supported. (#81)
+  * Fixed an issue with more than one session being active at once. (#83)
+* `ChatAnthropic()` no longer chokes after receiving an output that consists only of whitespace. (#86)
+* `orjson` is now used for JSON loading and dumping. (#87)
+
+### Changes
+
+* The `echo` argument of the `.chat()` method defaults to a new value of `"output"`. As a result, tool requests and results are now echoed by default. To revert to the previous behavior, set `echo="text"`. (#78)
+* Tool results are now dumped to JSON by default before being sent to the model. To revert to the previous behavior, have the tool return a `ContentToolResult` with `model_format="str"`. (#87)
+
+### Breaking changes
+
+* The `.export()` method's `include` argument has been renamed to `content` (to match `.stream()`). (#75)
+
 ## [0.6.1] - 2025-04-03
 
 ### Bug fixes

{chatlas-0.6.1 → chatlas-0.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chatlas
-Version: 0.6.1
+Version: 0.7.0
 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
@@ -18,6 +18,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.9
 Requires-Dist: jinja2
+Requires-Dist: orjson
 Requires-Dist: pydantic>=2.0
 Requires-Dist: requests
 Requires-Dist: rich
@@ -27,9 +28,13 @@ Provides-Extra: azure-openai
 Requires-Dist: openai; extra == 'azure-openai'
 Provides-Extra: bedrock-anthropic
 Requires-Dist: anthropic[bedrock]; extra == 'bedrock-anthropic'
+Provides-Extra: databricks
+Requires-Dist: databricks-sdk[openai]; extra == 'databricks'
 Provides-Extra: dev
 Requires-Dist: anthropic[bedrock]; extra == 'dev'
+Requires-Dist: databricks-sdk; extra == 'dev'
 Requires-Dist: google-genai>=1.2.0; extra == 'dev'
+Requires-Dist: htmltools; extra == 'dev'
 Requires-Dist: matplotlib; extra == 'dev'
 Requires-Dist: numpy>1.24.4; extra == 'dev'
 Requires-Dist: openai; extra == 'dev'
@@ -38,6 +43,7 @@ Requires-Dist: python-dotenv; extra == 'dev'
 Requires-Dist: ruff>=0.6.5; extra == 'dev'
 Requires-Dist: shiny; extra == 'dev'
 Requires-Dist: snowflake-ml-python; extra == 'dev'
+Requires-Dist: tenacity; extra == 'dev'
 Requires-Dist: tiktoken; extra == 'dev'
 Requires-Dist: torch; (python_version <= '3.11') and extra == 'dev'
 Provides-Extra: docs
@@ -123,6 +129,7 @@ It also supports the following enterprise cloud providers:
 
 * AWS Bedrock: [`ChatBedrockAnthropic()`](https://posit-dev.github.io/chatlas/reference/ChatBedrockAnthropic.html).
 * Azure OpenAI: [`ChatAzureOpenAI()`](https://posit-dev.github.io/chatlas/reference/ChatAzureOpenAI.html).
+* Databricks: [`ChatDatabricks()`](https://posit-dev.github.io/chatlas/reference/ChatDatabricks.html).
 * Snowflake Cortex: [`ChatSnowflake()`](https://posit-dev.github.io/chatlas/reference/ChatSnowflake.html).
 * Vertex AI: [`ChatVertex()`](https://posit-dev.github.io/chatlas/reference/ChatVertex.html).
 

{chatlas-0.6.1 → chatlas-0.7.0}/README.md

@@ -47,6 +47,7 @@ It also supports the following enterprise cloud providers:
 
 * AWS Bedrock: [`ChatBedrockAnthropic()`](https://posit-dev.github.io/chatlas/reference/ChatBedrockAnthropic.html).
 * Azure OpenAI: [`ChatAzureOpenAI()`](https://posit-dev.github.io/chatlas/reference/ChatAzureOpenAI.html).
+* Databricks: [`ChatDatabricks()`](https://posit-dev.github.io/chatlas/reference/ChatDatabricks.html).
 * Snowflake Cortex: [`ChatSnowflake()`](https://posit-dev.github.io/chatlas/reference/ChatSnowflake.html).
 * Vertex AI: [`ChatVertex()`](https://posit-dev.github.io/chatlas/reference/ChatVertex.html).
 

{chatlas-0.6.1 → chatlas-0.7.0}/chatlas/__init__.py

@@ -2,8 +2,10 @@ from . import types
 from ._anthropic import ChatAnthropic, ChatBedrockAnthropic
 from ._auto import ChatAuto
 from ._chat import Chat
+from ._content import ContentToolRequest, ContentToolResult
 from ._content_image import content_image_file, content_image_plot, content_image_url
 from ._content_pdf import content_pdf_file, content_pdf_url
+from ._databricks import ChatDatabricks
 from ._github import ChatGithub
 from ._google import ChatGoogle, ChatVertex
 from ._groq import ChatGroq
@@ -26,6 +28,7 @@ __all__ = (
     "ChatAnthropic",
     "ChatAuto",
     "ChatBedrockAnthropic",
+    "ChatDatabricks",
     "ChatGithub",
     "ChatGoogle",
     "ChatGroq",
@@ -41,6 +44,8 @@ __all__ = (
     "content_image_url",
     "content_pdf_file",
     "content_pdf_url",
+    "ContentToolRequest",
+    "ContentToolResult",
     "interpolate",
     "interpolate_file",
     "Provider",

{chatlas-0.6.1 → chatlas-0.7.0}/chatlas/_anthropic.py

@@ -1,10 +1,10 @@
 from __future__ import annotations
 
 import base64
-import json
 import warnings
 from typing import TYPE_CHECKING, Any, Literal, Optional, Union, cast, overload
 
+import orjson
 from pydantic import BaseModel
 
 from ._chat import Chat
@@ -366,8 +366,8 @@ class AnthropicProvider(Provider[Message, RawMessageStreamEvent, Message]):
             this_content = completion.content[chunk.index]
             if this_content.type == "tool_use" and isinstance(this_content.input, str):
                 try:
-                    this_content.input = json.loads(this_content.input or "{}")
-                except json.JSONDecodeError as e:
+                    this_content.input = orjson.loads(this_content.input or "{}")
+                except orjson.JSONDecodeError as e:
                     raise ValueError(f"Invalid JSON input: {e}")
         elif chunk.type == "message_delta":
             completion.stop_reason = chunk.delta.stop_reason
@@ -451,7 +451,10 @@ class AnthropicProvider(Provider[Message, RawMessageStreamEvent, Message]):
     @staticmethod
     def _as_content_block(content: Content) -> "ContentBlockParam":
         if isinstance(content, ContentText):
-            return {"text": content.text, "type": "text"}
+            text = content.text
+            if text == "" or text.isspace():
+                text = "[empty string]"
+            return {"type": "text", "text": text}
         elif isinstance(content, ContentJson):
             return {"text": "<structured data/>", "type": "text"}
         elif isinstance(content, ContentPDF):
@@ -485,12 +488,15 @@ class AnthropicProvider(Provider[Message, RawMessageStreamEvent, Message]):
                 "input": content.arguments,
             }
         elif isinstance(content, ContentToolResult):
-            return {
+            res: ToolResultBlockParam = {
                 "type": "tool_result",
                 "tool_use_id": content.id,
-                "content": content.get_final_value(),
                 "is_error": content.error is not None,
             }
+            # Anthropic supports non-text contents like ImageBlockParam
+            res["content"] = content.get_model_value()  # type: ignore
+            return res
+
         raise ValueError(f"Unknown content type: {type(content)}")
 
     @staticmethod

{chatlas-0.6.1 → chatlas-0.7.0}/chatlas/_auto.py

@@ -1,11 +1,13 @@
 from __future__ import annotations
 
-import json
 import os
 from typing import Callable, Literal, Optional
 
+import orjson
+
 from ._anthropic import ChatAnthropic, ChatBedrockAnthropic
 from ._chat import Chat
+from ._databricks import ChatDatabricks
 from ._github import ChatGithub
 from ._google import ChatGoogle, ChatVertex
 from ._groq import ChatGroq
@@ -18,6 +20,7 @@ from ._turn import Turn
 AutoProviders = Literal[
     "anthropic",
     "bedrock-anthropic",
+    "databricks",
     "github",
     "google",
     "groq",
@@ -32,6 +35,7 @@ AutoProviders = Literal[
 _provider_chat_model_map: dict[AutoProviders, Callable[..., Chat]] = {
     "anthropic": ChatAnthropic,
     "bedrock-anthropic": ChatBedrockAnthropic,
+    "databricks": ChatDatabricks,
     "github": ChatGithub,
     "google": ChatGoogle,
     "groq": ChatGroq,
@@ -55,7 +59,7 @@ def ChatAuto(
     """
     Use environment variables (env vars) to configure the Chat provider and model.
 
-    Creates a `:class:~chatlas.Chat` instance based on the specified provider.
+    Creates a :class:`~chatlas.Chat` instance based on the specified provider.
     The provider may be specified through the `provider` parameter and/or the
     `CHATLAS_CHAT_PROVIDER` env var. If both are set, the env var takes
     precedence. Similarly, the provider's model may be specified through the
@@ -175,7 +179,7 @@ def ChatAuto(
 
     env_kwargs = {}
     if env_kwargs_str := os.environ.get("CHATLAS_CHAT_ARGS"):
-        env_kwargs = json.loads(env_kwargs_str)
+        env_kwargs = orjson.loads(env_kwargs_str)
 
     kwargs = {**kwargs, **env_kwargs, **base_args}
     kwargs = {k: v for k, v in kwargs.items() if v is not None}