Module `cogsgpt.utils`

Expand source code

from __future__ import annotations

import os
from urllib.parse import urlparse

from cogsgpt.schema import FileSource

def detect_file_source(file_path: str) -> FileSource:
    """
    Detect whether the input is a local file path or a web URL.

    Args:
        file_path (str): The input file path or web URL.

    Returns:
        FileSource: The detected file source.

    Raises:
        ValueError: If the input is neither a local file path nor a web URL.
    """
    # Check if the input is a local file path
    if os.path.isfile(file_path):
        return FileSource.LOCAL

    # Check if the input is a web URL
    parsed_url = urlparse(file_path)
    if parsed_url.scheme and parsed_url.netloc:
        return FileSource.REMOTE

    # If the input is neither a local file path nor a web URL, raise an exception
    raise ValueError("Input must be a local file path or a web URL")

Functions

def detect_file_source(file_path: str) ‑> FileSource

Detect whether the input is a local file path or a web URL.

Args

file_path : str: The input file path or web URL.

Returns

FileSource: The detected file source.

Raises

ValueError: If the input is neither a local file path nor a web URL.

Expand source code

def detect_file_source(file_path: str) -> FileSource:
    """
    Detect whether the input is a local file path or a web URL.

    Args:
        file_path (str): The input file path or web URL.

    Returns:
        FileSource: The detected file source.

    Raises:
        ValueError: If the input is neither a local file path nor a web URL.
    """
    # Check if the input is a local file path
    if os.path.isfile(file_path):
        return FileSource.LOCAL

    # Check if the input is a web URL
    parsed_url = urlparse(file_path)
    if parsed_url.scheme and parsed_url.netloc:
        return FileSource.REMOTE

    # If the input is neither a local file path nor a web URL, raise an exception
    raise ValueError("Input must be a local file path or a web URL")