strawberry-graphql 0.263.0.dev1743450503__tar.gz → 0.263.0.dev1743582446__tar.gz

{strawberry_graphql-0.263.0.dev1743450503 → strawberry_graphql-0.263.0.dev1743582446}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: strawberry-graphql
-Version: 0.263.0.dev1743450503
+Version: 0.263.0.dev1743582446
 Summary: A library for creating GraphQL APIs
 License: MIT
 Keywords: graphql,api,rest,starlette,async

{strawberry_graphql-0.263.0.dev1743450503 → strawberry_graphql-0.263.0.dev1743582446}/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "strawberry-graphql"
 packages = [ { include = "strawberry" } ]
-version = "0.263.0.dev.1743450503"
+version = "0.263.0.dev.1743582446"
 description = "A library for creating GraphQL APIs"
 authors = ["Patrick Arminio <patrick.arminio@gmail.com>"]
 license = "MIT"

{strawberry_graphql-0.263.0.dev1743450503 → strawberry_graphql-0.263.0.dev1743582446}/strawberry/__init__.py

@@ -11,6 +11,7 @@ from .permission import BasePermission
 from .scalars import ID
 from .schema import Schema
 from .schema_directive import schema_directive
+from .streamable import Streamable
 from .types.arguments import argument
 from .types.auto import auto
 from .types.cast import cast
@@ -34,6 +35,7 @@ __all__ = [
     "Parent",
     "Private",
     "Schema",
+    "Streamable",
     "argument",
     "asdict",
     "auto",

{strawberry_graphql-0.263.0.dev1743450503 → strawberry_graphql-0.263.0.dev1743582446}/strawberry/annotation.py

@@ -17,6 +17,7 @@ from typing import (
 )
 from typing_extensions import Self, get_args, get_origin
 
+from strawberry.streamable import StrawberryStreamable
 from strawberry.types.base import (
     StrawberryList,
     StrawberryObjectDefinition,
@@ -141,6 +142,8 @@ class StrawberryAnnotation:
 
         if self._is_lazy_type(evaled_type):
             return evaled_type
+        if self._is_streamable(evaled_type, args):
+            return self.create_list(list[evaled_type])
         if self._is_list(evaled_type):
             return self.create_list(evaled_type)
 
@@ -310,6 +313,10 @@ class StrawberryAnnotation:
             or is_list
         )
 
+    @classmethod
+    def _is_streamable(cls, annotation: Any, args: list[Any]) -> bool:
+        return any(isinstance(arg, StrawberryStreamable) for arg in args)
+
     @classmethod
     def _is_strawberry_type(cls, evaled_type: Any) -> bool:
         # Prevent import cycles

{strawberry_graphql-0.263.0.dev1743450503 → strawberry_graphql-0.263.0.dev1743582446}/strawberry/http/async_base_view.py

@@ -367,6 +367,8 @@ class AsyncBaseHTTPView(
 
                 yield self.encode_multipart_data(response, "-")
 
+                all_pending = result.initial_result.pending
+
                 async for value in result.subsequent_results:
                     response = {
                         "hasNext": value.has_next,
@@ -382,11 +384,13 @@ class AsyncBaseHTTPView(
                     if value.incremental:
                         incremental = []
 
+                        all_pending.extend(value.pending)
+
                         for incremental_value in value.incremental:
                             pending_value = next(
                                 (
                                     v
-                                    for v in result.initial_result.pending
+                                    for v in all_pending
                                     if v.id == incremental_value.id
                                 ),
                                 None,
@@ -397,8 +401,6 @@ class AsyncBaseHTTPView(
                             incremental.append(
                                 {
                                     **incremental_value.formatted,
-                                    # for Apollo
-                                    # content type is `multipart/mixed;deferSpec=20220824,application/json`
                                     "path": pending_value.path,
                                     "label": pending_value.label,
                                 }

{strawberry_graphql-0.263.0.dev1743450503 → strawberry_graphql-0.263.0.dev1743582446}/strawberry/schema/_graphql_core.py

@@ -22,10 +22,7 @@ try:
     )
 
 except ImportError:
-    from typing_extensions import Literal, TypeAlias
-
-    NoneType: TypeAlias = Literal[None]
-    GraphQLIncrementalExecutionResults = NoneType
+    GraphQLIncrementalExecutionResults = type(None)
 
     incremental_execution_directives = []
     experimental_execute_incrementally = None

strawberry_graphql-0.263.0.dev1743582446/strawberry/streamable.py

@@ -0,0 +1,36 @@
+from collections.abc import AsyncGenerator
+from typing import Annotated, TypeVar
+
+
+class StrawberryStreamable: ...
+
+
+T = TypeVar("T")
+
+Streamable = Annotated[AsyncGenerator[T, None], StrawberryStreamable()]
+"""Represents a list that can be streamed using @stream.
+
+Example:
+
+```python
+import strawberry
+from dataclasses import dataclass
+
+
+@strawberry.type
+class Comment:
+    id: strawberry.ID
+    content: str
+
+
+@strawberry.type
+class Article:
+    @strawberry.field
+    @staticmethod
+    async def comments() -> strawberry.Streamable[Comment]:
+        for comment in fetch_comments():
+            yield comment
+```
+"""
+
+__all__ = ["Streamable"]