|
@@ -37,7 +37,7 @@ class CompletionAppGenerator(MessageBasedAppGenerator):
|
|
|
args: Mapping[str, Any],
|
|
|
invoke_from: InvokeFrom,
|
|
|
streaming: Literal[True],
|
|
|
- ) -> Generator[str, None, None]: ...
|
|
|
+ ) -> Generator[str | Mapping[str, Any], None, None]: ...
|
|
|
|
|
|
@overload
|
|
|
def generate(
|
|
@@ -57,7 +57,7 @@ class CompletionAppGenerator(MessageBasedAppGenerator):
|
|
|
args: Mapping[str, Any],
|
|
|
invoke_from: InvokeFrom,
|
|
|
streaming: bool = False,
|
|
|
- ) -> Union[Mapping[str, Any], Generator[str, None, None]]: ...
|
|
|
+ ) -> Union[Mapping[str, Any], Generator[str | Mapping[str, Any], None, None]]: ...
|
|
|
|
|
|
def generate(
|
|
|
self,
|
|
@@ -66,7 +66,7 @@ class CompletionAppGenerator(MessageBasedAppGenerator):
|
|
|
args: Mapping[str, Any],
|
|
|
invoke_from: InvokeFrom,
|
|
|
streaming: bool = True,
|
|
|
- ) -> Union[Mapping[str, Any], Generator[str, None, None]]:
|
|
|
+ ) -> Union[Mapping[str, Any], Generator[str | Mapping[str, Any], None, None]]:
|
|
|
"""
|
|
|
Generate App response.
|
|
|
|
|
@@ -231,7 +231,7 @@ class CompletionAppGenerator(MessageBasedAppGenerator):
|
|
|
user: Union[Account, EndUser],
|
|
|
invoke_from: InvokeFrom,
|
|
|
stream: bool = True,
|
|
|
- ) -> Union[Mapping, Generator[str, None, None]]:
|
|
|
+ ) -> Union[Mapping, Generator[Mapping | str, None, None]]:
|
|
|
"""
|
|
|
Generate App response.
|
|
|
|