Add support for stream timeouts, defaults to None

Author: tsmith023

weaviate/config.py

@@ -56,6 +56,9 @@ class Timeout(BaseModel):
     query: Union[int, float] = Field(default=30, ge=0)
     insert: Union[int, float] = Field(default=90, ge=0)
     init: Union[int, float] = Field(default=2, ge=0)
+    stream: Union[int, float, None] = Field(
+        default=None, ge=0, description="Timeout for streaming operations."
+    )
 
 
 class Proxies(BaseModel):

weaviate/connect/v4.py

@@ -1012,7 +1012,9 @@ def grpc_batch_stream(
         try:
             assert self.grpc_stub is not None
             for msg in self.grpc_stub.BatchStream(
-                request_iterator=requests, metadata=self.grpc_headers()
+                request_iterator=requests,
+                timeout=self.timeout_config.stream,
+                metadata=self.grpc_headers(),
             ):
                 yield msg
         except RpcError as e:
@@ -1232,7 +1234,7 @@ def grpc_batch_stream(
             response_deserializer=batch_pb2.BatchStreamReply.FromString,
         )(
             request_iterator=None,
-            timeout=self.timeout_config.insert,
+            timeout=self.timeout_config.stream,
             metadata=self.grpc_headers(),
         )