chainlit 0.4.2__tar.gz → 0.4.3__tar.gz

{chainlit-0.4.2 → chainlit-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: chainlit
-Version: 0.4.2
+Version: 0.4.3
 Summary: A faster way to build chatbot UIs.
 Home-page: https://github.com/Chainlit/chainlit
 License: Apache-2.0 license
@@ -19,8 +19,9 @@ Requires-Dist: asyncer (>=0.0.2,<0.0.3)
 Requires-Dist: auth0-python (>=4.1.1,<5.0.0)
 Requires-Dist: click (>=8.1.3,<9.0.0)
 Requires-Dist: dataclasses_json (>=0.5.7,<0.6.0)
-Requires-Dist: fastapi (>=0.96.0,<0.97.0)
+Requires-Dist: fastapi (>=0.97.0,<0.98.0)
 Requires-Dist: fastapi-socketio (>=0.0.10,<0.0.11)
+Requires-Dist: nest-asyncio (>=1.5.6,<2.0.0)
 Requires-Dist: openai (>=0.27.7,<0.28.0)
 Requires-Dist: prisma (>=0.9.0,<0.10.0)
 Requires-Dist: pydantic (>=1.10.8,<2.0.0)

chainlit-0.4.3/chainlit/__init__.py

@@ -0,0 +1,165 @@
+from dotenv import load_dotenv
+from typing import Callable, Any, TYPE_CHECKING
+import os
+import asyncio
+
+if TYPE_CHECKING:
+    from chainlit.client.base import BaseClient
+
+from chainlit.lc import (
+    LANGCHAIN_INSTALLED,
+    langchain_factory,
+    langchain_postprocess,
+    langchain_run,
+    langchain_rename,
+)
+from chainlit.llama_index import LLAMA_INDEX_INSTALLED, llama_index_factory
+from chainlit.langflow import langflow_factory
+from chainlit.utils import wrap_user_function
+from chainlit.config import config
+from chainlit.telemetry import trace
+from chainlit.version import __version__
+from chainlit.logger import logger
+from chainlit.types import LLMSettings
+from chainlit.action import Action
+from chainlit.element import Image, Text, Pdf, Avatar, Pyplot
+from chainlit.message import Message, ErrorMessage, AskUserMessage, AskFileMessage
+from chainlit.user_session import user_session
+from chainlit.sync import run_sync, make_async
+from chainlit.cache import cache
+
+if LANGCHAIN_INSTALLED:
+    from chainlit.lc.callbacks import (
+        LangchainCallbackHandler,
+        AsyncLangchainCallbackHandler,
+    )
+
+if LLAMA_INDEX_INSTALLED:
+    from chainlit.llama_index.callbacks import (
+        LlamaIndexCallbackHandler,
+    )
+
+
+env_found = load_dotenv(dotenv_path=os.path.join(os.getcwd(), ".env"))
+
+if env_found:
+    logger.info("Loaded .env file")
+
+
+@trace
+def on_message(func: Callable) -> Callable:
+    """
+    Framework agnostic decorator to react to messages coming from the UI.
+    The decorated function is called every time a new message is received.
+
+    Args:
+        func (Callable[[str], Any]): The function to be called when a new message is received. Takes the input message.
+
+    Returns:
+        Callable[[str], Any]: The decorated on_message function.
+    """
+
+    config.code.on_message = wrap_user_function(func)
+    return func
+
+
+@trace
+def on_chat_start(func: Callable) -> Callable:
+    """
+    Hook to react to the user websocket connection event.
+
+    Args:
+        func (Callable[], Any]): The connection hook to execute.
+
+    Returns:
+        Callable[], Any]: The decorated hook.
+    """
+
+    config.code.on_chat_start = wrap_user_function(func, with_task=True)
+    return func
+
+
+@trace
+def on_stop(func: Callable) -> Callable:
+    """
+    Hook to react to the user stopping a conversation.
+
+    Args:
+        func (Callable[[], Any]): The stop hook to execute.
+
+    Returns:
+        Callable[[], Any]: The decorated stop hook.
+    """
+
+    config.code.on_stop = wrap_user_function(func)
+    return func
+
+
+def action_callback(name: str) -> Callable:
+    """
+    Callback to call when an action is clicked in the UI.
+
+    Args:
+        func (Callable[[Action], Any]): The action callback to execute. First parameter is the action.
+    """
+
+    def decorator(func: Callable[[Action], Any]):
+        config.code.action_callbacks[name] = wrap_user_function(func, with_task=True)
+        return func
+
+    return decorator
+
+
+@trace
+def client_factory(func: Callable[[], "BaseClient"]) -> Callable[[], "BaseClient"]:
+    """
+    Callback to call when to initialize the custom client.
+
+    Args:
+        func (Callable[[str], BaseClient]): The action callback to execute. First parameter is the session id.
+    """
+
+    config.code.client_factory = func
+    return func
+
+
+def sleep(duration: int):
+    """
+    Sleep for a given duration.
+    Args:
+        duration (int): The duration in seconds.
+    """
+    return asyncio.sleep(duration)
+
+
+__all__ = [
+    "user_session",
+    "LLMSettings",
+    "Action",
+    "Pdf",
+    "Image",
+    "Text",
+    "Avatar",
+    "Pyplot",
+    "Message",
+    "ErrorMessage",
+    "AskUserMessage",
+    "AskFileMessage",
+    "langchain_factory",
+    "langchain_postprocess",
+    "langchain_run",
+    "langchain_rename",
+    "llama_index_factory",
+    "langflow_factory",
+    "on_chat_start",
+    "on_stop",
+    "action_callback",
+    "sleep",
+    "LangchainCallbackHandler",
+    "AsyncLangchainCallbackHandler",
+    "LlamaIndexCallbackHandler",
+    "client_factory",
+    "run_sync",
+    "make_async",
+    "cache",
+]

{chainlit-0.4.2 → chainlit-0.4.3}/chainlit/cli/__init__.py

@@ -2,6 +2,10 @@ import click
 import os
 import sys
 import uvicorn
+import asyncio
+import nest_asyncio
+
+nest_asyncio.apply()
 
 from chainlit.config import (
     config,
@@ -52,7 +56,15 @@ def run_chainlit(target: str):
 
     log_level = "debug" if config.run.debug else "error"
 
-    uvicorn.run(app, host=host, port=port, log_level=log_level)
+    # Start the server
+    async def start():
+        config = uvicorn.Config(app, host=host, port=port, log_level=log_level)
+        server = uvicorn.Server(config)
+        await server.serve()
+
+    # Run the asyncio event loop instead of uvloop to enable re entrance
+    asyncio.run(start())
+    # uvicorn.run(app, host=host, port=port, log_level=log_level)
 
 
 # Define the "run" command for Chainlit CLI