seekrai 0.0.1__py3-none-any.whl → 0.1.0__py3-none-any.whl

seekrai/cli/api/files.py

@@ -1,125 +0,0 @@
-import json
-import pathlib
-from textwrap import wrap
-
-import click
-from tabulate import tabulate
-
-from seekrai import seekrflow
-from seekrai.types import FilePurpose
-from seekrai.utils import check_file, convert_bytes, convert_unix_timestamp
-
-
-@click.group()
-@click.pass_context
-def files(ctx: click.Context) -> None:
-    """File API commands"""
-    pass
-
-
-@files.command()
-@click.pass_context
-@click.argument(
-    "file",
-    type=click.Path(
-        exists=True, file_okay=True, resolve_path=True, readable=True, dir_okay=False
-    ),
-    required=True,
-)
-@click.option(
-    "--purpose",
-    type=str,
-    default=FilePurpose.FineTune.value,
-    help="Purpose of file upload. Acceptable values in enum `seekrai.types.FilePurpose`. Defaults to `fine-tunes`.",
-)
-def upload(ctx: click.Context, file: pathlib.Path, purpose: str) -> None:
-    """Upload file"""
-
-    client: SeekrFlow = ctx.obj
-
-    response = client.files.upload(file=file, purpose=purpose)
-
-    click.echo(json.dumps(response.model_dump(), indent=4))
-
-
-@files.command()
-@click.pass_context
-def list(ctx: click.Context) -> None:
-    """List files"""
-    client: SeekrFlow = ctx.obj
-
-    response = client.files.list()
-
-    display_list = []
-    for i in response.data or []:
-        display_list.append(
-            {
-                "File name": "\n".join(wrap(i.filename or "", width=30)),
-                "File ID": i.id,
-                "Size": convert_bytes(
-                    float(str(i.bytes))
-                ),  # convert to string for mypy typing
-                "Created At": convert_unix_timestamp(i.created_at or 0),
-                "Line Count": i.line_count,
-            }
-        )
-    table = tabulate(display_list, headers="keys", tablefmt="grid", showindex=True)
-
-    click.echo(table)
-
-
-@files.command()
-@click.pass_context
-@click.argument("id", type=str, required=True)
-def retrieve(ctx: click.Context, id: str) -> None:
-    """Upload file"""
-
-    client: SeekrFlow = ctx.obj
-
-    response = client.files.retrieve(id=id)
-
-    click.echo(json.dumps(response.model_dump(), indent=4))
-
-
-@files.command()
-@click.pass_context
-@click.argument("id", type=str, required=True)
-@click.option("--output", type=str, default=None, help="Output filename")
-def retrieve_content(ctx: click.Context, id: str, output: str) -> None:
-    """Retrieve file content and output to file"""
-
-    client: SeekrFlow = ctx.obj
-
-    response = client.files.retrieve_content(id=id, output=output)
-
-    click.echo(json.dumps(response.model_dump(), indent=4))
-
-
-@files.command()
-@click.pass_context
-@click.argument("id", type=str, required=True)
-def delete(ctx: click.Context, id: str) -> None:
-    """Delete remote file"""
-
-    client: SeekrFlow = ctx.obj
-
-    response = client.files.delete(id=id)
-
-    click.echo(json.dumps(response.model_dump(), indent=4))
-
-
-@files.command()
-@click.pass_context
-@click.argument(
-    "file",
-    type=click.Path(
-        exists=True, file_okay=True, resolve_path=True, readable=True, dir_okay=False
-    ),
-    required=True,
-)
-def check(ctx: click.Context, file: pathlib.Path) -> None:
-    """Check file for issues"""
-
-    report = check_file(file)
-
-    click.echo(json.dumps(report, indent=4))

seekrai/cli/api/finetune.py

@@ -1,175 +0,0 @@
-import json
-from textwrap import wrap
-
-import click
-from tabulate import tabulate
-
-from seekrai import seekrflow
-from seekrai.utils import finetune_price_to_dollars, parse_timestamp
-
-
-@click.group(name="fine-tuning")
-@click.pass_context
-def fine_tuning(ctx: click.Context) -> None:
-    """Fine-tunes API commands"""
-    pass
-
-
-@fine_tuning.command()
-@click.pass_context
-@click.option(
-    "--training-file", type=str, required=True, help="Training file ID from Files API"
-)
-@click.option("--model", type=str, required=True, help="Base model name")
-@click.option("--n-epochs", type=int, default=1, help="Number of epochs to train for")
-@click.option(
-    "--n-checkpoints", type=int, default=1, help="Number of checkpoints to save"
-)
-@click.option("--batch-size", type=int, default=32, help="Train batch size")
-@click.option("--learning-rate", type=float, default=3e-5, help="Learning rate")
-@click.option(
-    "--suffix", type=str, default=None, help="Suffix for the fine-tuned model name"
-)
-@click.option("--wandb-api-key", type=str, default=None, help="Wandb API key")
-def create(
-    ctx: click.Context,
-    training_file: str,
-    model: str,
-    n_epochs: int,
-    n_checkpoints: int,
-    batch_size: int,
-    learning_rate: float,
-    suffix: str,
-    wandb_api_key: str,
-) -> None:
-    """Start fine-tuning"""
-    client: SeekrFlow = ctx.obj
-
-    response = client.fine_tuning.create(
-        training_file=training_file,
-        model=model,
-        n_epochs=n_epochs,
-        n_checkpoints=n_checkpoints,
-        batch_size=batch_size,
-        learning_rate=learning_rate,
-        suffix=suffix,
-        wandb_api_key=wandb_api_key,
-    )
-
-    click.echo(json.dumps(response.model_dump(), indent=4))
-
-
-@fine_tuning.command()
-@click.pass_context
-def list(ctx: click.Context) -> None:
-    """List fine-tuning jobs"""
-    client: SeekrFlow = ctx.obj
-
-    response = client.fine_tuning.list()
-
-    response.data = response.data or []
-
-    response.data.sort(key=lambda x: parse_timestamp(x.created_at or ""))
-
-    display_list = []
-    for i in response.data:
-        display_list.append(
-            {
-                "Fine-tune ID": i.id,
-                "Model Output Name": "\n".join(wrap(i.output_name or "", width=30)),
-                "Status": i.status,
-                "Created At": i.created_at,
-                "Price": f"""${finetune_price_to_dollars(
-                    float(str(i.total_price))
-                )}""",  # convert to string for mypy typing
-            }
-        )
-    table = tabulate(display_list, headers="keys", tablefmt="grid", showindex=True)
-
-    click.echo(table)
-
-
-@fine_tuning.command()
-@click.pass_context
-@click.argument("fine_tune_id", type=str, required=True)
-def retrieve(ctx: click.Context, fine_tune_id: str) -> None:
-    """Retrieve fine-tuning job details"""
-    client: SeekrFlow = ctx.obj
-
-    response = client.fine_tuning.retrieve(fine_tune_id)
-
-    # remove events from response for cleaner output
-    response.events = None
-
-    click.echo(json.dumps(response.model_dump(), indent=4))
-
-
-@fine_tuning.command()
-@click.pass_context
-@click.argument("fine_tune_id", type=str, required=True)
-def cancel(ctx: click.Context, fine_tune_id: str) -> None:
-    """Cancel fine-tuning job"""
-    client: SeekrFlow = ctx.obj
-
-    response = client.fine_tuning.cancel(fine_tune_id)
-
-    click.echo(json.dumps(response.model_dump(), indent=4))
-
-
-@fine_tuning.command()
-@click.pass_context
-@click.argument("fine_tune_id", type=str, required=True)
-def list_events(ctx: click.Context, fine_tune_id: str) -> None:
-    """List fine-tuning events"""
-    client: SeekrFlow = ctx.obj
-
-    response = client.fine_tuning.list_events(fine_tune_id)
-
-    response.data = response.data or []
-
-    display_list = []
-    for i in response.data:
-        display_list.append(
-            {
-                "Message": "\n".join(wrap(i.message or "", width=50)),
-                "Type": i.type,
-                "Created At": parse_timestamp(i.created_at or ""),
-                "Hash": i.hash,
-            }
-        )
-    table = tabulate(display_list, headers="keys", tablefmt="grid", showindex=True)
-
-    click.echo(table)
-
-
-@fine_tuning.command()
-@click.pass_context
-@click.argument("fine_tune_id", type=str, required=True)
-@click.option(
-    "--output_dir",
-    type=click.Path(exists=True, file_okay=False, resolve_path=True),
-    required=False,
-    default=None,
-    help="Output directory",
-)
-@click.option(
-    "--checkpoint-step",
-    type=int,
-    required=False,
-    default=-1,
-    help="Download fine-tuning checkpoint. Defaults to latest.",
-)
-def download(
-    ctx: click.Context,
-    fine_tune_id: str,
-    output_dir: str,
-    checkpoint_step: int,
-) -> None:
-    """Download fine-tuning checkpoint"""
-    client: SeekrFlow = ctx.obj
-
-    response = client.fine_tuning.download(
-        fine_tune_id, output=output_dir, checkpoint_step=checkpoint_step
-    )
-
-    click.echo(json.dumps(response.model_dump(), indent=4))

seekrai/cli/api/images.py

@@ -1,82 +0,0 @@
-import base64
-import pathlib
-
-import click
-from PIL import Image
-
-from seekrai import seekrflow
-from seekrai.types import ImageResponse
-from seekrai.types.images import ImageChoicesData
-
-
-@click.group()
-@click.pass_context
-def images(ctx: click.Context) -> None:
-    """Images generations API commands"""
-    pass
-
-
-@images.command()
-@click.pass_context
-@click.argument("prompt", type=str, required=True)
-@click.option("--model", type=str, required=True, help="Model name")
-@click.option("--steps", type=int, default=20, help="Number of steps to run generation")
-@click.option("--seed", type=int, default=None, help="Random seed")
-@click.option("--n", type=int, default=1, help="Number of images to generate")
-@click.option("--height", type=int, default=1024, help="Image height")
-@click.option("--width", type=int, default=1024, help="Image width")
-@click.option("--negative-prompt", type=str, default=None, help="Negative prompt")
-@click.option(
-    "--output",
-    type=click.Path(exists=True, file_okay=False, resolve_path=True),
-    required=False,
-    default=pathlib.Path("."),
-    help="Output directory",
-)
-@click.option("--prefix", type=str, required=False, default="image-")
-@click.option("--no-show", is_flag=True, help="Do not open images in viewer")
-def generate(
-    ctx: click.Context,
-    prompt: str,
-    model: str,
-    steps: int,
-    seed: int,
-    n: int,
-    height: int,
-    width: int,
-    negative_prompt: str,
-    output: pathlib.Path,
-    prefix: str,
-    no_show: bool,
-) -> None:
-    """Generate image"""
-
-    client: SeekrFlow = ctx.obj
-
-    response = client.images.generate(
-        prompt=prompt,
-        model=model,
-        steps=steps,
-        seed=seed,
-        n=n,
-        height=height,
-        width=width,
-        negative_prompt=negative_prompt,
-    )
-
-    assert isinstance(response, ImageResponse)
-    assert isinstance(response.data, list)
-
-    for i, choice in enumerate(response.data):
-        assert isinstance(choice, ImageChoicesData)
-
-        with open(f"{output}/{prefix}{choice.index}.png", "wb") as f:
-            f.write(base64.b64decode(choice.b64_json))
-
-        click.echo(
-            f"Image [{i + 1}/{len(response.data)}] saved to {output}/{prefix}{choice.index}.png"
-        )
-
-        if not no_show:
-            image = Image.open(f"{output}/{prefix}{choice.index}.png")
-            image.show()

seekrai/cli/api/models.py

@@ -1,42 +0,0 @@
-from textwrap import wrap
-
-import click
-from tabulate import tabulate
-
-from seekrai import seekrflow
-from seekrai.types.models import ModelObject
-
-
-@click.group()
-@click.pass_context
-def models(ctx: click.Context) -> None:
-    """Models API commands"""
-    pass
-
-
-@models.command()
-@click.pass_context
-def list(ctx: click.Context) -> None:
-    """List models"""
-    client: SeekrFlow = ctx.obj
-
-    response = client.models.list()
-
-    display_list = []
-
-    model: ModelObject
-    for model in response:
-        display_list.append(
-            {
-                "ID": "\n".join(wrap(model.id or "", width=30)),
-                "Name": "\n".join(wrap(model.display_name or "", width=30)),
-                "Organization": model.organization,
-                "Type": model.type,
-                "Context Length": model.context_length,
-                "License": "\n".join(wrap(model.license or "", width=30)),
-                "Input per 1M token": model.pricing.input,
-                "Output per 1M token": model.pricing.output,
-            }
-        )
-
-    click.echo(tabulate(display_list, headers="keys", tablefmt="grid"))

seekrai/cli/cli.py

@@ -1,77 +0,0 @@
-from __future__ import annotations
-
-import os
-from typing import Any
-
-import click
-
-import seekrai
-from seekrai.cli.api.chat import chat, interactive
-from seekrai.cli.api.completions import completions
-from seekrai.cli.api.files import files
-from seekrai.cli.api.finetune import fine_tuning
-from seekrai.cli.api.images import images
-from seekrai.cli.api.models import models
-from seekrai.constants import MAX_RETRIES, TIMEOUT_SECS
-
-
-def print_version(ctx: click.Context, params: Any, value: Any) -> None:
-    if not value or ctx.resilient_parsing:
-        return
-    click.echo(f"Version {seekrai.version}")
-    ctx.exit()
-
-
-@click.group()
-@click.pass_context
-@click.option(
-    "--api-key",
-    type=str,
-    help="API Key. Defaults to environment variable `SEEKRFLOW_API_KEY`",
-    default=os.getenv("SEEKRFLOW_API_KEY"),
-)
-@click.option(
-    "--base-url", type=str, help="API Base URL. Defaults to SeekrFlow AI endpoint."
-)
-@click.option(
-    "--timeout", type=int, help=f"Request timeout. Defaults to {TIMEOUT_SECS} seconds"
-)
-@click.option(
-    "--max-retries",
-    type=int,
-    help=f"Maximum number of HTTP retries. Defaults to {MAX_RETRIES}.",
-)
-@click.option(
-    "--version",
-    is_flag=True,
-    callback=print_version,
-    expose_value=False,
-    is_eager=True,
-    help="Print version",
-)
-@click.option("--debug", help="Debug mode", is_flag=True)
-def main(
-    ctx: click.Context,
-    api_key: str | None,
-    base_url: str | None,
-    timeout: int | None,
-    max_retries: int | None,
-    debug: bool | None,
-) -> None:
-    """This is a sample CLI tool."""
-    seekrai.log = "debug" if debug else None
-    ctx.obj = seekrai.SeekrFlow(
-        api_key=api_key, base_url=base_url, timeout=timeout, max_retries=max_retries
-    )
-
-
-main.add_command(chat)
-main.add_command(interactive)
-main.add_command(completions)
-main.add_command(images)
-main.add_command(files)
-main.add_command(fine_tuning)
-main.add_command(models)
-
-if __name__ == "__main__":
-    main()

seekrai/legacy/base.py

@@ -1,27 +0,0 @@
-import functools
-import warnings
-
-
-API_KEY_WARNING = (
-    "The use of seekrai.api_key is deprecated and will be removed in the next major release. "
-    "Please set the SEEKRFLOW_API_KEY environment variable instead."
-)
-
-
-def deprecated(func):  # type: ignore
-    """
-    This is a decorator which can be used to mark functions
-    as deprecated. It will result in a warning being emitted
-    when the function is used.
-    """
-
-    @functools.wraps(func)
-    def new_func(*args, **kwargs):  # type: ignore
-        warnings.warn(
-            f"Call to deprecated function {func.__name__}.",
-            category=DeprecationWarning,
-            stacklevel=2,
-        )
-        return func(*args, **kwargs)
-
-    return new_func

seekrai/legacy/complete.py

@@ -1,91 +0,0 @@
-from __future__ import annotations
-
-import warnings
-from typing import Any, AsyncGenerator, Dict, Iterator
-
-import seekrai
-from seekrai.legacy.base import API_KEY_WARNING, deprecated
-from seekrai.types import CompletionChunk, CompletionResponse
-
-
-class Complete:
-    @classmethod
-    @deprecated  # type: ignore
-    def create(
-        cls,
-        prompt: str,
-        **kwargs,
-    ) -> Dict[str, Any]:
-        """Legacy completion function."""
-
-        api_key = None
-        if seekrai.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = seekrai.api_key
-
-        client = seekrai.SeekrFlow(api_key=api_key)
-
-        return client.completions.create(
-            prompt=prompt, stream=False, **kwargs
-        ).model_dump()  # type: ignore
-
-    @classmethod
-    @deprecated  # type: ignore
-    def create_streaming(
-        cls,
-        prompt: str,
-        **kwargs,
-    ) -> Iterator[Dict[str, Any]]:
-        """Legacy streaming completion function."""
-
-        api_key = None
-        if seekrai.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = seekrai.api_key
-
-        client = seekrai.SeekrFlow(api_key=api_key)
-
-        return (
-            token.model_dump()  # type: ignore
-            for token in client.completions.create(prompt=prompt, stream=True, **kwargs)
-        )
-
-
-class Completion:
-    @classmethod
-    @deprecated  # type: ignore
-    def create(
-        cls,
-        prompt: str,
-        **kwargs,
-    ) -> CompletionResponse | Iterator[CompletionChunk]:
-        """Completion function."""
-
-        api_key = None
-        if seekrai.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = seekrai.api_key
-
-        client = seekrai.SeekrFlow(api_key=api_key)
-
-        return client.completions.create(prompt=prompt, **kwargs)
-
-
-class AsyncComplete:
-    @classmethod
-    @deprecated  # type: ignore
-    async def create(
-        cls,
-        prompt: str,
-        **kwargs,
-    ) -> CompletionResponse | AsyncGenerator[CompletionChunk, None]:
-        """Async completion function."""
-
-        api_key = None
-        if seekrai.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = seekrai.api_key
-
-        client = seekrai.AsyncSeekrFlow(api_key=api_key)
-
-        return await client.completions.create(prompt=prompt, **kwargs)

seekrai/legacy/embeddings.py

@@ -1,25 +0,0 @@
-import warnings
-from typing import Any, Dict
-
-import seekrai
-from seekrai.legacy.base import API_KEY_WARNING, deprecated
-
-
-class Embeddings:
-    @classmethod
-    @deprecated  # type: ignore
-    def create(
-        cls,
-        input: str,
-        **kwargs,
-    ) -> Dict[str, Any]:
-        """Legacy embeddings function."""
-
-        api_key = None
-        if seekrai.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = seekrai.api_key
-
-        client = seekrai.SeekrFlow(api_key=api_key)
-
-        return client.embeddings.create(input=input, **kwargs).model_dump()