From 504088bd825f6b671b268a911f719144c810c922 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 22 Nov 2023 21:10:11 -0300
Subject: [PATCH] Refactor schema.py to import typing and reorder imports

---
 src/backend/langflow/interface/custom/schema.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/backend/langflow/interface/custom/schema.py b/src/backend/langflow/interface/custom/schema.py
index 418309cee..b055dd386 100644
--- a/src/backend/langflow/interface/custom/schema.py
+++ b/src/backend/langflow/interface/custom/schema.py
@@ -1,9 +1,8 @@
+from typing import Any, Optional
+
 from pydantic import BaseModel, Field
 
 
-from typing import Optional
-
-
 class ClassCodeDetails(BaseModel):
     """
     A dataclass for storing details about a class.
@@ -26,4 +25,4 @@ class CallableCodeDetails(BaseModel):
     doc: Optional[str] = None
     args: list
     body: list
-    return_type: Optional[str] = None
+    return_type: Optional[Any] = None