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

seekrai/legacy/files.py

@@ -1,140 +0,0 @@
-from __future__ import annotations
-
-import json
-import warnings
-from typing import Any, Dict, List
-
-import seekrai
-from seekrai.legacy.base import API_KEY_WARNING, deprecated
-from seekrai.utils.files import check_file as check_json
-
-
-class Files:
-    @classmethod
-    @deprecated  # type: ignore
-    def list(
-        cls,
-    ) -> Dict[str, Any]:
-        """Legacy file list 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.files.list().model_dump()
-
-    @classmethod
-    def check(self, file: str) -> Dict[str, object]:
-        return check_json(file)
-
-    @classmethod
-    @deprecated  # type: ignore
-    def upload(
-        cls,
-        file: str,
-        check: bool = True,
-    ) -> Dict[str, Any]:
-        """Legacy file upload function."""
-
-        api_key = None
-        if seekrai.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = seekrai.api_key
-
-        if check:
-            report_dict = check_json(file)
-            if not report_dict["is_check_passed"]:
-                raise seekrai.error.FileTypeError(
-                    f"Invalid file supplied. Failed to upload.\nReport:\n {report_dict}"
-                )
-
-        client = seekrai.SeekrFlow(api_key=api_key)
-
-        response = client.files.upload(file=file).model_dump()
-
-        response["report_dict"] = report_dict
-
-        return response
-
-    @classmethod
-    @deprecated  # type: ignore
-    def delete(
-        cls,
-        file_id: str,
-    ) -> Dict[str, Any]:
-        """Legacy file delete 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.files.delete(id=file_id).model_dump()
-
-    @classmethod
-    @deprecated  # type: ignore
-    def retrieve(
-        cls,
-        file_id: str,
-    ) -> Dict[str, Any]:
-        """Legacy file retrieve 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.files.retrieve(id=file_id).model_dump()
-
-    @classmethod
-    @deprecated  # type: ignore
-    def retrieve_content(
-        cls,
-        file_id: str,
-        output: str | None = None,
-    ) -> Dict[str, Any]:
-        """Legacy file retrieve content 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.files.retrieve_content(id=file_id, output=output).model_dump()
-
-    @classmethod
-    @deprecated  # type: ignore
-    def save_jsonl(
-        self, data: Dict[str, str], output_path: str, append: bool = False
-    ) -> None:
-        """
-        Write list of objects to a JSON lines file.
-        """
-        mode = "a+" if append else "w"
-        with open(output_path, mode, encoding="utf-8") as f:
-            for line in data:
-                json_record = json.dumps(line, ensure_ascii=False)
-                f.write(json_record + "\n")
-        print("Wrote {} records to {}".format(len(data), output_path))
-
-    @classmethod
-    @deprecated  # type: ignore
-    def load_jsonl(self, input_path: str) -> List[Dict[str, str]]:
-        """
-        Read list of objects from a JSON lines file.
-        """
-        data = []
-        with open(input_path, "r", encoding="utf-8") as f:
-            for line in f:
-                data.append(json.loads(line.rstrip("\n|\r")))
-        print("Loaded {} records from {}".format(len(data), input_path))
-        return data

seekrai/legacy/finetune.py

@@ -1,173 +0,0 @@
-from __future__ import annotations
-
-import warnings
-from typing import Any, Dict, List
-
-import seekrai
-from seekrai.legacy.base import API_KEY_WARNING, deprecated
-
-
-class Finetune:
-    @classmethod
-    @deprecated  # type: ignore
-    def create(
-        cls,
-        training_file: str,  # training file_id
-        model: str,
-        n_epochs: int = 1,
-        n_checkpoints: int | None = 1,
-        batch_size: int | None = 32,
-        learning_rate: float = 0.00001,
-        suffix: (
-            str | None
-        ) = None,  # resulting finetuned model name will include the suffix
-        estimate_price: bool = False,
-        wandb_api_key: str | None = None,
-        confirm_inputs: bool = False,
-    ):
-        api_key = None
-        if seekrai.api_key:
-            warnings.warn(API_KEY_WARNING)
-            api_key = seekrai.api_key
-
-        if estimate_price:
-            raise ValueError("Price estimation is not supported in version >= 1.0.1")
-
-        if confirm_inputs:
-            raise ValueError("Input confirmation is not supported in version >= 1.0.1")
-
-        client = seekrai.SeekrFlow(api_key=api_key)
-
-        return 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,
-        ).model_dump()
-
-    @classmethod
-    @deprecated  # type: ignore
-    def list(
-        cls,
-    ) -> Dict[str, Any]:
-        """Legacy finetuning list 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.fine_tuning.list().model_dump()
-
-    @classmethod
-    @deprecated  # type: ignore
-    def retrieve(
-        cls,
-        fine_tune_id: str,
-    ) -> Dict[str, Any]:
-        """Legacy finetuning retrieve 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.fine_tuning.retrieve(id=fine_tune_id).model_dump()
-
-    @classmethod
-    @deprecated  # type: ignore
-    def cancel(
-        cls,
-        fine_tune_id: str,
-    ) -> Dict[str, Any]:
-        """Legacy finetuning cancel 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.fine_tuning.cancel(id=fine_tune_id).model_dump()
-
-    @classmethod
-    @deprecated  # type: ignore
-    def list_events(
-        cls,
-        fine_tune_id: str,
-    ) -> Dict[str, Any]:
-        """Legacy finetuning list events 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.fine_tuning.list_events(id=fine_tune_id).model_dump()
-
-    @classmethod
-    @deprecated  # type: ignore
-    def get_checkpoints(
-        cls,
-        fine_tune_id: str,
-    ) -> List[Any]:
-        """Legacy finetuning get checkpoints function."""
-
-        finetune_events = list(cls.retrieve(fine_tune_id=fine_tune_id)["events"])
-
-        saved_events = [i for i in finetune_events if i["type"] in ["CHECKPOINT_SAVE"]]
-
-        return saved_events
-
-    @classmethod
-    @deprecated  # type: ignore
-    def get_job_status(cls, fine_tune_id: str) -> str:
-        """Legacy finetuning get job status function."""
-        return str(cls.retrieve(fine_tune_id=fine_tune_id)["status"])
-
-    @classmethod
-    @deprecated  # type: ignore
-    def is_final_model_available(cls, fine_tune_id: str) -> bool:
-        """Legacy finetuning is final model available function."""
-
-        finetune_events = list(cls.retrieve(fine_tune_id=fine_tune_id)["events"])
-
-        for i in finetune_events:
-            if i["type"] in ["JOB_COMPLETE", "JOB_ERROR"]:
-                if i["checkpoint_path"] != "":
-                    return False
-                else:
-                    return True
-        return False
-
-    @classmethod
-    @deprecated  # type: ignore
-    def download(
-        cls,
-        fine_tune_id: str,
-        output: str | None = None,
-        step: int = -1,
-    ) -> Dict[str, Any]:
-        """Legacy finetuning download 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.fine_tuning.download(
-            id=fine_tune_id, output=output, checkpoint_step=step
-        ).model_dump()

seekrai/legacy/images.py

@@ -1,25 +0,0 @@
-import warnings
-from typing import Any, Dict
-
-import seekrai
-from seekrai.legacy.base import API_KEY_WARNING, deprecated
-
-
-class Image:
-    @classmethod
-    @deprecated  # type: ignore
-    def create(
-        cls,
-        prompt: str,
-        **kwargs,
-    ) -> Dict[str, Any]:
-        """Legacy image 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.images.generate(prompt=prompt, **kwargs).model_dump()

seekrai/legacy/models.py

@@ -1,44 +0,0 @@
-import warnings
-from typing import Any, Dict, List
-
-import seekrai
-from seekrai.legacy.base import API_KEY_WARNING, deprecated
-
-
-class Models:
-    @classmethod
-    @deprecated  # type: ignore
-    def list(
-        cls,
-    ) -> List[Dict[str, Any]]:
-        """Legacy model list 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 [item.model_dump() for item in client.models.list()]
-
-    @classmethod
-    @deprecated  # type: ignore
-    def info(
-        cls,
-        model: str,
-    ) -> Dict[str, Any]:
-        """Legacy model info 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)
-
-        model_list = client.models.list()
-
-        for item in model_list:
-            if item.id == model:
-                return item.model_dump()