From cada4e8570614d7b198a0792b68ea0afb7ce9258 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 22 Jun 2024 00:37:49 -0300 Subject: [PATCH] refactor: Add optional type hint for extra parameter in add_result_pair method --- src/backend/base/langflow/graph/edge/schema.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/graph/edge/schema.py b/src/backend/base/langflow/graph/edge/schema.py index ea691beae..628073d13 100644 --- a/src/backend/base/langflow/graph/edge/schema.py +++ b/src/backend/base/langflow/graph/edge/schema.py @@ -1,4 +1,4 @@ -from typing import Any, List +from typing import Optional, Any, List from pydantic import BaseModel @@ -13,7 +13,7 @@ class Payload(BaseModel): def __iter__(self): return iter(self.result_pairs) - def add_result_pair(self, result: Any, extra: Any = None) -> None: + def add_result_pair(self, result: Any, extra: Optional[Any] = None) -> None: self.result_pairs.append(ResultPair(result=result, extra=extra)) def get_last_result_pair(self) -> ResultPair: