Merge pull request #1052 from parea-ai/fix-anthropic-types

fix: import anthropic json delta type

Author: joschkabraun

cookbook/anthropic/tracing_anthropic_tool_use.py

@@ -17,17 +17,24 @@
 
 
 def anthropic_sync(create_kwargs):
-    message = client.beta.tools.messages.create(**create_kwargs)
+    message = client.messages.create(**create_kwargs)
     print(message.content)
 
 
+def anthropic_sync_stream(create_kwargs):
+    message = client.messages.create(stream=True, **create_kwargs)
+    for m in message:
+        print(m)
+
+
 async def async_anthropic(create_kwargs):
-    message = await aclient.beta.tools.messages.create(**create_kwargs)
+    message = await aclient.messages.create(**create_kwargs)
     print(message.content)
 
 
 if __name__ == "__main__":
     anthropic_sync(single_tool_use)
+    anthropic_sync_stream(single_tool_use)
     anthropic_sync(multiple_tool_use)
     anthropic_sync(missing_information)
     # asyncio.run(async_anthropic(single_tool_use))

parea/wrapper/anthropic/anthropic.py

@@ -5,7 +5,12 @@
 from datetime import datetime
 
 from anthropic import AsyncMessageStreamManager, AsyncStream, Client, MessageStreamManager, Stream
-from anthropic.types import ContentBlockDeltaEvent, InputJSONDelta, Message, MessageDeltaEvent, MessageStartEvent, TextBlock, ToolUseBlock
+from anthropic.types import ContentBlockDeltaEvent, Message, MessageDeltaEvent, MessageStartEvent, TextBlock, ToolUseBlock
+
+try:
+    from anthropic.types import InputJSONDelta
+except ImportError:
+    from anthropic.types import InputJsonDelta as InputJSONDelta
 
 from parea.cache.cache import Cache
 from parea.helpers import timezone_aware_now

pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
 [tool.poetry]
 name = "parea-ai"
 packages = [{ include = "parea" }]
-version = "0.2.202"
+version = "0.2.203"
 description = "Parea python sdk"
 readme = "README.md"
 authors = ["joel-parea-ai <[email protected]>"]