aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/sdk/asnyc_methods/conversations.py
diff options
context:
space:
mode:
Diffstat (limited to '.venv/lib/python3.12/site-packages/sdk/asnyc_methods/conversations.py')
-rw-r--r--.venv/lib/python3.12/site-packages/sdk/asnyc_methods/conversations.py310
1 files changed, 310 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/sdk/asnyc_methods/conversations.py b/.venv/lib/python3.12/site-packages/sdk/asnyc_methods/conversations.py
new file mode 100644
index 00000000..885f9fc5
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/sdk/asnyc_methods/conversations.py
@@ -0,0 +1,310 @@
+from builtins import list as _list
+from pathlib import Path
+from typing import Any, Optional
+from uuid import UUID
+
+import aiofiles
+
+from shared.api.models import (
+ WrappedBooleanResponse,
+ WrappedConversationMessagesResponse,
+ WrappedConversationResponse,
+ WrappedConversationsResponse,
+ WrappedMessageResponse,
+)
+
+
+class ConversationsSDK:
+ def __init__(self, client):
+ self.client = client
+
+ async def create(
+ self,
+ name: Optional[str] = None,
+ ) -> WrappedConversationResponse:
+ """Create a new conversation.
+
+ Returns:
+ WrappedConversationResponse
+ """
+ data: dict[str, Any] = {}
+ if name:
+ data["name"] = name
+
+ response_dict = await self.client._make_request(
+ "POST",
+ "conversations",
+ data=data,
+ version="v3",
+ )
+
+ return WrappedConversationResponse(**response_dict)
+
+ async def list(
+ self,
+ ids: Optional[list[str | UUID]] = None,
+ offset: Optional[int] = 0,
+ limit: Optional[int] = 100,
+ ) -> WrappedConversationsResponse:
+ """List conversations with pagination and sorting options.
+
+ Args:
+ ids (Optional[list[str | UUID]]): List of conversation IDs to retrieve
+ offset (int, optional): Specifies the number of objects to skip. Defaults to 0.
+ limit (int, optional): Specifies a limit on the number of objects to return, ranging between 1 and 100. Defaults to 100.
+
+ Returns:
+ WrappedConversationsResponse
+ """
+ params: dict = {
+ "offset": offset,
+ "limit": limit,
+ }
+ if ids:
+ params["ids"] = ids
+
+ response_dict = await self.client._make_request(
+ "GET",
+ "conversations",
+ params=params,
+ version="v3",
+ )
+
+ return WrappedConversationsResponse(**response_dict)
+
+ async def retrieve(
+ self,
+ id: str | UUID,
+ ) -> WrappedConversationMessagesResponse:
+ """Get detailed information about a specific conversation.
+
+ Args:
+ id (str | UUID): The ID of the conversation to retrieve
+
+ Returns:
+ WrappedConversationMessagesResponse
+ """
+ response_dict = await self.client._make_request(
+ "GET",
+ f"conversations/{str(id)}",
+ version="v3",
+ )
+
+ return WrappedConversationMessagesResponse(**response_dict)
+
+ async def update(
+ self,
+ id: str | UUID,
+ name: str,
+ ) -> WrappedConversationResponse:
+ """Update an existing conversation.
+
+ Args:
+ id (str | UUID): The ID of the conversation to update
+ name (str): The new name of the conversation
+
+ Returns:
+ WrappedConversationResponse
+ """
+ data: dict[str, Any] = {
+ "name": name,
+ }
+
+ response_dict = await self.client._make_request(
+ "POST",
+ f"conversations/{str(id)}",
+ json=data,
+ version="v3",
+ )
+
+ return WrappedConversationResponse(**response_dict)
+
+ async def delete(
+ self,
+ id: str | UUID,
+ ) -> WrappedBooleanResponse:
+ """Delete a conversation.
+
+ Args:
+ id (str | UUID): The ID of the conversation to delete
+
+ Returns:
+ WrappedBooleanResponse
+ """
+ response_dict = await self.client._make_request(
+ "DELETE",
+ f"conversations/{str(id)}",
+ version="v3",
+ )
+
+ return WrappedBooleanResponse(**response_dict)
+
+ async def add_message(
+ self,
+ id: str | UUID,
+ content: str,
+ role: str,
+ metadata: Optional[dict] = None,
+ parent_id: Optional[str] = None,
+ ) -> WrappedMessageResponse:
+ """Add a new message to a conversation.
+
+ Args:
+ id (str | UUID): The ID of the conversation to add the message to
+ content (str): The content of the message
+ role (str): The role of the message (e.g., "user" or "assistant")
+ parent_id (Optional[str]): The ID of the parent message
+ metadata (Optional[dict]): Additional metadata to attach to the message
+
+ Returns:
+ WrappedMessageResponse
+ """
+ data: dict[str, Any] = {
+ "content": content,
+ "role": role,
+ }
+ if parent_id:
+ data["parent_id"] = parent_id
+ if metadata:
+ data["metadata"] = metadata
+
+ response_dict = await self.client._make_request(
+ "POST",
+ f"conversations/{str(id)}/messages",
+ json=data,
+ version="v3",
+ )
+
+ return WrappedMessageResponse(**response_dict)
+
+ async def update_message(
+ self,
+ id: str | UUID,
+ message_id: str,
+ content: Optional[str] = None,
+ metadata: Optional[dict] = None,
+ ) -> WrappedMessageResponse:
+ """Update an existing message in a conversation.
+
+ Args:
+ id (str | UUID): The ID of the conversation containing the message
+ message_id (str): The ID of the message to update
+ content (str): The new content of the message
+ metadata (dict): Additional metadata to attach to the message
+
+ Returns:
+ WrappedMessageResponse
+ """
+ data: dict[str, Any] = {"content": content}
+ if metadata:
+ data["metadata"] = metadata
+ response_dict = await self.client._make_request(
+ "POST",
+ f"conversations/{str(id)}/messages/{message_id}",
+ json=data,
+ version="v3",
+ )
+
+ return WrappedMessageResponse(**response_dict)
+
+ async def export(
+ self,
+ output_path: str | Path,
+ columns: Optional[_list[str]] = None,
+ filters: Optional[dict] = None,
+ include_header: bool = True,
+ ) -> None:
+ """Export conversations to a CSV file, streaming the results directly
+ to disk.
+
+ Args:
+ output_path (str | Path): Local path where the CSV file should be saved
+ columns (Optional[list[str]]): Specific columns to export. If None, exports default columns
+ filters (Optional[dict]): Optional filters to apply when selecting conversations
+ include_header (bool): Whether to include column headers in the CSV (default: True)
+
+ Returns:
+ None
+ """
+ # Convert path to string if it's a Path object
+ output_path = (
+ str(output_path) if isinstance(output_path, Path) else output_path
+ )
+
+ # Prepare request data
+ data: dict[str, Any] = {"include_header": include_header}
+ if columns:
+ data["columns"] = columns
+ if filters:
+ data["filters"] = filters
+
+ # Stream response directly to file
+ async with aiofiles.open(output_path, "wb") as f:
+ async with self.client.session.post(
+ f"{self.client.base_url}/v3/conversations/export",
+ json=data,
+ headers={
+ "Accept": "text/csv",
+ **self.client._get_auth_header(),
+ },
+ ) as response:
+ if response.status != 200:
+ raise ValueError(
+ f"Export failed with status {response.status}",
+ response,
+ )
+
+ async for chunk in response.content.iter_chunks():
+ if chunk:
+ await f.write(chunk[0])
+
+ async def export_messages(
+ self,
+ output_path: str | Path,
+ columns: Optional[_list[str]] = None,
+ filters: Optional[dict] = None,
+ include_header: bool = True,
+ ) -> None:
+ """Export messages to a CSV file, streaming the results directly to
+ disk.
+
+ Args:
+ output_path (str | Path): Local path where the CSV file should be saved
+ columns (Optional[list[str]]): Specific columns to export. If None, exports default columns
+ filters (Optional[dict]): Optional filters to apply when selecting messages
+ include_header (bool): Whether to include column headers in the CSV (default: True)
+
+ Returns:
+ None
+ """
+ # Convert path to string if it's a Path object
+ output_path = (
+ str(output_path) if isinstance(output_path, Path) else output_path
+ )
+
+ # Prepare request data
+ data: dict[str, Any] = {"include_header": include_header}
+ if columns:
+ data["columns"] = columns
+ if filters:
+ data["filters"] = filters
+
+ # Stream response directly to file
+ async with aiofiles.open(output_path, "wb") as f:
+ async with self.client.session.post(
+ f"{self.client.base_url}/v3/conversations/export_messages",
+ json=data,
+ headers={
+ "Accept": "text/csv",
+ **self.client._get_auth_header(),
+ },
+ ) as response:
+ if response.status != 200:
+ raise ValueError(
+ f"Export failed with status {response.status}",
+ response,
+ )
+
+ async for chunk in response.content.iter_chunks():
+ if chunk:
+ await f.write(chunk[0])
generated by cgit v1.2.3 (git 2.47.1) at 2025-06-08 22:59:07 +0000