chatlas 0.2.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

chatlas/_openai.py

@@ -8,6 +8,7 @@ from pydantic import BaseModel
 from ._chat import Chat
 from ._content import (
     Content,
+    ContentImage,
     ContentImageInline,
     ContentImageRemote,
     ContentJson,
@@ -20,7 +21,7 @@ from ._merge import merge_dicts
 from ._provider import Provider
 from ._tokens import tokens_log
 from ._tools import Tool, basemodel_to_param_schema
-from ._turn import Turn, normalize_turns
+from ._turn import Turn, normalize_turns, user_turn
 from ._utils import MISSING, MISSING_TYPE, is_testing
 
 if TYPE_CHECKING:
@@ -294,10 +295,12 @@ class OpenAIProvider(Provider[ChatCompletion, ChatCompletionChunk, ChatCompletio
             "stream": stream,
             "messages": self._as_message_param(turns),
             "model": self._model,
-            "seed": self._seed,
             **(kwargs or {}),
         }
 
+        if self._seed is not None:
+            kwargs_full["seed"] = self._seed
+
         if tool_schemas:
             kwargs_full["tools"] = tool_schemas
 
@@ -335,7 +338,7 @@ class OpenAIProvider(Provider[ChatCompletion, ChatCompletionChunk, ChatCompletio
             return chunkd
         return merge_dicts(completion, chunkd)
 
-    def stream_turn(self, completion, has_data_model, stream) -> Turn:
+    def stream_turn(self, completion, has_data_model) -> Turn:
         from openai.types.chat import ChatCompletion
 
         delta = completion["choices"][0].pop("delta")  # type: ignore
@@ -343,12 +346,60 @@ class OpenAIProvider(Provider[ChatCompletion, ChatCompletionChunk, ChatCompletio
         completion = ChatCompletion.construct(**completion)
         return self._as_turn(completion, has_data_model)
 
-    async def stream_turn_async(self, completion, has_data_model, stream):
-        return self.stream_turn(completion, has_data_model, stream)
-
     def value_turn(self, completion, has_data_model) -> Turn:
         return self._as_turn(completion, has_data_model)
 
+    def token_count(
+        self,
+        *args: Content | str,
+        tools: dict[str, Tool],
+        data_model: Optional[type[BaseModel]],
+    ) -> int:
+        try:
+            import tiktoken
+        except ImportError:
+            raise ImportError(
+                "The tiktoken package is required for token counting. "
+                "Please install it with `pip install tiktoken`."
+            )
+
+        encoding = tiktoken.encoding_for_model(self._model)
+
+        turn = user_turn(*args)
+
+        # Count the tokens in image contents
+        image_tokens = sum(
+            self._image_token_count(x)
+            for x in turn.contents
+            if isinstance(x, ContentImage)
+        )
+
+        # For other contents, get the token count from the actual message param
+        other_contents = [x for x in turn.contents if not isinstance(x, ContentImage)]
+        other_full = self._as_message_param([Turn("user", other_contents)])
+        other_tokens = len(encoding.encode(str(other_full)))
+
+        return other_tokens + image_tokens
+
+    async def token_count_async(
+        self,
+        *args: Content | str,
+        tools: dict[str, Tool],
+        data_model: Optional[type[BaseModel]],
+    ) -> int:
+        return self.token_count(*args, tools=tools, data_model=data_model)
+
+    @staticmethod
+    def _image_token_count(image: ContentImage) -> int:
+        if isinstance(image, ContentImageRemote) and image.detail == "low":
+            return 85
+        else:
+            # This is just the max token count for an image The highest possible
+            # resolution is 768 x 2048, and 8 tiles of size 512px can fit inside
+            # TODO: this is obviously a very conservative estimate and could be improved
+            # https://platform.openai.com/docs/guides/vision/calculating-costs
+            return 170 * 8 + 85
+
     @staticmethod
     def _as_message_param(turns: list[Turn]) -> list["ChatCompletionMessageParam"]:
         from openai.types.chat import (
@@ -412,7 +463,13 @@ class OpenAIProvider(Provider[ChatCompletion, ChatCompletionChunk, ChatCompletio
                         contents.append({"type": "text", "text": ""})
                     elif isinstance(x, ContentImageRemote):
                         contents.append(
-                            {"type": "image_url", "image_url": {"url": x.url}}
+                            {
+                                "type": "image_url",
+                                "image_url": {
+                                    "url": x.url,
+                                    "detail": x.detail,
+                                },
+                            }
                         )
                     elif isinstance(x, ContentImageInline):
                         contents.append(

chatlas/_provider.py

@@ -14,6 +14,7 @@ from typing import (
 
 from pydantic import BaseModel
 
+from ._content import Content
 from ._tools import Tool
 from ._turn import Turn
 
@@ -124,20 +125,27 @@ class Provider(
         self,
         completion: ChatCompletionDictT,
         has_data_model: bool,
-        stream: Any,
     ) -> Turn: ...
 
     @abstractmethod
-    async def stream_turn_async(
+    def value_turn(
         self,
-        completion: ChatCompletionDictT,
+        completion: ChatCompletionT,
         has_data_model: bool,
-        stream: Any,
     ) -> Turn: ...
 
     @abstractmethod
-    def value_turn(
+    def token_count(
         self,
-        completion: ChatCompletionT,
-        has_data_model: bool,
-    ) -> Turn: ...
+        *args: Content | str,
+        tools: dict[str, Tool],
+        data_model: Optional[type[BaseModel]],
+    ) -> int: ...
+
+    @abstractmethod
+    async def token_count_async(
+        self,
+        *args: Content | str,
+        tools: dict[str, Tool],
+        data_model: Optional[type[BaseModel]],
+    ) -> int: ...

chatlas/types/__init__.py

@@ -1,4 +1,8 @@
-from .._chat import ChatResponse, ChatResponseAsync, SubmitInputArgsT
+from .._chat import (  # noqa: A005
+  ChatResponse,
+  ChatResponseAsync,
+  SubmitInputArgsT,
+)
 from .._content import (
     Content,
     ContentImage,

chatlas/types/anthropic/_client.py

@@ -18,12 +18,4 @@ class ChatClientArgs(TypedDict, total=False):
     default_headers: Optional[Mapping[str, str]]
     default_query: Optional[Mapping[str, object]]
     http_client: httpx.AsyncClient
-    transport: httpx.AsyncBaseTransport
-    proxies: Union[
-        str,
-        httpx.Proxy,
-        dict[str | httpx.URL, Union[None, str, httpx.URL, httpx.Proxy]],
-        None,
-    ]
-    connection_pool_limits: httpx.Limits
     _strict_response_validation: bool

chatlas/types/anthropic/_submit.py

@@ -6,7 +6,6 @@
 from typing import Iterable, Literal, Mapping, Optional, TypedDict, Union
 
 import anthropic
-import anthropic._types
 import anthropic.types.message_param
 import anthropic.types.text_block_param
 import anthropic.types.tool_choice_any_param
@@ -19,7 +18,6 @@ class SubmitInputArgs(TypedDict, total=False):
     max_tokens: int
     messages: Iterable[anthropic.types.message_param.MessageParam]
     model: Union[
-        str,
         Literal[
             "claude-3-5-haiku-latest",
             "claude-3-5-haiku-20241022",
@@ -33,6 +31,7 @@ class SubmitInputArgs(TypedDict, total=False):
             "claude-2.1",
             "claude-2.0",
         ],
+        str,
     ]
     stop_sequences: Union[list[str], anthropic.NotGiven]
     stream: Union[Literal[False], Literal[True], anthropic.NotGiven]
@@ -51,7 +50,7 @@ class SubmitInputArgs(TypedDict, total=False):
     tools: Union[Iterable[anthropic.types.tool_param.ToolParam], anthropic.NotGiven]
     top_k: int | anthropic.NotGiven
     top_p: float | anthropic.NotGiven
-    extra_headers: Optional[Mapping[str, Union[str, anthropic._types.Omit]]]
+    extra_headers: Optional[Mapping[str, Union[str, anthropic.Omit]]]
     extra_query: Optional[Mapping[str, object]]
     extra_body: object | None
     timeout: float | anthropic.Timeout | None | anthropic.NotGiven

chatlas/types/google/_client.py

@@ -3,99 +3,20 @@
 # ---------------------------------------------------------
 
 
-from typing import Any, Callable, Iterable, TypedDict, Union
+from typing import Optional, TypedDict, Union
 
-import google.ai.generativelanguage_v1beta.types.content
-import google.ai.generativelanguage_v1beta.types.file
-import google.ai.generativelanguage_v1beta.types.generative_service
-import google.generativeai.types.content_types
-import google.generativeai.types.file_types
-import google.generativeai.types.generation_types
+import google.auth.credentials
+import google.genai.client
+import google.genai.types
 
 
 class ChatClientArgs(TypedDict, total=False):
-    model_name: str
-    generation_config: Union[
-        google.ai.generativelanguage_v1beta.types.generative_service.GenerationConfig,
-        google.generativeai.types.generation_types.GenerationConfigDict,
-        google.generativeai.types.generation_types.GenerationConfig,
-        None,
-    ]
-    tools: Union[
-        google.generativeai.types.content_types.FunctionLibrary,
-        Iterable[
-            Union[
-                str,
-                google.generativeai.types.content_types.Tool,
-                google.ai.generativelanguage_v1beta.types.content.Tool,
-                google.generativeai.types.content_types.ToolDict,
-                Iterable[
-                    Union[
-                        google.generativeai.types.content_types.FunctionDeclaration,
-                        google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-                        dict[str, Any],
-                        Callable[..., Any],
-                    ]
-                ],
-                google.generativeai.types.content_types.FunctionDeclaration,
-                google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-                dict[str, Any],
-                Callable[..., Any],
-            ]
-        ],
-        str,
-        google.generativeai.types.content_types.Tool,
-        google.ai.generativelanguage_v1beta.types.content.Tool,
-        google.generativeai.types.content_types.ToolDict,
-        Iterable[
-            Union[
-                google.generativeai.types.content_types.FunctionDeclaration,
-                google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-                dict[str, Any],
-                Callable[..., Any],
-            ]
-        ],
-        google.generativeai.types.content_types.FunctionDeclaration,
-        google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-        dict[str, Any],
-        Callable[..., Any],
-        None,
-    ]
-    tool_config: Union[
-        google.generativeai.types.content_types.ToolConfigDict,
-        google.ai.generativelanguage_v1beta.types.content.ToolConfig,
-        None,
-    ]
-    system_instruction: Union[
-        google.ai.generativelanguage_v1beta.types.content.Content,
-        google.generativeai.types.content_types.ContentDict,
-        Iterable[
-            Union[
-                google.ai.generativelanguage_v1beta.types.content.Part,
-                google.generativeai.types.content_types.PartDict,
-                google.ai.generativelanguage_v1beta.types.content.Blob,
-                google.generativeai.types.content_types.BlobDict,
-                Any,
-                str,
-                google.ai.generativelanguage_v1beta.types.content.FunctionCall,
-                google.ai.generativelanguage_v1beta.types.content.FunctionResponse,
-                google.generativeai.types.file_types.FileDataDict,
-                google.ai.generativelanguage_v1beta.types.content.FileData,
-                google.ai.generativelanguage_v1beta.types.file.File,
-                google.generativeai.types.file_types.File,
-            ]
-        ],
-        google.ai.generativelanguage_v1beta.types.content.Part,
-        google.generativeai.types.content_types.PartDict,
-        google.ai.generativelanguage_v1beta.types.content.Blob,
-        google.generativeai.types.content_types.BlobDict,
-        Any,
-        str,
-        google.ai.generativelanguage_v1beta.types.content.FunctionCall,
-        google.ai.generativelanguage_v1beta.types.content.FunctionResponse,
-        google.generativeai.types.file_types.FileDataDict,
-        google.ai.generativelanguage_v1beta.types.content.FileData,
-        google.ai.generativelanguage_v1beta.types.file.File,
-        google.generativeai.types.file_types.File,
-        None,
+    vertexai: Optional[bool]
+    api_key: Optional[str]
+    credentials: Optional[google.auth.credentials.Credentials]
+    project: Optional[str]
+    location: Optional[str]
+    debug_config: Optional[google.genai.client.DebugConfig]
+    http_options: Union[
+        google.genai.types.HttpOptions, google.genai.types.HttpOptionsDict, None
     ]

chatlas/types/google/_submit.py

@@ -3,111 +3,64 @@
 # ---------------------------------------------------------
 
 
-from typing import Any, Callable, Iterable, TypedDict, Union
+from typing import TypedDict, Union
 
-import google.ai.generativelanguage_v1beta.types.content
-import google.ai.generativelanguage_v1beta.types.file
-import google.ai.generativelanguage_v1beta.types.generative_service
-import google.generativeai.types.content_types
-import google.generativeai.types.file_types
-import google.generativeai.types.generation_types
-import google.generativeai.types.helper_types
+import google.genai.types
+import PIL.Image
 
 
 class SubmitInputArgs(TypedDict, total=False):
+    model: str
     contents: Union[
-        google.ai.generativelanguage_v1beta.types.content.Content,
-        google.generativeai.types.content_types.ContentDict,
-        Iterable[
+        list[
             Union[
-                google.ai.generativelanguage_v1beta.types.content.Part,
-                google.generativeai.types.content_types.PartDict,
-                google.ai.generativelanguage_v1beta.types.content.Blob,
-                google.generativeai.types.content_types.BlobDict,
-                Any,
+                google.genai.types.Content,
+                list[
+                    Union[
+                        google.genai.types.File,
+                        google.genai.types.Part,
+                        PIL.Image.Image,
+                        str,
+                    ]
+                ],
+                google.genai.types.File,
+                google.genai.types.Part,
+                PIL.Image.Image,
                 str,
-                google.ai.generativelanguage_v1beta.types.content.FunctionCall,
-                google.ai.generativelanguage_v1beta.types.content.FunctionResponse,
-                google.generativeai.types.file_types.FileDataDict,
-                google.ai.generativelanguage_v1beta.types.content.FileData,
-                google.ai.generativelanguage_v1beta.types.file.File,
-                google.generativeai.types.file_types.File,
             ]
         ],
-        google.ai.generativelanguage_v1beta.types.content.Part,
-        google.generativeai.types.content_types.PartDict,
-        google.ai.generativelanguage_v1beta.types.content.Blob,
-        google.generativeai.types.content_types.BlobDict,
-        Any,
-        str,
-        google.ai.generativelanguage_v1beta.types.content.FunctionCall,
-        google.ai.generativelanguage_v1beta.types.content.FunctionResponse,
-        google.generativeai.types.file_types.FileDataDict,
-        google.ai.generativelanguage_v1beta.types.content.FileData,
-        google.ai.generativelanguage_v1beta.types.file.File,
-        google.generativeai.types.file_types.File,
-        Iterable[
+        google.genai.types.Content,
+        list[
             Union[
-                google.ai.generativelanguage_v1beta.types.content.Content,
-                google.generativeai.types.content_types.ContentDict,
+                google.genai.types.File, google.genai.types.Part, PIL.Image.Image, str
             ]
         ],
-        None,
-    ]
-    generation_config: Union[
-        google.ai.generativelanguage_v1beta.types.generative_service.GenerationConfig,
-        google.generativeai.types.generation_types.GenerationConfigDict,
-        google.generativeai.types.generation_types.GenerationConfig,
-        None,
-    ]
-    stream: bool
-    tools: Union[
-        google.generativeai.types.content_types.FunctionLibrary,
-        Iterable[
+        google.genai.types.File,
+        google.genai.types.Part,
+        PIL.Image.Image,
+        str,
+        list[
             Union[
-                str,
-                google.generativeai.types.content_types.Tool,
-                google.ai.generativelanguage_v1beta.types.content.Tool,
-                google.generativeai.types.content_types.ToolDict,
-                Iterable[
+                google.genai.types.Content,
+                list[
                     Union[
-                        google.generativeai.types.content_types.FunctionDeclaration,
-                        google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-                        dict[str, Any],
-                        Callable[..., Any],
+                        google.genai.types.File,
+                        google.genai.types.Part,
+                        PIL.Image.Image,
+                        str,
                     ]
                 ],
-                google.generativeai.types.content_types.FunctionDeclaration,
-                google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-                dict[str, Any],
-                Callable[..., Any],
-            ]
-        ],
-        str,
-        google.generativeai.types.content_types.Tool,
-        google.ai.generativelanguage_v1beta.types.content.Tool,
-        google.generativeai.types.content_types.ToolDict,
-        Iterable[
-            Union[
-                google.generativeai.types.content_types.FunctionDeclaration,
-                google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-                dict[str, Any],
-                Callable[..., Any],
+                google.genai.types.File,
+                google.genai.types.Part,
+                PIL.Image.Image,
+                str,
+                google.genai.types.ContentDict,
             ]
         ],
-        google.generativeai.types.content_types.FunctionDeclaration,
-        google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-        dict[str, Any],
-        Callable[..., Any],
-        None,
-    ]
-    tool_config: Union[
-        google.generativeai.types.content_types.ToolConfigDict,
-        google.ai.generativelanguage_v1beta.types.content.ToolConfig,
-        None,
+        google.genai.types.ContentDict,
     ]
-    request_options: Union[
-        google.generativeai.types.helper_types.RequestOptions,
-        google.generativeai.types.helper_types.RequestOptionsDict,
+    config: Union[
+        google.genai.types.GenerateContentConfig,
+        google.genai.types.GenerateContentConfigDict,
         None,
     ]

chatlas/types/openai/_client.py

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

chatlas/types/openai/_client_azure.py

@@ -17,6 +17,7 @@ class ChatAzureClientArgs(TypedDict, total=False):
     organization: str | None
     project: str | None
     base_url: str | None
+    websocket_base_url: str | httpx.URL | None
     timeout: float | openai.Timeout | None | openai.NotGiven
     max_retries: int
     default_headers: Optional[Mapping[str, str]]

chatlas/types/openai/_submit.py

@@ -8,6 +8,7 @@ from typing import Iterable, Literal, Mapping, Optional, TypedDict, Union
 import openai
 import openai.types.chat.chat_completion_assistant_message_param
 import openai.types.chat.chat_completion_audio_param
+import openai.types.chat.chat_completion_developer_message_param
 import openai.types.chat.chat_completion_function_call_option_param
 import openai.types.chat.chat_completion_function_message_param
 import openai.types.chat.chat_completion_named_tool_choice_param
@@ -26,6 +27,7 @@ import openai.types.shared_params.response_format_text
 class SubmitInputArgs(TypedDict, total=False):
     messages: Iterable[
         Union[
+            openai.types.chat.chat_completion_developer_message_param.ChatCompletionDeveloperMessageParam,
             openai.types.chat.chat_completion_system_message_param.ChatCompletionSystemMessageParam,
             openai.types.chat.chat_completion_user_message_param.ChatCompletionUserMessageParam,
             openai.types.chat.chat_completion_assistant_message_param.ChatCompletionAssistantMessageParam,
@@ -36,6 +38,10 @@ class SubmitInputArgs(TypedDict, total=False):
     model: Union[
         str,
         Literal[
+            "o3-mini",
+            "o3-mini-2025-01-31",
+            "o1",
+            "o1-2024-12-17",
             "o1-preview",
             "o1-preview-2024-09-12",
             "o1-mini",
@@ -44,10 +50,11 @@ class SubmitInputArgs(TypedDict, total=False):
             "gpt-4o-2024-11-20",
             "gpt-4o-2024-08-06",
             "gpt-4o-2024-05-13",
-            "gpt-4o-realtime-preview",
-            "gpt-4o-realtime-preview-2024-10-01",
             "gpt-4o-audio-preview",
             "gpt-4o-audio-preview-2024-10-01",
+            "gpt-4o-audio-preview-2024-12-17",
+            "gpt-4o-mini-audio-preview",
+            "gpt-4o-mini-audio-preview-2024-12-17",
             "chatgpt-4o-latest",
             "gpt-4o-mini",
             "gpt-4o-mini-2024-07-18",
@@ -100,6 +107,7 @@ class SubmitInputArgs(TypedDict, total=False):
         openai.NotGiven,
     ]
     presence_penalty: Union[float, None, openai.NotGiven]
+    reasoning_effort: Union[Literal["low", "medium", "high"], None, openai.NotGiven]
     response_format: Union[
         openai.types.shared_params.response_format_text.ResponseFormatText,
         openai.types.shared_params.response_format_json_object.ResponseFormatJSONObject,

{chatlas-0.2.0.dist-info → chatlas-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: chatlas
-Version: 0.2.0
+Version: 0.4.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
@@ -22,7 +22,7 @@ Requires-Dist: pydantic>=2.0
 Requires-Dist: rich
 Provides-Extra: dev
 Requires-Dist: anthropic[bedrock]; extra == 'dev'
-Requires-Dist: google-generativeai>=0.8.3; extra == 'dev'
+Requires-Dist: google-genai>=1.2.0; extra == 'dev'
 Requires-Dist: matplotlib; extra == 'dev'
 Requires-Dist: numpy>1.24.4; extra == 'dev'
 Requires-Dist: openai; extra == 'dev'
@@ -30,15 +30,18 @@ Requires-Dist: pillow; extra == 'dev'
 Requires-Dist: python-dotenv; extra == 'dev'
 Requires-Dist: ruff>=0.6.5; extra == 'dev'
 Requires-Dist: shiny; extra == 'dev'
+Requires-Dist: tiktoken; extra == 'dev'
 Provides-Extra: docs
 Requires-Dist: griffe>=1; extra == 'docs'
 Requires-Dist: ipykernel; extra == 'docs'
 Requires-Dist: ipywidgets; extra == 'docs'
 Requires-Dist: nbclient; extra == 'docs'
 Requires-Dist: nbformat; extra == 'docs'
+Requires-Dist: numpy; extra == 'docs'
 Requires-Dist: pandas; extra == 'docs'
 Requires-Dist: pyyaml; extra == 'docs'
 Requires-Dist: quartodoc>=0.7; extra == 'docs'
+Requires-Dist: sentence-transformers; extra == 'docs'
 Provides-Extra: test
 Requires-Dist: pyright>=1.1.379; extra == 'test'
 Requires-Dist: pytest-asyncio; extra == 'test'
@@ -46,14 +49,24 @@ Requires-Dist: pytest>=8.3.2; extra == 'test'
 Requires-Dist: syrupy>=4; extra == 'test'
 Description-Content-Type: text/markdown
 
-# chatlas
+<h1 class="unnumbered unlisted"> chatlas <a href="https://posit-dev.github.io/chatlas"><img src="docs/images/logo.png" align="right" height="138" alt="chatlas website" /></a> </h1> 
+
+
+
+<p>
+<!-- badges start -->
+<a href="https://pypi.org/project/chatlas/"><img alt="PyPI" src="https://img.shields.io/pypi/v/chatlas?logo=python&logoColor=white&color=orange"></a>
+<a href="https://choosealicense.com/licenses/mit/"><img src="https://img.shields.io/badge/License-MIT-blue.svg" alt="MIT License"></a>
+<a href="https://github.com/posit-dev/chatlas"><img src="https://github.com/posit-dev/chatlas/actions/workflows/test.yml/badge.svg?branch=main" alt="Python Tests"></a>
+<!-- badges end -->
+</p>
 
 chatlas provides a simple and unified interface across large language model (llm) providers in Python. 
-It abstracts away complexity from common tasks like streaming chat interfaces, tool calling, structured output, and much more.
-chatlas helps you prototype faster without painting you into a corner; for example, switching providers is as easy as changing one line of code, but provider specific features are still accessible when needed.
-Developer experience is also a key focus of chatlas: typing support, rich console output, and built-in tooling are all included.
+It helps you prototype faster by abstracting away complexity from common tasks like streaming chat interfaces, tool calling, structured output, and much more.
+Switching providers is also as easy as changing one line of code, but you can also reach for provider-specific features when you need them.
+Developer experience is also a key focus of chatlas: typing support, rich console output, and extension points are all included.
 
-(Looking for something similar to chatlas, but in R? Check out [elmer](https://elmer.tidyverse.org/)!)
+(Looking for something similar to chatlas, but in R? Check out [ellmer](https://ellmer.tidyverse.org/)!)
 
 ## Install
 
@@ -85,6 +98,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).
+* Vertex AI: [`ChatVertex()`](https://posit-dev.github.io/chatlas/reference/ChatVertex.html).
 
 To use a model provider that isn't listed here, you have two options:
 
@@ -123,7 +137,7 @@ From a `chat` instance, it's simple to start a web-based or terminal-based chat
 chat.app()
 ```
 
-<div style="display:flex;justify-content:center;">
+<div align="center">
 <img width="500" alt="A web app for chatting with an LLM via chatlas" src="https://github.com/user-attachments/assets/e43f60cb-3686-435a-bd11-8215cb024d2e" class="border rounded">
 </div>
 
@@ -279,7 +293,7 @@ asyncio.run(main())
 
 `chatlas` has full typing support, meaning that, among other things, autocompletion just works in your favorite editor:
 
-<div style="display:flex;justify-content:center;">
+<div align="center">
 <img width="500" alt="Autocompleting model options in ChatOpenAI" src="https://github.com/user-attachments/assets/163d6d8a-7d58-422d-b3af-cc9f2adee759" class="rounded">
 </div>
 
@@ -299,7 +313,7 @@ This shows important information like tool call results, finish reasons, and mor
 If the problem isn't self-evident, you can also reach into the `.get_last_turn()`, which contains the full response object, with full details about the completion.
 
 
-<div style="display:flex;justify-content:center;">
+<div align="center">
   <img width="500" alt="Turn completion details with typing support" src="https://github.com/user-attachments/assets/eaea338d-e44a-4e23-84a7-2e998d8af3ba" class="rounded">
 </div>