django_ai_assistant.helpers.use_cases

get_assistant_cls(assistant_id, user, request=None)

Get assistant class by id.

Uses AI_ASSISTANT_CAN_RUN_ASSISTANT_FN permission to check if user can run the assistant.

Parameters:

Name	Type	Description	Default
assistant_id	str	Assistant id to get	required
user	Any	Current user	required
request	HttpRequest | None	Current request, if any	None

Returns: type[AIAssistant]: Assistant class with the given id Raises: AIAssistantNotDefinedError: If assistant with the given id is not found AIUserNotAllowedError: If user is not allowed to use the assistant

Source code in django_ai_assistant/helpers/use_cases.py

def get_assistant_cls(
    assistant_id: str,
    user: Any,
    request: HttpRequest | None = None,
) -> type[AIAssistant]:
    """Get assistant class by id.\n
    Uses `AI_ASSISTANT_CAN_RUN_ASSISTANT_FN` permission to check if user can run the assistant.

    Args:
        assistant_id (str): Assistant id to get
        user (Any): Current user
        request (HttpRequest | None): Current request, if any
    Returns:
        type[AIAssistant]: Assistant class with the given id
    Raises:
        AIAssistantNotDefinedError: If assistant with the given id is not found
        AIUserNotAllowedError: If user is not allowed to use the assistant
    """
    if assistant_id not in AIAssistant.get_cls_registry():
        raise AIAssistantNotDefinedError(f"Assistant with id={assistant_id} not found")
    assistant_cls = AIAssistant.get_cls(assistant_id)
    if not can_run_assistant(
        assistant_cls=assistant_cls,
        user=user,
        request=request,
    ):
        raise AIUserNotAllowedError("User is not allowed to use this assistant")
    return assistant_cls

get_single_assistant_info(assistant_id, user, request=None)

Get assistant info id. Returns a dictionary with the assistant id and name.

Uses AI_ASSISTANT_CAN_RUN_ASSISTANT_FN permission to check if user can see the assistant.

Parameters:

Name	Type	Description	Default
assistant_id	str	Assistant id to get	required
user	Any	Current user	required
request	HttpRequest | None	Current request, if any	None

Returns: dict[str, str]: dict like {"id": "personal_ai", "name": "Personal AI"} Raises: AIAssistantNotDefinedError: If assistant with the given id is not found AIUserNotAllowedError: If user is not allowed to see the assistant

Source code in django_ai_assistant/helpers/use_cases.py

def get_single_assistant_info(
    assistant_id: str,
    user: Any,
    request: HttpRequest | None = None,
) -> dict[str, str]:
    """Get assistant info id. Returns a dictionary with the assistant id and name.\n
    Uses `AI_ASSISTANT_CAN_RUN_ASSISTANT_FN` permission to check if user can see the assistant.

    Args:
        assistant_id (str): Assistant id to get
        user (Any): Current user
        request (HttpRequest | None): Current request, if any
    Returns:
        dict[str, str]: dict like `{"id": "personal_ai", "name": "Personal AI"}`
    Raises:
        AIAssistantNotDefinedError: If assistant with the given id is not found
        AIUserNotAllowedError: If user is not allowed to see the assistant
    """
    assistant_cls = get_assistant_cls(assistant_id, user, request)

    return {
        "id": assistant_id,
        "name": assistant_cls.name,
    }

get_assistants_info(user, request=None)

Get all assistants info. Returns a list of dictionaries with the assistant id and name.

Uses AI_ASSISTANT_CAN_RUN_ASSISTANT_FN permission to check the assistants the user can see, and returns only the ones the user can see.

Parameters:

Name	Type	Description	Default
user	Any	Current user	required
request	HttpRequest | None	Current request, if any	None

Returns: list[dict[str, str]]: List of dicts like [{"id": "personal_ai", "name": "Personal AI"}, ...]

Source code in django_ai_assistant/helpers/use_cases.py

def get_assistants_info(
    user: Any,
    request: HttpRequest | None = None,
) -> list[dict[str, str]]:
    """Get all assistants info. Returns a list of dictionaries with the assistant id and name.\n
    Uses `AI_ASSISTANT_CAN_RUN_ASSISTANT_FN` permission to check the assistants the user can see,
    and returns only the ones the user can see.

    Args:
        user (Any): Current user
        request (HttpRequest | None): Current request, if any
    Returns:
        list[dict[str, str]]: List of dicts like `[{"id": "personal_ai", "name": "Personal AI"}, ...]`
    """
    assistant_info_list = []
    for assistant_id in AIAssistant.get_cls_registry().keys():
        try:
            info = get_single_assistant_info(assistant_id, user, request)
            assistant_info_list.append(info)
        except AIUserNotAllowedError:
            continue
    return assistant_info_list

create_message(assistant_id, thread, user, content, request=None)

Create a message in a thread, and right after runs the assistant to get the AI response.

Uses AI_ASSISTANT_CAN_RUN_ASSISTANT_FN permission to check if user can run the assistant.

Uses AI_ASSISTANT_CAN_CREATE_MESSAGE_FN permission to check if user can create a message in the thread.

Parameters:

Name	Type	Description	Default
assistant_id	str	Assistant id to use to get the AI response	required
thread	Thread	Thread where to create the message	required
user	Any	Current user	required
content	Any	Message content, usually a string	required
request	HttpRequest | None	Current request, if any	None

Returns: dict: The output of the assistant chain, structured like {"output": "assistant response", "history": ...} Raises: AIUserNotAllowedError: If user is not allowed to create messages in the thread

Source code in django_ai_assistant/helpers/use_cases.py

def create_message(
    assistant_id: str,
    thread: Thread,
    user: Any,
    content: Any,
    request: HttpRequest | None = None,
) -> dict:
    """Create a message in a thread, and right after runs the assistant to get the AI response.\n
    Uses `AI_ASSISTANT_CAN_RUN_ASSISTANT_FN` permission to check if user can run the assistant.\n
    Uses `AI_ASSISTANT_CAN_CREATE_MESSAGE_FN` permission to check if user can create a message in the thread.

    Args:
        assistant_id (str): Assistant id to use to get the AI response
        thread (Thread): Thread where to create the message
        user (Any): Current user
        content (Any): Message content, usually a string
        request (HttpRequest | None): Current request, if any
    Returns:
        dict: The output of the assistant chain,
            structured like `{"output": "assistant response", "history": ...}`
    Raises:
        AIUserNotAllowedError: If user is not allowed to create messages in the thread
    """
    assistant_cls = get_assistant_cls(assistant_id, user, request)

    if not can_create_message(thread=thread, user=user, request=request):
        raise AIUserNotAllowedError("User is not allowed to create messages in this thread")

    # TODO: Check if we can separate the message creation from the chain invoke
    assistant = assistant_cls(user=user, request=request)
    assistant_message = assistant.invoke(
        {"input": content},
        thread_id=thread.id,
    )
    return assistant_message

create_thread(name, user, request=None)

Create a thread.

Uses AI_ASSISTANT_CAN_CREATE_THREAD_FN permission to check if user can create a thread.

Parameters:

Name	Type	Description	Default
name	str	Thread name	required
user	Any	Current user	required
request	HttpRequest | None	Current request, if any	None

Returns: Thread: Created thread model instance Raises: AIUserNotAllowedError: If user is not allowed to create threads

Source code in django_ai_assistant/helpers/use_cases.py

def create_thread(
    name: str,
    user: Any,
    request: HttpRequest | None = None,
) -> Thread:
    """Create a thread.\n
    Uses `AI_ASSISTANT_CAN_CREATE_THREAD_FN` permission to check if user can create a thread.

    Args:
        name (str): Thread name
        user (Any): Current user
        request (HttpRequest | None): Current request, if any
    Returns:
        Thread: Created thread model instance
    Raises:
        AIUserNotAllowedError: If user is not allowed to create threads
    """
    if not can_create_thread(user=user, request=request):
        raise AIUserNotAllowedError("User is not allowed to create threads")

    thread = Thread.objects.create(name=name, created_by=user)
    return thread

get_single_thread(thread_id, user, request=None)

Get a single thread by id.

Uses AI_ASSISTANT_CAN_VIEW_THREAD_FN permission to check if user can view the thread.

Parameters:

Name	Type	Description	Default
thread_id	str	Thread id to get	required
user	Any	Current user	required
request	HttpRequest | None	Current request, if any	None

Returns: Thread: Thread model instance Raises: AIUserNotAllowedError: If user is not allowed to view the thread

Source code in django_ai_assistant/helpers/use_cases.py

def get_single_thread(
    thread_id: Any,
    user: Any,
    request: HttpRequest | None = None,
) -> Thread:
    """Get a single thread by id.\n
    Uses `AI_ASSISTANT_CAN_VIEW_THREAD_FN` permission to check if user can view the thread.

    Args:
        thread_id (str): Thread id to get
        user (Any): Current user
        request (HttpRequest | None): Current request, if any
    Returns:
        Thread: Thread model instance
    Raises:
        AIUserNotAllowedError: If user is not allowed to view the thread
    """
    thread = Thread.objects.get(id=thread_id)

    if not can_view_thread(thread=thread, user=user, request=request):
        raise AIUserNotAllowedError("User is not allowed to view this thread")

    return thread

get_threads(user)

Get all user owned threads.

Parameters:

Name	Type	Description	Default
user	Any	Current user	required

Returns: list[Thread]: List of thread model instances

Source code in django_ai_assistant/helpers/use_cases.py

def get_threads(user: Any) -> list[Thread]:
    """Get all user owned threads.\n

    Args:
        user (Any): Current user
    Returns:
        list[Thread]: List of thread model instances
    """
    return list(Thread.objects.filter(created_by=user))

update_thread(thread, name, user, request=None)

Update thread name.

Uses AI_ASSISTANT_CAN_UPDATE_THREAD_FN permission to check if user can update the thread.

Parameters:

Name	Type	Description	Default
thread	Thread	Thread model instance to update	required
name	str	New thread name	required
user	Any	Current user	required
request	HttpRequest | None	Current request, if any	None

Returns: Thread: Updated thread model instance Raises: AIUserNotAllowedError: If user is not allowed to update the thread

Source code in django_ai_assistant/helpers/use_cases.py

def update_thread(
    thread: Thread,
    name: str,
    user: Any,
    request: HttpRequest | None = None,
) -> Thread:
    """Update thread name.\n
    Uses `AI_ASSISTANT_CAN_UPDATE_THREAD_FN` permission to check if user can update the thread.

    Args:
        thread (Thread): Thread model instance to update
        name (str): New thread name
        user (Any): Current user
        request (HttpRequest | None): Current request, if any
    Returns:
        Thread: Updated thread model instance
    Raises:
        AIUserNotAllowedError: If user is not allowed to update the thread
    """
    if not can_update_thread(thread=thread, user=user, request=request):
        raise AIUserNotAllowedError("User is not allowed to update this thread")

    thread.name = name
    thread.save()
    return thread

delete_thread(thread, user, request=None)

Delete a thread.

Uses AI_ASSISTANT_CAN_DELETE_THREAD_FN permission to check if user can delete the thread.

Parameters:

Name	Type	Description	Default
thread	Thread	Thread model instance to delete	required
user	Any	Current user	required
request	HttpRequest | None	Current request, if any	None

Raises: AIUserNotAllowedError: If user is not allowed to delete the thread

Source code in django_ai_assistant/helpers/use_cases.py

def delete_thread(
    thread: Thread,
    user: Any,
    request: HttpRequest | None = None,
) -> None:
    """Delete a thread.\n
    Uses `AI_ASSISTANT_CAN_DELETE_THREAD_FN` permission to check if user can delete the thread.

    Args:
        thread (Thread): Thread model instance to delete
        user (Any): Current user
        request (HttpRequest | None): Current request, if any
    Raises:
        AIUserNotAllowedError: If user is not allowed to delete the thread
    """
    if not can_delete_thread(thread=thread, user=user, request=request):
        raise AIUserNotAllowedError("User is not allowed to delete this thread")

    thread.delete()

get_thread_messages(thread, user, request=None)

Get all messages in a thread.

Uses AI_ASSISTANT_CAN_VIEW_THREAD_FN permission to check if user can view the thread.

Parameters:

Name	Type	Description	Default
thread	Thread	Thread model instance to get messages from	required
user	Any	Current user	required
request	HttpRequest | None	Current request, if any	None

Returns: list[BaseMessage]: List of message instances

Source code in django_ai_assistant/helpers/use_cases.py

def get_thread_messages(
    thread: Thread,
    user: Any,
    request: HttpRequest | None = None,
) -> list[BaseMessage]:
    """Get all messages in a thread.\n
    Uses `AI_ASSISTANT_CAN_VIEW_THREAD_FN` permission to check if user can view the thread.

    Args:
        thread (Thread): Thread model instance to get messages from
        user (Any): Current user
        request (HttpRequest | None): Current request, if any
    Returns:
        list[BaseMessage]: List of message instances
    """
    # TODO: have more permissions for threads? View thread permission?
    if user != thread.created_by:
        raise AIUserNotAllowedError("User is not allowed to view messages in this thread")

    return DjangoChatMessageHistory(thread.id).get_messages()

delete_message(message, user, request=None)

Delete a message.

Uses AI_ASSISTANT_CAN_DELETE_MESSAGE_FN permission to check if user can delete the message.

Parameters:

Name	Type	Description	Default
message	Message	Message model instance to delete	required
user	Any	Current user	required
request	HttpRequest | None	Current request, if any	None

Raises: AIUserNotAllowedError: If user is not allowed to delete the message

Source code in django_ai_assistant/helpers/use_cases.py

def delete_message(
    message: Message,
    user: Any,
    request: HttpRequest | None = None,
):
    """Delete a message.\n
    Uses `AI_ASSISTANT_CAN_DELETE_MESSAGE_FN` permission to check if user can delete the message.

    Args:
        message (Message): Message model instance to delete
        user (Any): Current user
        request (HttpRequest | None): Current request, if any
    Raises:
        AIUserNotAllowedError: If user is not allowed to delete the message
    """
    if not can_delete_message(message=message, user=user, request=request):
        raise AIUserNotAllowedError("User is not allowed to delete this message")

    return DjangoChatMessageHistory(thread_id=message.thread_id).remove_messages(
        message_ids=[str(message.id)]
    )