feat: change typing

ComposioHQ · Apr 11, 2024 · e867ed9 · e867ed9
1 parent 982071f
commit e867ed9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/core/composio/sdk/core.py b/core/composio/sdk/core.py
@@ -157,7 +157,7 @@ def execute_action(self, action: Action, params: dict):
         resp = account.execute_action(action, params)
         return resp
 
-    def get_list_of_connections(self, app_name: list[App | str] = None) -> list[ConnectedAccount]:
+    def get_list_of_connections(self, app_name: list[Union[App, str]] = None) -> list[ConnectedAccount]:
         for i, item in enumerate(app_name):
             if isinstance(item, App):
                 app_name[i] = item.value

diff --git a/core/composio/sdk/utils.py b/core/composio/sdk/utils.py
@@ -4,6 +4,8 @@
 from pydantic import BaseModel
 import subprocess
 
+from typing import Union
+
 def _get_enum_key(name):
     return name.upper().replace(' ', '_').replace('-', '_')
 
@@ -55,8 +57,8 @@ def generate_enums():
         f.write(enum_content)
 
 class GitUserInfo(BaseModel):
-    name: str | None
-    email: str | None
+    name: Union[None, str] 
+    email: Union[None, str] 
 
 def get_git_user_info() -> GitUserInfo:
     try: