Just require python 3.11 with Anthropic

Author: cpsievert

pkg-py/src/shinychat/_chat_normalize.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import sys
 from functools import singledispatch
 
 from htmltools import HTML, Tagifiable
@@ -180,25 +181,6 @@ def _(chunk: ChatCompletionChunk) -> ChatMessage:
 
 try:
     from anthropic.types import Message as AnthropicMessage
-    from anthropic.types import (
-        RawContentBlockDeltaEvent,
-        RawContentBlockStartEvent,
-        RawContentBlockStopEvent,
-        RawMessageDeltaEvent,
-        RawMessageStartEvent,
-        RawMessageStopEvent,
-    )
-
-    # Create a non-annotated type alias for RawMessageStreamEvent
-    # (so it works with singledispatch)
-    RawMessageStreamEvent = (
-        RawMessageStartEvent
-        | RawMessageDeltaEvent
-        | RawMessageStopEvent
-        | RawContentBlockStartEvent
-        | RawContentBlockDeltaEvent
-        | RawContentBlockStopEvent
-    )
 
     @get_message_content.register
     def _(message: AnthropicMessage) -> ChatMessage:
@@ -210,18 +192,22 @@ def _(message: AnthropicMessage) -> ChatMessage:
             )
         return ChatMessage(content=content.text, role="assistant")
 
-    @get_message_chunk_content.register
-    def _(chunk: RawMessageStreamEvent) -> ChatMessage:
-        content = ""
-        if chunk.type == "content_block_delta":
-            if chunk.delta.type != "text_delta":
-                raise ValueError(
-                    f"Anthropic message delta type {chunk.delta.type} not supported. "
-                    "Only 'text_delta' type is supported"
-                )
-            content = chunk.delta.text
-
-        return ChatMessage(content=content, role="assistant")
+    # Old versions of singledispatch doesn't seem to support union types
+    if sys.version_info >= (3, 11):
+        from anthropic.types import RawMessageStreamEvent
+
+        @get_message_chunk_content.register
+        def _(chunk: RawMessageStreamEvent) -> ChatMessage:
+            content = ""
+            if chunk.type == "content_block_delta":
+                if chunk.delta.type != "text_delta":
+                    raise ValueError(
+                        f"Anthropic message delta type {chunk.delta.type} not supported. "
+                        "Only 'text_delta' type is supported"
+                    )
+                content = chunk.delta.text
+
+            return ChatMessage(content=content, role="assistant")
 except ImportError:
     pass
 

pyproject.toml

@@ -24,9 +24,9 @@ Changelog = "https://github.com/posit-dev/shinychat/blob/main/pkg-py/CHANGELOG.m
 
 [project.optional-dependencies]
 providers = [
-    "anthropic",
+    "anthropic;python_version>='3.11'",
     "chatlas>=0.6.1",
-    "google-generativeai;python_version>='3.9'",
+    "google-generativeai",
     "langchain-core",
     "ollama>=0.4.0",
     "openai",