fix: backport some fixes + version bump to release (#952)

* fix: ensure response format passed in aio. Ensure reconnecting rather than breaking. Correctly bubble up workflow name if set

Signed-off-by: Casper Nielsen <casper@diagrid.io>

* chore: fix f-string

Signed-off-by: Casper Nielsen <casper@diagrid.io>

* chore: add todo string

Signed-off-by: Casper Nielsen <casper@diagrid.io>

* fix: address conflicts

Signed-off-by: Samantha Coyle <sam@diagrid.io>

* chore(test): increase coverage

Signed-off-by: Casper Nielsen <casper@diagrid.io>

* chore(format): ruff

Signed-off-by: Casper Nielsen <casper@diagrid.io>

* fix: bump versions everywhere for new release

Signed-off-by: Samantha Coyle <sam@diagrid.io>

---------

Signed-off-by: Casper Nielsen <casper@diagrid.io>
Signed-off-by: Samantha Coyle <sam@diagrid.io>
Co-authored-by: Casper Nielsen <casper@diagrid.io>

Author: sicoyle

dapr/aio/clients/grpc/client.py

@@ -25,8 +25,10 @@
 
 import grpc.aio  # type: ignore
 from google.protobuf.any_pb2 import Any as GrpcAny
+from google.protobuf.duration_pb2 import Duration as GrpcDuration
 from google.protobuf.empty_pb2 import Empty as GrpcEmpty
 from google.protobuf.message import Message as GrpcMessage
+from google.protobuf.struct_pb2 import Struct as GrpcStruct
 from grpc import StatusCode  # type: ignore
 from grpc.aio import (  # type: ignore
     AioRpcError,
@@ -1883,6 +1885,8 @@ async def converse_alpha2(
         temperature: Optional[float] = None,
         tools: Optional[List[conversation.ConversationTools]] = None,
         tool_choice: Optional[str] = None,
+        response_format: Optional[GrpcStruct] = None,
+        prompt_cache_retention: Optional[GrpcDuration] = None,
     ) -> conversation.ConversationResponseAlpha2:
         """Invoke an LLM using the conversation API (Alpha2) with tool calling support.
 
@@ -1896,6 +1900,8 @@ async def converse_alpha2(
             temperature: Optional temperature setting for the LLM to optimize for creativity or predictability
             tools: Optional list of tools available for the LLM to call
             tool_choice: Optional control over which tools can be called ('none', 'auto', 'required', or specific tool name)
+            response_format: Optional response format (google.protobuf.struct_pb2.Struct, ex: json_schema for structured output)
+            prompt_cache_retention: Optional retention for prompt cache (google.protobuf.duration_pb2.Duration)
 
         Returns:
             ConversationResponseAlpha2 containing the conversation results with choices and tool calls
@@ -1951,6 +1957,10 @@ async def converse_alpha2(
             request.temperature = temperature
         if tool_choice is not None:
             request.tool_choice = tool_choice
+        if response_format is not None and hasattr(request, 'response_format'):
+            request.response_format.CopyFrom(response_format)
+        if prompt_cache_retention is not None and hasattr(request, 'prompt_cache_retention'):
+            request.prompt_cache_retention.CopyFrom(prompt_cache_retention)
 
         try:
             response, call = await self.retry_policy.run_rpc_async(

dapr/clients/grpc/client.py

@@ -640,6 +640,16 @@ def stream_messages(sub):
                     break
                 except StreamCancelledError:
                     break
+                except Exception:
+                    # Stream died — reconnect via the subscription's own
+                    # reconnect logic (which waits for the sidecar to be healthy).
+                    try:
+                        sub.reconnect_stream()
+                    except Exception:
+                        # Sidecar still unavailable — back off before retrying
+                        # TODO: Make this configurable
+                        time.sleep(5)
+                    continue
 
         def close_subscription():
             subscription.close()

dapr/clients/grpc/subscription.py

@@ -1,3 +1,4 @@
+import logging
 import queue
 import threading
 from typing import Optional
@@ -13,6 +14,8 @@
 )
 from dapr.proto import api_v1, appcallback_v1
 
+logger = logging.getLogger(__name__)
+
 
 class Subscription:
     def __init__(self, stub, pubsub_name, topic, metadata=None, dead_letter_topic=None):
@@ -67,7 +70,7 @@ def outgoing_request_iterator():
     def reconnect_stream(self):
         self.close()
         DaprHealth.wait_for_sidecar()
-        print('Attempting to reconnect...')
+        logger.info('Subscription stream reconnecting...')
         self.start()
 
     def next_message(self):
@@ -84,10 +87,17 @@ def next_message(self):
             message = next(self._stream)
             return SubscriptionMessage(message.event_message)
         except RpcError as e:
-            # If Dapr can't be reached, wait until it's ready and reconnect the stream
-            if e.code() == StatusCode.UNAVAILABLE or e.code() == StatusCode.UNKNOWN:
-                print(
-                    f'gRPC error while reading from stream: {e.details()}, Status Code: {e.code()}'
+            # If Dapr can't be reached, wait until it's ready and reconnect the stream.
+            # INTERNAL covers RST_STREAM from cloud proxies (e.g. Diagrid Cloud).
+            if e.code() in (
+                StatusCode.UNAVAILABLE,
+                StatusCode.UNKNOWN,
+                StatusCode.INTERNAL,
+            ):
+                logger.warning(
+                    'Subscription stream error (%s): %s — reconnecting',
+                    e.code(),
+                    e.details(),
                 )
                 self.reconnect_stream()
             elif e.code() == StatusCode.CANCELLED:
@@ -111,7 +121,7 @@ def respond(self, message, status):
                 raise StreamInactiveError('Stream is not active')
             self._send_queue.put(msg)
         except Exception as e:
-            print(f"Can't send message on inactive stream: {e}")
+            logger.warning(f"Can't send message on inactive stream: {e}")
 
     def respond_success(self, message):
         self.respond(message, TopicEventResponse('success').status)
@@ -135,15 +145,12 @@ def _is_stream_active(self):
             return self._stream_active
 
     def close(self):
+        self._set_stream_inactive()
         if self._stream:
             try:
                 self._stream.cancel()
-                self._set_stream_inactive()
-            except RpcError as e:
-                if e.code() != StatusCode.CANCELLED:
-                    raise Exception(f'Error while closing stream: {e}')
-            except Exception as e:
-                raise Exception(f'Error while closing stream: {e}')
+            except Exception:
+                pass  # Stream already dead — safe to ignore
 
     def __iter__(self):
         return self

dapr/version/version.py

@@ -13,4 +13,4 @@
 limitations under the License.
 """
 
-__version__ = '1.17.1'
+__version__ = '1.17.2'

dev-requirements.txt

@@ -14,7 +14,7 @@ Flask>=1.1
 # needed for auto fix
 ruff===0.14.1
 # needed for dapr-ext-workflow
-durabletask-dapr >= 0.17.1
+durabletask-dapr >= 0.17.2
 # needed for .env file loading in examples
 python-dotenv>=1.0.0
 # needed for enhanced schema generation from function features

ext/dapr-ext-fastapi/dapr/ext/fastapi/version.py

@@ -13,4 +13,4 @@
 limitations under the License.
 """
 
-__version__ = '1.17.1'
+__version__ = '1.17.2'

ext/dapr-ext-fastapi/setup.cfg

@@ -25,7 +25,7 @@ python_requires = >=3.10
 packages = find_namespace:
 include_package_data = True
 install_requires =
-    dapr >= 1.17.1
+    dapr >= 1.17.2
     uvicorn >= 0.11.6
     fastapi >= 0.60.1
 

ext/dapr-ext-grpc/dapr/ext/grpc/version.py

@@ -13,4 +13,4 @@
 limitations under the License.
 """
 
-__version__ = '1.17.1'
+__version__ = '1.17.2'

ext/dapr-ext-grpc/setup.cfg

@@ -24,7 +24,7 @@ python_requires = >=3.10
 packages = find_namespace:
 include_package_data = True
 install_requires =
-    dapr >= 1.17.1
+    dapr >= 1.17.2
     cloudevents >= 1.0.0
 
 [options.packages.find]

ext/dapr-ext-langgraph/dapr/ext/langgraph/version.py

@@ -13,4 +13,4 @@
 limitations under the License.
 """
 
-__version__ = '1.17.1'
+__version__ = '1.17.2'