Avoid using is_typeddict

posit-dev · Nov 26, 2024 · b26a435 · b26a435
1 parent 9efe068
commit b26a435
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/shiny/ui/_chat_normalize.py b/shiny/ui/_chat_normalize.py
@@ -1,6 +1,6 @@
 import sys
 from abc import ABC, abstractmethod
-from typing import TYPE_CHECKING, Any, Optional, cast, is_typeddict
+from typing import TYPE_CHECKING, Any, Optional, cast
 
 from htmltools import HTML
 
@@ -236,7 +236,7 @@ def can_normalize(self, message: Any) -> bool:
 
             # Ollama<0.4 used TypedDict (now it uses pydantic)
             # https://github.com/ollama/ollama-python/pull/276
-            if is_typeddict(ChatResponse):
+            if isinstance(ChatResponse, dict):
                 return "message" in message and super().can_normalize(
                     message["message"]
                 )