primethink-cli 1.0.0__py3-none-any.whl

primethink.py

@@ -0,0 +1,1002 @@
+#!/usr/bin/env python3
+
+import os
+import sys
+import json
+import contextlib
+import click
+import requests
+from pathlib import Path
+from typing import Optional, Dict, Any, List
+
+# Define default config file location
+CONFIG_DIR = Path.home() / ".primethink"
+CONFIG_FILE = CONFIG_DIR / "config.json"
+
+# Default API URL
+DEFAULT_API_URL = "https://api.primethink.ai"
+
+# Default timeout for HTTP requests (seconds)
+REQUEST_TIMEOUT = 30
+
+
+def _sanitize_filename(name: str) -> str:
+    """Strip path components and unsafe characters from a server-provided filename."""
+    basename = Path(name).name
+    basename = basename.lstrip('.')
+    if not basename:
+        return "unnamed_file"
+    return basename
+
+@click.group()
+def cli():
+    """PrimeThink CLI - A tool for interacting with PrimeThink API."""
+    pass
+
+@cli.command()
+def version():
+    """Display the version of PrimeThink CLI."""
+    click.echo("PrimeThink CLI v1.0.0")
+
+# ============================================================================
+# Token Management Commands
+# ============================================================================
+
+@cli.command()
+@click.option('--token', '-t', required=True, help='API token')
+@click.option('--profile', '-p', default='default', help='Profile name (default: default)')
+@click.option('--api-url', '-u', default=None, help='Custom API URL (default: https://api.primethink.ai)')
+def configure(token, profile, api_url):
+    """Configure API token for a profile."""
+    CONFIG_DIR.mkdir(exist_ok=True)
+
+    # Load existing config
+    config = load_config()
+
+    # Initialize profiles if not exists
+    if 'profiles' not in config:
+        config['profiles'] = {}
+
+    # Save token and API URL for the profile
+    profile_config = {'token': token}
+    if api_url:
+        profile_config['api_url'] = api_url
+
+    config['profiles'][profile] = profile_config
+
+    # Set as active profile if it's the first one or default
+    if 'active_profile' not in config or profile == 'default':
+        config['active_profile'] = profile
+
+    # Save config
+    save_config(config)
+
+    api_display = api_url if api_url else DEFAULT_API_URL
+    click.echo(f"✓ Token configured for profile '{profile}' (API: {api_display})")
+    click.echo(f"✓ Profile '{profile}' set as active")
+
+@cli.command()
+@click.argument('profile')
+def use(profile):
+    """Switch to a different token profile."""
+    config = load_config()
+
+    if 'profiles' not in config or profile not in config['profiles']:
+        click.echo(f"Error: Profile '{profile}' not found. Use 'pt configure' to create it.")
+        sys.exit(1)
+
+    config['active_profile'] = profile
+    save_config(config)
+
+    api_url = config['profiles'][profile].get('api_url', DEFAULT_API_URL)
+    click.echo(f"✓ Switched to profile '{profile}' (API: {api_url})")
+
+@cli.command()
+def list_profiles():
+    """List all configured profiles."""
+    config = load_config()
+
+    if 'profiles' not in config or not config['profiles']:
+        click.echo("No profiles configured. Use 'pt configure' to add one.")
+        return
+
+    active_profile = config.get('active_profile', '')
+
+    click.echo("Configured profiles:")
+    for name, data in config['profiles'].items():
+        active = "* " if name == active_profile else "  "
+        api_url = data.get('api_url', DEFAULT_API_URL)
+        click.echo(f"{active}{name} ({api_url})")
+
+@cli.command()
+@click.argument('profile')
+def remove_profile(profile):
+    """Remove a token profile."""
+    config = load_config()
+
+    if 'profiles' not in config or profile not in config['profiles']:
+        click.echo(f"Error: Profile '{profile}' not found.")
+        sys.exit(1)
+
+    del config['profiles'][profile]
+
+    # If the removed profile was active, switch to another one
+    if config.get('active_profile') == profile:
+        if config['profiles']:
+            config['active_profile'] = list(config['profiles'].keys())[0]
+            click.echo(f"✓ Switched to profile '{config['active_profile']}'")
+        else:
+            del config['active_profile']
+
+    save_config(config)
+    click.echo(f"✓ Profile '{profile}' removed")
+
+# ============================================================================
+# API Commands
+# ============================================================================
+
+@cli.command()
+@click.option('--profile', '-p', default=None, help='Profile to use for this request')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def available_actions(profile, api_url):
+    """Get list of available task actions."""
+    config = get_active_config(profile)
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    try:
+        response = requests.get(
+            f"{api_url}/api/v1/tasks/available_task_actions",
+            headers={
+                "accept": "application/json",
+                "Authorization": f"Token {token}"
+            },
+            timeout=REQUEST_TIMEOUT
+        )
+
+        if response.status_code == 200:
+            actions = response.json()
+            click.echo(json.dumps(actions, indent=2))
+        else:
+            click.echo(f"Error: {response.status_code} - {response.text}")
+            sys.exit(1)
+    except requests.exceptions.RequestException as e:
+        click.echo(f"Error connecting to API: {e}")
+        sys.exit(1)
+
+@cli.command()
+@click.option('--action', '-a', required=True, help='Task action name')
+@click.option('--message', '-m', required=True, help='Message input')
+@click.option('--files', '-f', multiple=True, type=click.Path(exists=True), help='Files to attach')
+@click.option('--return-original', is_flag=True, help='Return original message')
+@click.option('--profile', '-p', default=None, help='Profile to use for this request')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def execute_action(action, message, files, return_original, profile, api_url):
+    """Execute a task action."""
+    config = get_active_config(profile)
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    # Prepare multipart form data
+    form_data = {
+        'task_action_name': action,
+        'message_input': message,
+        'return_original_message': str(return_original).lower()
+    }
+
+    with contextlib.ExitStack() as stack:
+        files_data = []
+        for file_path in files:
+            f = stack.enter_context(open(file_path, 'rb'))
+            files_data.append(('files', f))
+
+        try:
+            response = requests.post(
+                f"{api_url}/api/v1/tasks/execute_task_action",
+                headers={
+                    "accept": "application/json",
+                    "Authorization": f"Token {token}"
+                },
+                data=form_data,
+                files=files_data if files_data else None,
+                timeout=REQUEST_TIMEOUT
+            )
+
+            if response.status_code == 200:
+                result = response.json()
+                click.echo(json.dumps(result, indent=2))
+            else:
+                click.echo(f"Error: {response.status_code} - {response.text}")
+                sys.exit(1)
+        except requests.exceptions.RequestException as e:
+            click.echo(f"Error connecting to API: {e}")
+            sys.exit(1)
+
+@cli.command()
+@click.argument('chat_id_or_mention', required=False)
+@click.option('--message', '-m', required=True, help='Message input')
+@click.option('--files', '-f', multiple=True, type=click.Path(exists=True), help='Files to attach')
+@click.option('--async', 'async_request', is_flag=True, default=False, help='Asynchronous request')
+@click.option('--agent', '-a', type=int, help='Agent ID to send message to')
+@click.option('--profile', '-p', default=None, help='Profile to use for this request')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def send_message(chat_id_or_mention, message, files, async_request, agent, profile, api_url):
+    """Send a message to a chat by ID/mention name, or to an agent by ID."""
+    # Validate that either chat_id_or_mention or agent is provided, but not both
+    if not chat_id_or_mention and not agent:
+        click.echo("Error: Either CHAT_ID_OR_MENTION or --agent must be provided")
+        sys.exit(1)
+
+    if chat_id_or_mention and agent:
+        click.echo("Error: Cannot specify both CHAT_ID_OR_MENTION and --agent")
+        sys.exit(1)
+
+    config = get_active_config(profile)
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    # Prepare multipart form data
+    form_data = {
+        'message_input': message
+    }
+
+    # Add is_sync only for chat messages
+    if chat_id_or_mention:
+        is_sync = str(not async_request).lower()
+        form_data['is_sync'] = is_sync
+
+    # Determine the endpoint based on whether we're sending to a chat or agent
+    if agent:
+        endpoint = f"{api_url}/api/v1/virtual-assistants/{agent}/messages"
+    else:
+        endpoint = f"{api_url}/api/v1/chats/{chat_id_or_mention}/messages"
+
+    with contextlib.ExitStack() as stack:
+        files_data = []
+        for file_path in files:
+            f = stack.enter_context(open(file_path, 'rb'))
+            files_data.append(('files', f))
+
+        try:
+            response = requests.post(
+                endpoint,
+                headers={
+                    "accept": "application/json",
+                    "Authorization": f"Token {token}"
+                },
+                data=form_data,
+                files=files_data if files_data else None,
+                timeout=REQUEST_TIMEOUT
+            )
+
+            if response.status_code == 200:
+                result = response.json()
+                click.echo(json.dumps(result, indent=2))
+            else:
+                click.echo(f"Error: {response.status_code} - {response.text}")
+                sys.exit(1)
+        except requests.exceptions.RequestException as e:
+            click.echo(f"Error connecting to API: {e}")
+            sys.exit(1)
+
+# ============================================================================
+# Chat Subcommands
+# ============================================================================
+
+@cli.group()
+def chat():
+    """Chat file management commands."""
+    pass
+
+@chat.command()
+@click.argument('chat_id')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+@click.option('--path', '-p', default=None, help='Directory path within the chat to list (e.g., /subfolder)')
+def list_files(chat_id, api_url, path):
+    """List files and directories in a chat."""
+    config = get_active_config()
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    params = {}
+    if path:
+        params['path'] = path
+
+    try:
+        response = requests.get(
+            f"{api_url}/api/v1/chats/{chat_id}/directories",
+            headers={
+                "accept": "application/json",
+                "Authorization": f"Token {token}"
+            },
+            params=params,
+            timeout=REQUEST_TIMEOUT
+        )
+
+        if response.status_code == 200:
+            result = response.json()
+            click.echo(json.dumps(result, indent=2))
+        else:
+            click.echo(f"Error: {response.status_code} - {response.text}")
+            sys.exit(1)
+    except requests.exceptions.RequestException as e:
+        click.echo(f"Error connecting to API: {e}")
+        sys.exit(1)
+
+@chat.command()
+@click.argument('chat_id')
+@click.argument('files', nargs=-1, required=True, type=click.Path(exists=True))
+@click.option('--path', '-p', default=None, help='Directory path within the chat to upload to (e.g., /subfolder)')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def upload_files(chat_id, files, path, api_url):
+    """Upload local files to a chat."""
+    config = get_active_config()
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    params = {}
+    if path:
+        params['path'] = path
+
+    with contextlib.ExitStack() as stack:
+        files_data = []
+        for file_path in files:
+            f = stack.enter_context(open(file_path, 'rb'))
+            files_data.append(('documents', f))
+
+        try:
+            response = requests.post(
+                f"{api_url}/api/v1/chats/{chat_id}/documents",
+                headers={
+                    "accept": "application/json",
+                    "Authorization": f"Token {token}"
+                },
+                files=files_data,
+                params=params,
+                timeout=REQUEST_TIMEOUT
+            )
+
+            if response.status_code == 200:
+                result = response.json()
+                click.echo(json.dumps(result, indent=2))
+                click.echo(f"Uploaded {len(files)} file(s) to chat {chat_id}")
+            else:
+                click.echo(f"Error: {response.status_code} - {response.text}")
+                sys.exit(1)
+        except requests.exceptions.RequestException as e:
+            click.echo(f"Error connecting to API: {e}")
+            sys.exit(1)
+
+@chat.command()
+@click.argument('chat_id')
+@click.argument('document_id')
+@click.option('--output', '-o', default=None, type=click.Path(), help='Output file path (default: use original filename)')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def download_file(chat_id, document_id, output, api_url):
+    """Download a file from a chat to the local filesystem."""
+    config = get_active_config()
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    try:
+        meta_response = requests.get(
+            f"{api_url}/api/v1/chats/{chat_id}/document/{document_id}",
+            headers={
+                "accept": "application/json",
+                "Authorization": f"Token {token}"
+            },
+            timeout=REQUEST_TIMEOUT
+        )
+
+        if meta_response.status_code != 200:
+            click.echo(f"Error fetching document metadata: {meta_response.status_code} - {meta_response.text}")
+            sys.exit(1)
+
+        doc_meta = meta_response.json()
+        filename = output or _sanitize_filename(
+            doc_meta.get('filename') or doc_meta.get('name') or f"document_{document_id}"
+        )
+
+        download_response = requests.get(
+            f"{api_url}/api/v1/documents/{document_id}/download",
+            headers={
+                "Authorization": f"Token {token}"
+            },
+            stream=True,
+            timeout=REQUEST_TIMEOUT
+        )
+
+        if download_response.status_code == 200:
+            output_path = Path(filename)
+            output_path.parent.mkdir(parents=True, exist_ok=True)
+            with open(output_path, 'wb') as f:
+                for chunk in download_response.iter_content(chunk_size=8192):
+                    f.write(chunk)
+            click.echo(f"Downloaded document {document_id} to {output_path}")
+        else:
+            click.echo(f"Error downloading document: {download_response.status_code} - {download_response.text}")
+            sys.exit(1)
+    except requests.exceptions.RequestException as e:
+        click.echo(f"Error connecting to API: {e}")
+        sys.exit(1)
+
+@chat.command()
+@click.argument('chat_id')
+@click.argument('local_dir', type=click.Path(exists=True, file_okay=False))
+@click.option('--path', '-p', default=None, help='Target directory path within the chat (e.g., /subfolder)')
+@click.option('--pattern', default='*', help='Glob pattern to filter files (default: *)')
+@click.option('--recursive', '-r', is_flag=True, help='Include files in subdirectories')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def sync_to(chat_id, local_dir, path, pattern, recursive, api_url):
+    """Sync local directory files to a chat."""
+    config = get_active_config()
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    local_path = Path(local_dir)
+
+    if recursive:
+        file_list = [f for f in local_path.rglob(pattern) if f.is_file()]
+    else:
+        file_list = [f for f in local_path.glob(pattern) if f.is_file()]
+
+    if not file_list:
+        click.echo(f"No files matching pattern '{pattern}' found in {local_dir}")
+        return
+
+    click.echo(f"Found {len(file_list)} file(s) to sync")
+
+    files_by_dir: Dict[str, List[Path]] = {}
+    for file in file_list:
+        rel_path = file.relative_to(local_path)
+        parent_dir = rel_path.parent.as_posix()
+        if parent_dir == '.':
+            parent_dir = ''
+        files_by_dir.setdefault(parent_dir, []).append(file)
+
+    uploaded_count = 0
+    error_count = 0
+
+    for sub_dir, dir_files in files_by_dir.items():
+        if path and sub_dir:
+            target_path = f"{path.rstrip('/')}/{sub_dir}"
+        elif path:
+            target_path = path
+        elif sub_dir:
+            target_path = f"/{sub_dir}"
+        else:
+            target_path = None
+
+        params = {}
+        if target_path:
+            params['path'] = target_path
+
+        with contextlib.ExitStack() as stack:
+            files_data = []
+            for file in dir_files:
+                f = stack.enter_context(open(file, 'rb'))
+                files_data.append(('documents', f))
+
+            try:
+                response = requests.post(
+                    f"{api_url}/api/v1/chats/{chat_id}/documents",
+                    headers={
+                        "accept": "application/json",
+                        "Authorization": f"Token {token}"
+                    },
+                    files=files_data,
+                    params=params,
+                    timeout=REQUEST_TIMEOUT
+                )
+
+                if response.status_code == 200:
+                    uploaded_count += len(dir_files)
+                    dir_display = target_path or '/'
+                    click.echo(f"  Uploaded {len(dir_files)} file(s) to {dir_display}")
+                else:
+                    error_count += len(dir_files)
+                    click.echo(f"  Error uploading to {target_path or '/'}: {response.status_code} - {response.text}")
+            except requests.exceptions.RequestException as e:
+                error_count += len(dir_files)
+                click.echo(f"  Error connecting to API: {e}")
+
+    click.echo(f"Sync complete: {uploaded_count} uploaded, {error_count} failed")
+
+@chat.command()
+@click.argument('chat_id')
+@click.argument('local_dir', type=click.Path(file_okay=False))
+@click.option('--path', '-p', default=None, help='Source directory path within the chat (e.g., /subfolder)')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def sync_from(chat_id, local_dir, path, api_url):
+    """Sync chat files to a local directory."""
+    config = get_active_config()
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    output_dir = Path(local_dir)
+    output_dir.mkdir(parents=True, exist_ok=True)
+
+    downloaded_count = 0
+    error_count = 0
+
+    def download_directory(chat_path, local_base):
+        nonlocal downloaded_count, error_count
+
+        params = {}
+        if chat_path:
+            params['path'] = chat_path
+
+        try:
+            response = requests.get(
+                f"{api_url}/api/v1/chats/{chat_id}/directories",
+                headers={
+                    "accept": "application/json",
+                    "Authorization": f"Token {token}"
+                },
+                params=params,
+                timeout=REQUEST_TIMEOUT
+            )
+
+            if response.status_code != 200:
+                error_count += 1
+                click.echo(f"  Error listing {chat_path or '/'}: {response.status_code} - {response.text}")
+                return
+
+            result = response.json()
+            dirs = result.get('dirs', [])
+            documents = result.get('documents', [])
+
+            for doc in documents:
+                doc_id = doc.get('id')
+                filename = _sanitize_filename(
+                    doc.get('filename') or doc.get('name') or f"document_{doc_id}"
+                )
+                file_output = local_base / filename
+
+                try:
+                    dl_response = requests.get(
+                        f"{api_url}/api/v1/documents/{doc_id}/download",
+                        headers={
+                            "Authorization": f"Token {token}"
+                        },
+                        stream=True,
+                        timeout=REQUEST_TIMEOUT
+                    )
+
+                    if dl_response.status_code == 200:
+                        file_output.parent.mkdir(parents=True, exist_ok=True)
+                        with open(file_output, 'wb') as f:
+                            for chunk in dl_response.iter_content(chunk_size=8192):
+                                f.write(chunk)
+                        downloaded_count += 1
+                        click.echo(f"  Downloaded: {file_output}")
+                    else:
+                        error_count += 1
+                        click.echo(f"  Error downloading {filename}: {dl_response.status_code}")
+                except requests.exceptions.RequestException as e:
+                    error_count += 1
+                    click.echo(f"  Error downloading {filename}: {e}")
+
+            for d in dirs:
+                dir_path = d.get('path', '')
+                dir_name = d.get('name') or dir_path.rstrip('/').split('/')[-1] or dir_path
+                sub_local = local_base / dir_name
+                sub_local.mkdir(parents=True, exist_ok=True)
+                download_directory(dir_path, sub_local)
+
+        except requests.exceptions.RequestException as e:
+            error_count += 1
+            click.echo(f"  Error connecting to API: {e}")
+
+    click.echo(f"Syncing files from chat {chat_id} to {local_dir}...")
+    download_directory(path, output_dir)
+    click.echo(f"Sync complete: {downloaded_count} downloaded, {error_count} failed")
+
+# ============================================================================
+# Collections Subcommands
+# ============================================================================
+
+@cli.group()
+def collections():
+    """Collection file management commands."""
+    pass
+
+@collections.command(name='list')
+@click.option('--page', default=1, type=int, help='Page number (default: 1)')
+@click.option('--page-size', default=20, type=int, help='Results per page (default: 20)')
+@click.option('--search', '-s', default=None, help='Search collections by name')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def collections_list(page, page_size, search, api_url):
+    """List all collections."""
+    config = get_active_config()
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    params = {'page': page, 'page_size': page_size}
+    if search:
+        params['search'] = search
+
+    try:
+        response = requests.get(
+            f"{api_url}/api/v1/collections",
+            headers={
+                "accept": "application/json",
+                "Authorization": f"Token {token}"
+            },
+            params=params,
+            timeout=REQUEST_TIMEOUT
+        )
+
+        if response.status_code == 200:
+            result = response.json()
+            click.echo(json.dumps(result, indent=2))
+        else:
+            click.echo(f"Error: {response.status_code} - {response.text}")
+            sys.exit(1)
+    except requests.exceptions.RequestException as e:
+        click.echo(f"Error connecting to API: {e}")
+        sys.exit(1)
+
+@collections.command()
+@click.argument('collection_id')
+@click.option('--path', '-p', default=None, help='Directory path within the collection (e.g., /subfolder)')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def list_files(collection_id, path, api_url):
+    """List files and directories in a collection."""
+    config = get_active_config()
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    params = {}
+    if path:
+        params['path'] = path
+
+    try:
+        response = requests.get(
+            f"{api_url}/api/v1/collections/{collection_id}/directories",
+            headers={
+                "accept": "application/json",
+                "Authorization": f"Token {token}"
+            },
+            params=params,
+            timeout=REQUEST_TIMEOUT
+        )
+
+        if response.status_code == 200:
+            result = response.json()
+            click.echo(json.dumps(result, indent=2))
+        else:
+            click.echo(f"Error: {response.status_code} - {response.text}")
+            sys.exit(1)
+    except requests.exceptions.RequestException as e:
+        click.echo(f"Error connecting to API: {e}")
+        sys.exit(1)
+
+@collections.command()
+@click.argument('collection_id')
+@click.argument('files', nargs=-1, required=True, type=click.Path(exists=True))
+@click.option('--path', '-p', default=None, help='Directory path within the collection to upload to')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def upload_files(collection_id, files, path, api_url):
+    """Upload local files to a collection."""
+    config = get_active_config()
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    form_data = {}
+    if path:
+        form_data['path'] = path
+
+    with contextlib.ExitStack() as stack:
+        files_data = []
+        for file_path in files:
+            f = stack.enter_context(open(file_path, 'rb'))
+            files_data.append(('documents', f))
+
+        try:
+            response = requests.post(
+                f"{api_url}/api/v1/collections/{collection_id}/documents",
+                headers={
+                    "accept": "application/json",
+                    "Authorization": f"Token {token}"
+                },
+                files=files_data,
+                data=form_data if form_data else None,
+                timeout=REQUEST_TIMEOUT
+            )
+
+            if response.status_code == 200:
+                result = response.json()
+                click.echo(json.dumps(result, indent=2))
+                click.echo(f"Uploaded {len(files)} file(s) to collection {collection_id}")
+            else:
+                click.echo(f"Error: {response.status_code} - {response.text}")
+                sys.exit(1)
+        except requests.exceptions.RequestException as e:
+            click.echo(f"Error connecting to API: {e}")
+            sys.exit(1)
+
+@collections.command()
+@click.argument('collection_id')
+@click.argument('document_id')
+@click.option('--output', '-o', default=None, type=click.Path(), help='Output file path (default: use original filename)')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def download_file(collection_id, document_id, output, api_url):
+    """Download a file from a collection to the local filesystem."""
+    config = get_active_config()
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    try:
+        meta_response = requests.get(
+            f"{api_url}/api/v1/collections/{collection_id}",
+            headers={
+                "accept": "application/json",
+                "Authorization": f"Token {token}"
+            },
+            timeout=REQUEST_TIMEOUT
+        )
+
+        if meta_response.status_code != 200:
+            click.echo(f"Error fetching collection details: {meta_response.status_code} - {meta_response.text}")
+            sys.exit(1)
+
+        collection_data = meta_response.json()
+        documents = collection_data.get('documents', [])
+        doc_meta = next((doc for doc in documents if str(doc.get('id')) == str(document_id)), None)
+
+        filename = output
+        if not filename and doc_meta:
+            filename = _sanitize_filename(
+                doc_meta.get('filename') or doc_meta.get('name') or f"document_{document_id}"
+            )
+        if not filename:
+            filename = f"document_{document_id}"
+
+        download_response = requests.get(
+            f"{api_url}/api/v1/documents/{document_id}/download",
+            headers={
+                "Authorization": f"Token {token}"
+            },
+            stream=True,
+            timeout=REQUEST_TIMEOUT
+        )
+
+        if download_response.status_code == 200:
+            output_path = Path(filename)
+            output_path.parent.mkdir(parents=True, exist_ok=True)
+            with open(output_path, 'wb') as f:
+                for chunk in download_response.iter_content(chunk_size=8192):
+                    f.write(chunk)
+            click.echo(f"Downloaded document {document_id} to {output_path}")
+        else:
+            click.echo(f"Error downloading document: {download_response.status_code} - {download_response.text}")
+            sys.exit(1)
+    except requests.exceptions.RequestException as e:
+        click.echo(f"Error connecting to API: {e}")
+        sys.exit(1)
+
+@collections.command()
+@click.argument('collection_id')
+@click.argument('local_dir', type=click.Path(exists=True, file_okay=False))
+@click.option('--path', '-p', default=None, help='Target directory path within the collection')
+@click.option('--pattern', default='*', help='Glob pattern to filter files (default: *)')
+@click.option('--recursive', '-r', is_flag=True, help='Include files in subdirectories')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def sync_to(collection_id, local_dir, path, pattern, recursive, api_url):
+    """Sync local directory files to a collection."""
+    config = get_active_config()
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    local_path = Path(local_dir)
+
+    if recursive:
+        file_list = [f for f in local_path.rglob(pattern) if f.is_file()]
+    else:
+        file_list = [f for f in local_path.glob(pattern) if f.is_file()]
+
+    if not file_list:
+        click.echo(f"No files matching pattern '{pattern}' found in {local_dir}")
+        return
+
+    click.echo(f"Found {len(file_list)} file(s) to sync")
+
+    files_by_dir: Dict[str, List[Path]] = {}
+    for file in file_list:
+        rel_path = file.relative_to(local_path)
+        parent_dir = rel_path.parent.as_posix()
+        if parent_dir == '.':
+            parent_dir = ''
+        files_by_dir.setdefault(parent_dir, []).append(file)
+
+    uploaded_count = 0
+    error_count = 0
+
+    for sub_dir, dir_files in files_by_dir.items():
+        if path and sub_dir:
+            target_path = f"{path.rstrip('/')}/{sub_dir}"
+        elif path:
+            target_path = path
+        elif sub_dir:
+            target_path = f"/{sub_dir}"
+        else:
+            target_path = None
+
+        form_data = {}
+        if target_path:
+            form_data['path'] = target_path
+
+        with contextlib.ExitStack() as stack:
+            files_data = []
+            for file in dir_files:
+                f = stack.enter_context(open(file, 'rb'))
+                files_data.append(('documents', f))
+
+            try:
+                response = requests.post(
+                    f"{api_url}/api/v1/collections/{collection_id}/documents",
+                    headers={
+                        "accept": "application/json",
+                        "Authorization": f"Token {token}"
+                    },
+                    files=files_data,
+                    data=form_data if form_data else None,
+                    timeout=REQUEST_TIMEOUT
+                )
+
+                if response.status_code == 200:
+                    uploaded_count += len(dir_files)
+                    dir_display = target_path or '/'
+                    click.echo(f"  Uploaded {len(dir_files)} file(s) to {dir_display}")
+                else:
+                    error_count += len(dir_files)
+                    click.echo(f"  Error uploading to {target_path or '/'}: {response.status_code} - {response.text}")
+            except requests.exceptions.RequestException as e:
+                error_count += len(dir_files)
+                click.echo(f"  Error connecting to API: {e}")
+
+    click.echo(f"Sync complete: {uploaded_count} uploaded, {error_count} failed")
+
+@collections.command()
+@click.argument('collection_id')
+@click.argument('local_dir', type=click.Path(file_okay=False))
+@click.option('--path', '-p', default=None, help='Source directory path within the collection')
+@click.option('--api-url', '-u', default=None, help='Override API URL for this request')
+def sync_from(collection_id, local_dir, path, api_url):
+    """Sync collection files to a local directory."""
+    config = get_active_config()
+    api_url = api_url or config['api_url']
+    token = config['token']
+
+    output_dir = Path(local_dir)
+    output_dir.mkdir(parents=True, exist_ok=True)
+
+    downloaded_count = 0
+    error_count = 0
+
+    def download_directory(coll_path, local_base):
+        nonlocal downloaded_count, error_count
+
+        params = {}
+        if coll_path:
+            params['path'] = coll_path
+
+        try:
+            response = requests.get(
+                f"{api_url}/api/v1/collections/{collection_id}/directories",
+                headers={
+                    "accept": "application/json",
+                    "Authorization": f"Token {token}"
+                },
+                params=params,
+                timeout=REQUEST_TIMEOUT
+            )
+
+            if response.status_code != 200:
+                error_count += 1
+                click.echo(f"  Error listing {coll_path or '/'}: {response.status_code} - {response.text}")
+                return
+
+            result = response.json()
+            dirs = result.get('dirs', [])
+            documents = result.get('documents', [])
+
+            for doc in documents:
+                doc_id = doc.get('id')
+                filename = _sanitize_filename(
+                    doc.get('filename') or doc.get('name') or f"document_{doc_id}"
+                )
+                file_output = local_base / filename
+
+                try:
+                    dl_response = requests.get(
+                        f"{api_url}/api/v1/documents/{doc_id}/download",
+                        headers={
+                            "Authorization": f"Token {token}"
+                        },
+                        stream=True,
+                        timeout=REQUEST_TIMEOUT
+                    )
+
+                    if dl_response.status_code == 200:
+                        file_output.parent.mkdir(parents=True, exist_ok=True)
+                        with open(file_output, 'wb') as f:
+                            for chunk in dl_response.iter_content(chunk_size=8192):
+                                f.write(chunk)
+                        downloaded_count += 1
+                        click.echo(f"  Downloaded: {file_output}")
+                    else:
+                        error_count += 1
+                        click.echo(f"  Error downloading {filename}: {dl_response.status_code}")
+                except requests.exceptions.RequestException as e:
+                    error_count += 1
+                    click.echo(f"  Error downloading {filename}: {e}")
+
+            for d in dirs:
+                dir_path = d.get('path', '')
+                dir_name = d.get('name') or dir_path.rstrip('/').split('/')[-1] or dir_path
+                sub_local = local_base / dir_name
+                sub_local.mkdir(parents=True, exist_ok=True)
+                download_directory(dir_path, sub_local)
+
+        except requests.exceptions.RequestException as e:
+            error_count += 1
+            click.echo(f"  Error connecting to API: {e}")
+
+    click.echo(f"Syncing files from collection {collection_id} to {local_dir}...")
+    download_directory(path, output_dir)
+    click.echo(f"Sync complete: {downloaded_count} downloaded, {error_count} failed")
+
+# ============================================================================
+# Helper Functions
+# ============================================================================
+
+def load_config() -> Dict[str, Any]:
+    """Load configuration from file."""
+    if not CONFIG_FILE.exists():
+        return {}
+
+    try:
+        with open(CONFIG_FILE, 'r') as f:
+            return json.load(f)
+    except json.JSONDecodeError:
+        return {}
+
+def save_config(config: Dict[str, Any]):
+    """Save configuration to file."""
+    CONFIG_DIR.mkdir(exist_ok=True)
+    with open(CONFIG_FILE, 'w') as f:
+        json.dump(config, f, indent=2)
+
+def get_active_config(profile: Optional[str] = None) -> Dict[str, str]:
+    """Get active profile configuration or specific profile if provided."""
+    config = load_config()
+
+    # Use specified profile or fall back to active profile
+    if profile:
+        target_profile = profile
+    else:
+        if 'active_profile' not in config:
+            click.echo("Error: No active profile. Use 'pt configure' to set one up.")
+            sys.exit(1)
+        target_profile = config['active_profile']
+
+    if 'profiles' not in config or target_profile not in config['profiles']:
+        available = ', '.join(config.get('profiles', {}).keys())
+        if available:
+            click.echo(f"Error: Profile '{target_profile}' not found.")
+            click.echo(f"Available profiles: {available}")
+            click.echo("Use 'pt list-profiles' to see all profiles.")
+        else:
+            click.echo("Error: No profiles configured. Use 'pt configure' to create one.")
+        sys.exit(1)
+
+    profile_data = config['profiles'][target_profile]
+    return {
+        'token': profile_data['token'],
+        'api_url': profile_data.get('api_url', DEFAULT_API_URL)
+    }
+
+if __name__ == '__main__':
+    cli()

primethink_cli-1.0.0.dist-info/METADATA

@@ -0,0 +1,307 @@
+Metadata-Version: 2.4
+Name: primethink-cli
+Version: 1.0.0
+Summary: PrimeThink CLI - A powerful tool for interacting with PrimeThink AI API
+Home-page: https://github.com/primethink-ai/primethink-cli
+Author: PrimeThink
+Author-email: PrimeThink <support@primethink.ai>
+License-Expression: MIT
+Project-URL: Homepage, https://primethink.ai
+Project-URL: Documentation, https://primethink.ai/cli/install
+Project-URL: Repository, https://github.com/primethink-ai/primethink-cli
+Project-URL: Issues, https://github.com/primethink-ai/primethink-cli/issues
+Keywords: cli,ai,primethink,api,assistant
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: Utilities
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: click>=8.0.0
+Requires-Dist: requests>=2.25.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Dynamic: author
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
+
+# PrimeThink CLI
+
+Command Line Interface for interacting with the PrimeThink API.
+
+## Installation
+
+```bash
+pip install -e .
+```
+
+Or install from requirements:
+
+```bash
+pip install -r requirements.txt
+```
+
+## Quick Start
+
+1. Configure your API token:
+```bash
+primethink configure --token YOUR_API_TOKEN
+```
+
+2. Use the CLI commands:
+```bash
+primethink available-actions
+primethink execute-action --action "action-name" --message "your message"
+primethink send-message CHAT_ID --message "your message"
+primethink send-message --agent AGENT_ID --message "your message"
+```
+
+## Configuration
+
+### Token Management
+
+The CLI supports multiple token profiles, allowing you to switch between different accounts or API endpoints.
+
+#### Configure a new profile
+
+```bash
+primethink configure --token YOUR_API_TOKEN --profile default
+```
+
+Options:
+- `--token, -t`: Your API token (required)
+- `--profile, -p`: Profile name (default: "default")
+- `--api-url, -u`: Custom API URL (optional, default: https://api.primethink.ai)
+
+#### List all profiles
+
+```bash
+primethink list-profiles
+```
+
+#### Switch to a different profile
+
+```bash
+primethink use PROFILE_NAME
+```
+
+#### Remove a profile
+
+```bash
+primethink remove-profile PROFILE_NAME
+```
+
+## Commands
+
+### Version
+
+Display the CLI version:
+
+```bash
+primethink version
+```
+
+### Available Actions
+
+Get a list of available task actions:
+
+```bash
+primethink available-actions
+```
+
+You can use a specific profile for a single request without switching the active profile:
+
+```bash
+primethink available-actions --profile production
+```
+
+You can also override the API URL for a single request:
+
+```bash
+primethink available-actions --api-url https://custom-api.example.com
+```
+
+### Execute Task Action
+
+Execute a task action with a message and optional files:
+
+```bash
+primethink execute-action --action "action-name" --message "your message"
+```
+
+With files:
+
+```bash
+primethink execute-action --action "action-name" --message "your message" --files file1.txt --files file2.txt
+```
+
+Options:
+- `--action, -a`: Task action name (required)
+- `--message, -m`: Message input (required)
+- `--files, -f`: Files to attach (can be specified multiple times)
+- `--return-original`: Return original message (flag)
+- `--profile, -p`: Profile to use for this request (optional)
+- `--api-url, -u`: Override API URL for this request (optional)
+
+### Send Message to Chat
+
+Send a message to an existing chat by ID or mention name:
+
+```bash
+primethink send-message CHAT_ID_OR_MENTION --message "your message"
+```
+
+You can use either a chat ID or mention name:
+
+```bash
+# By chat ID
+primethink send-message 123 --message "your message"
+
+# By mention name
+primethink send-message @my-assistant --message "your message"
+```
+
+With files:
+
+```bash
+primethink send-message 123 --message "your message" --files file1.txt --files file2.txt
+```
+
+Options:
+- `CHAT_ID_OR_MENTION`: The chat ID or mention name (optional, positional argument)
+- `--message, -m`: Message input (required)
+- `--files, -f`: Files to attach (can be specified multiple times)
+- `--async`: Asynchronous request (flag, default: false, applies only to chat messages)
+- `--agent, -a`: Agent ID to send message to (optional, use instead of CHAT_ID_OR_MENTION)
+- `--profile, -p`: Profile to use for this request (optional)
+- `--api-url, -u`: Override API URL for this request (optional)
+
+**Note:** You must provide either `CHAT_ID_OR_MENTION` or `--agent`, but not both.
+
+### Send Message to Agent
+
+Send a message directly to an agent using the `--agent` option:
+
+```bash
+primethink send-message --agent 1 --message "your message"
+```
+
+With files:
+
+```bash
+primethink send-message --agent 1 --message "Analyze this data" --files data.csv
+```
+
+## Configuration File
+
+The CLI stores configuration in `~/.primethink/config.json`. This file contains:
+- Active profile
+- All configured profiles with their tokens and optional custom API URLs
+
+Example structure:
+
+```json
+{
+  "active_profile": "default",
+  "profiles": {
+    "default": {
+      "token": "your-api-token"
+    },
+    "custom": {
+      "token": "your-custom-token",
+      "api_url": "https://custom-api.example.com"
+    }
+  }
+}
+```
+
+## Development
+
+### Install development dependencies
+
+```bash
+pip install -e ".[dev]"
+```
+
+### Run tests
+
+```bash
+pytest
+```
+
+### Run tests with coverage
+
+```bash
+pytest --cov=primethink --cov-report=html
+```
+
+## Examples
+
+### Example 1: Configure and use the CLI
+
+```bash
+# Configure your token
+primethink configure --token YOUR_API_TOKEN
+
+# List available actions
+primethink available-actions
+
+# Execute an action
+primethink execute-action --action "summarize" --message "Summarize this text"
+```
+
+### Example 2: Using multiple profiles
+
+```bash
+# Configure default profile
+primethink configure --token YOUR_TOKEN --profile default
+
+# Configure a custom API endpoint profile
+primethink configure --token CUSTOM_TOKEN --profile production --api-url https://prod-api.example.com
+
+# List profiles
+primethink list-profiles
+
+# Switch to production profile
+primethink use production
+
+# Or use a specific profile for a single command without switching
+primethink available-actions --profile production
+primethink send-message 123 --message "Hello" --profile default
+
+# Switch back to default
+primethink use default
+```
+
+### Example 3: Sending messages with files
+
+```bash
+# Send a message to a chat with multiple files
+primethink send-message 123 --message "Here are the documents" --files report.pdf --files data.csv
+
+# Send message by mention name
+primethink send-message @my-assistant --message "Review this"
+
+# Send to agent with a specific profile
+primethink send-message --agent 1 --message "Help me with this task" --files data.csv --profile production
+
+# Use a custom API URL for a single request
+primethink send-message 123 --message "Hello" --api-url https://custom-api.example.com
+```
+
+## License
+
+MIT License
+
+## Support
+
+For issues and questions, please visit: https://primethink.ai

primethink_cli-1.0.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+primethink.py,sha256=bv-eOYO271YQGOo_FaSFrSMrCov_visQlWj8UytobyY,37657
+primethink_cli-1.0.0.dist-info/licenses/LICENSE,sha256=UxGf3lQlPQ5LS5gHeJ1qpVvlutxtTeNGCSdgWlCB4gg,1067
+primethink_cli-1.0.0.dist-info/METADATA,sha256=NsxrHDdnfli-dph2RE8kGhk4E0VInlesfijtxcvYz_s,7378
+primethink_cli-1.0.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+primethink_cli-1.0.0.dist-info/entry_points.txt,sha256=UPFuP23YqanhsicGLir0lX-6qLlA88Z60Xl1QddOAzY,38
+primethink_cli-1.0.0.dist-info/top_level.txt,sha256=nzEk_Oen-qTbKFMoifnqyauyzPuwYAI3kIxr7DQHC2s,11
+primethink_cli-1.0.0.dist-info/RECORD,,

primethink_cli-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

primethink_cli-1.0.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+pt = primethink:cli

primethink_cli-1.0.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 PrimeThink
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

primethink_cli-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+primethink