fix: more ci fixes

Chainlit · Nov 28, 2024 · c5faa02 · c5faa02
1 parent cf98d74
commit c5faa02
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/literalai/api/asynchronous.py b/literalai/api/asynchronous.py
@@ -3,7 +3,6 @@
 
 from typing_extensions import deprecated
 from typing import (
-    TYPE_CHECKING,
     Any,
     Callable,
     Dict,
@@ -148,7 +147,7 @@ def raise_error(error):
             logger.error(f"Failed to {description}: {error}")
             raise Exception(error)
 
-        variables = _prepare_variables(variables)
+        variables = prepare_variables(variables)
 
         async with httpx.AsyncClient(follow_redirects=True) as client:
             response = await client.post(
@@ -443,7 +442,7 @@ async def upload_file(
         # Prepare form data
         form_data = (
             {}
-        )  # type: Dict[str, Union[Tuple[Union[str, None], Any], Tuple[Union[str, None], Any, Any]]]
+        )  # type: Dict[str, Union[tuple[Union[str, None], Any], tuple[Union[str, None], Any, Any]]]
         for field_name, field_value in fields.items():
             form_data[field_name] = (None, field_value)
 

diff --git a/literalai/api/synchronous.py b/literalai/api/synchronous.py
@@ -3,7 +3,6 @@
 
 from typing_extensions import deprecated
 from typing import (
-    TYPE_CHECKING,
     Any,
     Callable,
     Dict,
@@ -105,9 +104,6 @@
 from literalai.observability.thread import Thread
 from literalai.prompt_engineering.prompt import Prompt, ProviderSettings
 
-if TYPE_CHECKING:
-    from typing import Tuple  # noqa: F401
-
 import httpx
 
 from literalai.my_types import PaginatedResponse, User
@@ -150,7 +146,7 @@ def raise_error(error):
             logger.error(f"Failed to {description}: {error}")
             raise Exception(error)
 
-        variables = _prepare_variables(variables)
+        variables = prepare_variables(variables)
         with httpx.Client(follow_redirects=True) as client:
             response = client.post(
                 self.graphql_endpoint,
@@ -441,7 +437,7 @@ def upload_file(
         # Prepare form data
         form_data = (
             {}
-        )  # type: Dict[str, Union[Tuple[Union[str, None], Any], Tuple[Union[str, None], Any, Any]]]
+        )  # type: Dict[str, Union[tuple[Union[str, None], Any], tuple[Union[str, None], Any, Any]]]
         for field_name, field_value in fields.items():
             form_data[field_name] = (None, field_value)