lionagi 0.5.5__py3-none-any.whl → 0.6.1__py3-none-any.whl

lionagi/integrations/openai_/api_endpoints/embeddings/response_body.py

@@ -1,67 +0,0 @@
-from typing import List, Literal
-
-from pydantic import BaseModel, ConfigDict, Field
-
-from ..data_models import OpenAIEndpointResponseBody
-
-
-class EmbeddingObject(BaseModel):
-    index: int = Field(
-        description="The index of the embedding in the list of embeddings."
-    )
-
-    embedding: list[float] = Field(
-        description=(
-            "The embedding vector, which is a list of floats. The length of "
-            "vector depends on the model as listed in the embedding guide."
-        ),
-    )
-
-    object: Literal["embedding"] = Field(
-        "embedding",
-        description='The object type, which is always "embedding".',
-    )
-
-
-class Usage(BaseModel):
-    prompt_tokens: int = Field(description="Number of tokens in the prompt.")
-
-    total_tokens: int = Field(
-        description="Total number of tokens used in "
-        "the request (prompt + completion)"
-    )
-
-
-class OpenAIEmbeddingResponseBody(OpenAIEndpointResponseBody):
-    object: Literal["list"] = Field(
-        description='The object type, which is always "list" for embeddings.'
-    )
-    data: list[EmbeddingObject] = Field(
-        description="The list of embedding objects."
-    )
-    model: str = Field(
-        description="The ID of the model used to generate the embeddings."
-    )
-    usage: Usage = Field(description="The usage statistics for the request.")
-
-    model_config = ConfigDict(
-        json_schema_extra={
-            "example": {
-                "object": "list",
-                "data": [
-                    {
-                        "index": 0,
-                        "embedding": [
-                            0.0023064255,
-                            -0.009327292,
-                            ...,
-                            -0.0028842222,
-                        ],
-                        "object": "embedding",
-                    }
-                ],
-                "model": "text-embedding-ada-002",
-                "usage": {"prompt_tokens": 8, "total_tokens": 8},
-            }
-        }
-    )

lionagi/integrations/openai_/api_endpoints/files/__init__.py

@@ -1,11 +0,0 @@
-from .delete_file import OpenAIDeleteFilePathParam
-from .list_files import OpenAIListFilesQueryParam
-from .retrieve_file import OpenAIRetrieveFilePathParam
-from .upload_file import OpenAIUploadFileRequestBody
-
-__all__ = [
-    "OpenAIUploadFileRequestBody",
-    "OpenAIListFilesQueryParam",
-    "OpenAIRetrieveFilePathParam",
-    "OpenAIDeleteFilePathParam",
-]

lionagi/integrations/openai_/api_endpoints/files/delete_file.py

@@ -1,20 +0,0 @@
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointPathParam, OpenAIEndpointResponseBody
-
-
-class OpenAIDeleteFilePathParam(OpenAIEndpointPathParam):
-    file_id: str = Field(
-        description="The ID of the file to use for this request."
-    )
-
-
-class OpenAIDeleteFileResponseBody(OpenAIEndpointResponseBody):
-    id: str = Field(
-        description="The file identifier, which can be referenced "
-        "in the API endpoints.",
-    )
-
-    object: str = Field(description="The object type, which is always file.")
-
-    deleted: bool = Field(description="The deletion status")

lionagi/integrations/openai_/api_endpoints/files/file_models.py

@@ -1,56 +0,0 @@
-from enum import Enum
-
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointResponseBody
-
-
-class Purpose(str, Enum):
-    ASSISTANTS = "assistants"
-    ASSISTANTS_OUTPUT = "assistants_output"
-    VISION = "vision"
-    BATCH = "batch"
-    BATCH_OUTPUT = "batch_output"
-    FINE_TUNE = "fine-tune"
-    FINE_TUNE_RESULTS = "fine-tune-results"
-
-
-class Status(str, Enum):
-    UPLOADED = "uploaded"
-    PROCESSED = "processed"
-    ERROR = "error"
-
-
-class OpenAIFileResponseBody(OpenAIEndpointResponseBody):
-    id: str = Field(
-        description="The file identifier, which can be "
-        "referenced in the API endpoints.",
-    )
-
-    bytes: int = Field(description="The size of the file, in bytes.")
-
-    created_at: int = Field(
-        description="The Unix timestamp (in seconds) for "
-        "when the file was created.",
-    )
-
-    filename: str = Field(description="The name of the file.")
-
-    object: str = Field(description="The object type, which is always file.")
-
-    purpose: Purpose = Field(description="The intended purpose of the file.")
-
-    status: Status = Field(
-        None,
-        description="The current status of the file, which can be "
-        "either uploaded, processed, or error.",
-        deprecated=True,
-    )
-
-    status_details: str | None = Field(
-        None,
-        description="For details on why a fine-tuning training "
-        "file failed validation, "
-        "see the error field on fine_tuning.job.",
-        deprecated=True,
-    )

lionagi/integrations/openai_/api_endpoints/files/list_files.py

@@ -1,27 +0,0 @@
-from typing import List, Literal, Optional
-
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointQueryParam, OpenAIEndpointResponseBody
-from .file_models import OpenAIFileResponseBody
-
-
-class OpenAIListFilesQueryParam(OpenAIEndpointQueryParam):
-    purpose: str | None = Field(
-        description="Only return files with the given purpose."
-    )
-
-
-class OpenAIListFilesResponseBody(OpenAIEndpointResponseBody):
-    data: list[OpenAIFileResponseBody] = Field(
-        description="The list of file objects."
-    )
-
-    object: Literal["list"] = Field(
-        description='The object type, which is always "list".'
-    )
-
-    has_more: bool = Field(
-        description="Whether there are more results"
-        " available after this batch."
-    )

lionagi/integrations/openai_/api_endpoints/files/retrieve_file.py

@@ -1,9 +0,0 @@
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointPathParam
-
-
-class OpenAIRetrieveFilePathParam(OpenAIEndpointPathParam):
-    file_id: str = Field(
-        description="The ID of the file to use for this request."
-    )

lionagi/integrations/openai_/api_endpoints/files/upload_file.py

@@ -1,38 +0,0 @@
-from enum import Enum
-from typing import IO
-
-from pydantic import ConfigDict, Field, field_validator
-
-from ..data_models import OpenAIEndpointRequestBody
-
-
-class Purpose(str, Enum):
-    ASSISTANT = "assistant"
-    VISION = "vision"
-    BATCH = "batch"
-    FINE_TUNE = "fine-tune"
-
-
-class OpenAIUploadFileRequestBody(OpenAIEndpointRequestBody):
-    file: str | IO = Field(
-        description="The File object (not file name) to be uploaded."
-    )
-
-    purpose: Purpose = Field(
-        description="The intended purpose of the uploaded file."
-    )
-
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-
-    @field_validator("file")
-    @classmethod
-    def get_file_object(cls, value):
-        if isinstance(value, str):
-            value = open(value, "rb")
-        return value
-
-    def __del__(self):
-        # Ensure the file is closed when the object is destroyed
-        file_obj = self.__dict__.get("file")
-        if file_obj and not file_obj.closed:
-            file_obj.close()

lionagi/integrations/openai_/api_endpoints/fine_tuning/__init__.py

@@ -1,37 +0,0 @@
-from .cancel_jobs import OpenAICancelFineTuningPathParam
-from .create_jobs import (
-    Hyperparameters,
-    Integration,
-    OpenAICreateFineTuningJobRequestBody,
-    Wandb,
-)
-from .list_fine_tuning_checkpoints import (
-    OpenAIListFineTuningCheckpointsPathParam,
-    OpenAIListFineTuningCheckpointsQueryParam,
-)
-from .list_fine_tuning_events import (
-    OpenAIListFineTuningEventsPathParam,
-    OpenAIListFineTuningEventsQueryParam,
-)
-from .list_fine_tuning_jobs import OpenAIListFineTuningJobsQueryParam
-from .retrieve_jobs import OpenAIRetrieveFineTuningJobPathParam
-from .training_format import (
-    OpenAIChatModelTrainingFormat,
-    OpenAICompletionsModelTrainingFormat,
-)
-
-__all__ = [
-    "OpenAICreateFineTuningJobRequestBody",
-    "Hyperparameters",
-    "Integration",
-    "Wandb",
-    "OpenAIListFineTuningJobsQueryParam",
-    "OpenAIListFineTuningEventsPathParam",
-    "OpenAIListFineTuningEventsQueryParam",
-    "OpenAIListFineTuningCheckpointsPathParam",
-    "OpenAIListFineTuningCheckpointsQueryParam",
-    "OpenAIRetrieveFineTuningJobPathParam",
-    "OpenAICancelFineTuningPathParam",
-    "OpenAIChatModelTrainingFormat",
-    "OpenAICompletionsModelTrainingFormat",
-]

lionagi/integrations/openai_/api_endpoints/fine_tuning/cancel_jobs.py

@@ -1,9 +0,0 @@
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointPathParam
-
-
-class OpenAICancelFineTuningPathParam(OpenAIEndpointPathParam):
-    fine_tuning_job_id: str = Field(
-        description="The ID of the fine-tuning job to get events for."
-    )

lionagi/integrations/openai_/api_endpoints/fine_tuning/create_jobs.py

@@ -1,133 +0,0 @@
-from typing import List, Literal, Optional
-
-from pydantic import BaseModel, ConfigDict, Field
-
-from ..data_models import OpenAIEndpointRequestBody
-
-
-class Wandb(BaseModel):
-    project: str = Field(
-        description="The name of the project for the new run."
-    )
-
-    name: str | None = Field(None, description="A display name for the run.")
-
-    entity: str | None = Field(
-        None, description="The entity (team or username) for the WandB run."
-    )
-
-    tags: list[str] | None = Field(
-        None,
-        description="A list of tags to be attached to the newly created run.",
-    )
-
-
-class Integration(BaseModel):
-    type: Literal["wandb"] = Field(
-        description="The type of integration to enable."
-        " Only 'wandb' is supported."
-    )
-
-    wandb: Wandb = Field(
-        description="Settings for Weights and Biases integration."
-    )
-
-
-class Hyperparameters(BaseModel):
-    batch_size: str | int | None = Field(
-        "auto",
-        description=(
-            "Number of examples in each batch. A larger"
-            " batch size means that "
-            "model parameters are updated less frequently, but with lower variance."
-        ),
-    )
-    learning_rate_multiplier: str | float | None = Field(
-        "auto",
-        description=(
-            "Scaling factor for the learning rate. A smaller learning rate "
-            "may be useful to avoid overfitting."
-        ),
-    )
-    n_epochs: str | int | None = Field(
-        "auto",
-        description=(
-            "The number of epochs to train the model for. An epoch refers "
-            "to one full cycle through the training dataset."
-        ),
-    )
-
-    # TODO: check hyperparameters
-    model_config = ConfigDict(extra="allow")
-
-
-class OpenAICreateFineTuningJobRequestBody(OpenAIEndpointRequestBody):
-    model: str = Field(description="The name of the model to fine-tune.")
-
-    training_file: str = Field(
-        description=(
-            "The ID of an uploaded file that contains training data. "
-            "Must be a JSONL file with 'purpose' set to 'fine-tune'."
-        ),
-    )
-
-    hyperparameters: Hyperparameters | None = Field(
-        None, description="The hyperparameters used for the fine-tuning job."
-    )
-
-    suffix: str | None = Field(
-        None,
-        max_length=64,
-        description=(
-            "A string of up to 64 characters that will be added to your "
-            "fine-tuned model name."
-        ),
-    )
-
-    validation_file: str | None = Field(
-        None,
-        description=(
-            "The ID of an uploaded file that contains validation data. "
-            "Must be a JSONL file with 'purpose' set to 'fine-tune'."
-        ),
-    )
-    integrations: list[Integration] | None = Field(
-        None,
-        description="A list of integrations to enable for your fine-tuning job.",
-    )
-
-    seed: int | None = Field(
-        None,
-        description=(
-            "The seed controls the reproducibility of the job. Passing in the "
-            "same seed and job parameters should produce the same results."
-        ),
-    )
-
-    model_config = ConfigDict(
-        json_schema_extra={
-            "example": {
-                "training_file": "file-BK7bzQj3FfZFXr7DbL6xJwfo",
-                "model": "gpt-4o-mini",
-                "hyperparameters": {
-                    "batch_size": "auto",
-                    "learning_rate_multiplier": "auto",
-                    "n_epochs": "auto",
-                },
-                "suffix": "custom-model-name",
-                "validation_file": None,
-                "integrations": [
-                    {
-                        "type": "wandb",
-                        "wandb": {
-                            "project": "my-project",
-                            "name": "my-run",
-                            "entity": "my-team",
-                            "tags": ["tag1", "tag2"],
-                        },
-                    }
-                ],
-                "seed": 42,
-            }
-        }
-    )

lionagi/integrations/openai_/api_endpoints/fine_tuning/fine_tuning_job_checkpoint_models.py

@@ -1,58 +0,0 @@
-from typing import Literal
-
-from pydantic import BaseModel, Field
-
-from ..data_models import OpenAIEndpointResponseBody
-
-
-class Metrics(BaseModel):
-    step: int = Field(
-        description="The step number associated with these metrics."
-    )
-
-    train_loss: float = Field(description="The training loss at this step.")
-
-    train_mean_token_accuracy: float = Field(
-        description="The mean token accuracy during training at this step."
-    )
-    valid_loss: float = Field(description="The validation loss at this step.")
-
-    valid_mean_token_accuracy: float = Field(
-        description="The mean token accuracy during validation at this step."
-    )
-
-    full_valid_loss: float = Field(
-        description="The full validation loss at this step."
-    )
-
-    full_valid_mean_token_accuracy: float = Field(
-        description="The full mean token accuracy during validation at this step."
-    )
-
-
-class OpenAIFineTuningJobCheckpointResponseBody(OpenAIEndpointResponseBody):
-    id: str = Field(description="The ID of the checkpoint.")
-
-    created_at: int = Field(
-        description="The Unix timestamp (in seconds) of when the checkpoint was created.",
-    )
-
-    fine_tuned_model_checkpoint: str = Field(
-        description="The name of the fine-tuned model checkpoint."
-    )
-
-    step_number: int = Field(
-        description="The step number of this checkpoint in the fine-tuning process.",
-    )
-
-    metrics: Metrics = Field(
-        description="Metrics associated with the checkpoint."
-    )
-
-    fine_tuning_job_id: str = Field(
-        description="The ID of the fine-tuning job this checkpoint belongs to."
-    )
-
-    object: Literal["fine_tuning.job.checkpoint"] = Field(
-        description="The object type, always 'fine_tuning.job.checkpoint'.",
-    )

lionagi/integrations/openai_/api_endpoints/fine_tuning/fine_tuning_job_event_models.py

@@ -1,31 +0,0 @@
-from typing import Literal, Optional
-
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointResponseBody
-
-
-class OpenAIFineTuningJobEventResponseBody(OpenAIEndpointResponseBody):
-    id: str = Field(description="The ID of the fine-tuning event.")
-
-    created_at: int = Field(
-        description="The Unix timestamp (in seconds) of when the event was created.",
-    )
-
-    level: str = Field(description="The level of the event (e.g., 'info').")
-
-    object: Literal["fine_tuning.job.event"] = Field(
-        "fine_tuning.job.event",
-        description="The object type, always 'fine_tuning.job.event'.",
-    )
-
-    message: str = Field(
-        ..., description="The message associated with the event."
-    )
-
-    # TODO: check Job Event Response
-    data: dict | None = Field(
-        None, description="Additional data associated with the event, if any."
-    )
-
-    type: str = Field(description="The type of the event (e.g., 'message').")

lionagi/integrations/openai_/api_endpoints/fine_tuning/fine_tuning_job_models.py

@@ -1,140 +0,0 @@
-from enum import Enum
-from typing import List, Literal, Optional, Union
-
-from pydantic import BaseModel, ConfigDict, Field
-
-from ..data_models import OpenAIEndpointResponseBody
-
-
-class Status(str, Enum):
-    VALIDATING_FILES = "validating_files"
-    QUEUED = "queued"
-    RUNNING = "running"
-    SUCCEEDED = "succeeded"
-    FAILED = "failed"
-    CANCELLED = "cancelled"
-
-
-class Error(BaseModel):
-    code: str | None = Field(
-        None, description="A machine-readable error code."
-    )
-
-    message: str | None = Field(
-        None, description="A human-readable error message."
-    )
-
-    param: str | None = Field(
-        None, description="The parameter that was invalid, if applicable."
-    )
-
-
-class Hyperparameters(BaseModel):
-    n_epochs: Literal["auto"] | int = Field(
-        description="The number of epochs to train the model for. 'auto' or 1-50.",
-    )
-
-    # TODO: check hyperparameters
-    batch_size: Literal["auto"] | int = Field(
-        None, description="The batch size."
-    )
-
-    learning_rate_multiplier: Literal["auto"] | int = Field(
-        None, description="The learning rate multiplier."
-    )
-
-    model_config = ConfigDict(extra="allow")
-
-
-class Wandb(BaseModel):
-    project: str = Field(
-        ..., description="The name of the project for the new run."
-    )
-
-    name: str | None = Field(None, description="A display name for the run.")
-
-    entity: str | None = Field(
-        None, description="The entity to use for the run."
-    )
-
-    tags: list[str] | None = Field(description="A list of tags for the run.")
-
-
-class Integration(BaseModel):
-    type: str = Field(
-        ..., description="The type of the integration being enabled."
-    )
-
-    wandb: Wandb | None = Field(
-        None, description="Settings for Weights and Biases integration."
-    )
-
-
-class OpenAIFineTuningJobResponseBody(OpenAIEndpointResponseBody):
-    id: str = Field(
-        description="The object identifier, referenceable in API endpoints."
-    )
-
-    created_at: int = Field(
-        ..., description="The Unix timestamp (in seconds) of job creation."
-    )
-
-    error: Error | None = Field(
-        None, description="Error information for failed jobs."
-    )
-
-    fine_tuned_model: str | None = Field(
-        None, description="The name of the fine-tuned model being created."
-    )
-
-    finished_at: int | None = Field(
-        None, description="The Unix timestamp (in seconds) of job completion."
-    )
-
-    hyperparameters: Hyperparameters = Field(
-        description="The hyperparameters used for the fine-tuning job."
-    )
-
-    model: str = Field(description="The base model being fine-tuned.")
-
-    object: Literal["fine_tuning.job"] = Field(
-        description="The object type, always 'fine_tuning.job'."
-    )
-
-    organization_id: str = Field(
-        ..., description="The organization that owns the fine-tuning job."
-    )
-
-    result_files: list[str] = Field(
-        ..., description="The compiled results file ID(s) for the job."
-    )
-
-    status: Status = Field(
-        description=(
-            "The current status of the fine-tuning job: validating_files, "
-            "queued, running, succeeded, failed, or cancelled."
-        ),
-    )
-
-    trained_tokens: int | None = Field(
-        None, description="The total number of billable tokens processed."
-    )
-
-    training_file: str = Field(description="The file ID used for training.")
-
-    validation_file: str | None = Field(
-        None, description="The file ID used for validation."
-    )
-
-    integrations: list[Integration] | None = Field(
-        None, description="A list of integrations enabled for this job."
-    )
-
-    seed: int = Field(
-        None, description="The seed used for the fine-tuning job."
-    )
-
-    estimated_finish: int | None = Field(
-        None,
-        description="The Unix timestamp (in seconds) for estimated job completion.",
-    )