Organize all imports with Trunk

backend/chainlit/haystack/callbacks.py

@@ -1,13 +1,14 @@
 import re
 from typing import Any, Generic, List, Optional, TypeVar
 
-from chainlit import Message
 from chainlit.step import Step
 from chainlit.sync import run_sync
 from haystack.agents import Agent, Tool
 from haystack.agents.agent_step import AgentStep
 from literalai.helper import utc_now
 
+from chainlit import Message
+
 T = TypeVar("T")
 
 

backend/tests/data/test_sql_alchemy.py

@@ -2,13 +2,14 @@
 from pathlib import Path
 
 import pytest
-from chainlit import User
 from chainlit.data.base import BaseStorageClient
 from chainlit.data.sql_alchemy import SQLAlchemyDataLayer
 from chainlit.element import Text
 from sqlalchemy import text
 from sqlalchemy.ext.asyncio import create_async_engine
 
+from chainlit import User
+
 
 @pytest.fixture
 async def data_layer(mock_storage_client: BaseStorageClient, tmp_path: Path):

backend/tests/test_callbacks.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
 
 import pytest
-from chainlit import config
 from chainlit.callbacks import password_auth_callback
 from chainlit.user import User
 
+from chainlit import config
+
 
 @pytest.fixture
 def test_config(monkeypatch: pytest.MonkeyPatch):

cypress/e2e/ask_file/main.py

@@ -1,4 +1,5 @@
 import aiofiles
+
 import chainlit as cl
 
 

cypress/e2e/chat_settings/main.py

@@ -1,6 +1,7 @@
-import chainlit as cl
 from chainlit.input_widget import Select, Slider, Switch
 
+import chainlit as cl
+
 
 @cl.on_chat_start
 async def start():

cypress/e2e/context/main.py

@@ -1,7 +1,8 @@
-import chainlit as cl
 from chainlit.context import context
 from chainlit.sync import make_async, run_sync
 
+import chainlit as cl
+
 
 async def async_function_from_sync():
     await cl.sleep(2)

cypress/e2e/custom_data_layer/sql_alchemy.py

@@ -1,10 +1,11 @@
 from typing import Optional
 
-import chainlit as cl
 import chainlit.data as cl_data
 from chainlit.data.sql_alchemy import SQLAlchemyDataLayer
 from chainlit.data.storage_clients import AzureStorageClient
 
+import chainlit as cl
+
 storage_client = AzureStorageClient(
     account_url="<your_account_url>", container="<your_container>"
 )

cypress/e2e/llama_index_cb/main.py

@@ -1,8 +1,9 @@
-import chainlit as cl
 from llama_index.core.callbacks.schema import CBEventType, EventPayload
 from llama_index.core.llms import ChatMessage, ChatResponse
 from llama_index.core.schema import NodeWithScore, TextNode
 
+import chainlit as cl
+
 
 @cl.on_chat_start
 async def start():

cypress/e2e/plotly/main.py

@@ -1,6 +1,7 @@
-import chainlit as cl
 import plotly.graph_objects as go
 
+import chainlit as cl
+
 
 @cl.on_chat_start
 async def start():