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

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

@@ -1,51 +0,0 @@
-from typing import List, Literal, Optional
-
-from pydantic import Field
-
-from ..data_models import (
-    OpenAIEndpointPathParam,
-    OpenAIEndpointQueryParam,
-    OpenAIEndpointResponseBody,
-)
-from .fine_tuning_job_checkpoint_models import (
-    OpenAIFineTuningJobCheckpointResponseBody,
-)
-
-
-class OpenAIListFineTuningCheckpointsPathParam(OpenAIEndpointPathParam):
-    fine_tuning_job_id: str = Field(
-        description="The ID of the fine-tuning job to get checkpoints for."
-    )
-
-
-class OpenAIListFineTuningCheckpointsQueryParam(OpenAIEndpointQueryParam):
-    after: str | None = Field(
-        None,
-        description="Identifier for the last checkpoint ID from the previous pagination request.",
-    )
-
-    limit: int | None = Field(
-        10, description="Number of checkpoints to retrieve.", ge=1
-    )
-
-
-class OpenAIListFineTuningCheckpointsResponseBody(OpenAIEndpointResponseBody):
-    object: Literal["list"] = Field(
-        description="The object type, always 'list' for a list of fine-tuning checkpoints.",
-    )
-
-    data: list[OpenAIFineTuningJobCheckpointResponseBody] = Field(
-        description="A list of fine-tuning checkpoints."
-    )
-
-    first_id: str | None = Field(
-        None, description="The ID of the first checkpoint in the list."
-    )
-
-    last_id: str | None = Field(
-        None, description="The ID of the last checkpoint in the list."
-    )
-
-    has_more: bool = Field(
-        description="Whether there are more results available after this batch."
-    )

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

@@ -1,42 +0,0 @@
-from typing import List, Literal, Optional
-
-from pydantic import Field
-
-from ..data_models import (
-    OpenAIEndpointPathParam,
-    OpenAIEndpointQueryParam,
-    OpenAIEndpointResponseBody,
-)
-from .fine_tuning_job_event_models import OpenAIFineTuningJobEventResponseBody
-
-
-# Request model
-class OpenAIListFineTuningEventsPathParam(OpenAIEndpointPathParam):
-    fine_tuning_job_id: str = Field(
-        description="The ID of the fine-tuning job to get events for."
-    )
-
-
-class OpenAIListFineTuningEventsQueryParam(OpenAIEndpointQueryParam):
-    after: str | None = Field(
-        None,
-        description="Identifier for the last event from the previous pagination request.",
-    )
-
-    limit: int | None = Field(
-        20, description="Number of events to retrieve.", ge=1
-    )
-
-
-class OpenAIListFineTuningEventsResponseBody(OpenAIEndpointResponseBody):
-    object: Literal["list"] = Field(
-        description="The object type, always 'list' for a list of fine-tuning events.",
-    )
-
-    data: list[OpenAIFineTuningJobEventResponseBody] = Field(
-        description="A list of fine-tuning events."
-    )
-
-    has_more: bool = Field(
-        description="Whether there are more results available after this batch."
-    )

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

@@ -1,31 +0,0 @@
-from typing import List, Literal, Optional
-
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointQueryParam, OpenAIEndpointResponseBody
-from .fine_tuning_job_models import OpenAIFineTuningJobResponseBody
-
-
-class OpenAIListFineTuningJobsQueryParam(OpenAIEndpointQueryParam):
-    after: str | None = Field(
-        None,
-        description="Identifier for the last job from the previous pagination request.",
-    )
-
-    limit: int | None = Field(
-        20, description="Number of fine-tuning jobs to retrieve.", ge=1
-    )
-
-
-class OpenAIListFineTuningJobsResponseBody(OpenAIEndpointResponseBody):
-    object: Literal["list"] = Field(
-        description="The object type, always 'list' for a list of fine-tuning jobs.",
-    )
-
-    data: list[OpenAIFineTuningJobResponseBody] = Field(
-        description="A list of fine-tuning job events."
-    )
-
-    has_more: bool = Field(
-        description="Whether there are more results available after this batch."
-    )

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

@@ -1,9 +0,0 @@
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointPathParam
-
-
-class OpenAIRetrieveFineTuningJobPathParam(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/training_format.py

@@ -1,30 +0,0 @@
-from typing import List, Optional
-
-from pydantic import BaseModel, Field, SerializeAsAny
-
-from ..chat_completions.request.request_body import Message, Tool
-
-
-class OpenAIChatModelTrainingFormat(BaseModel):
-    messages: SerializeAsAny[list[Message]] = Field(
-        description="A list of messages comprising the conversation so far."
-    )
-
-    tools: list[Tool] | None = Field(
-        None, description="A list of tools may generate JSON inputs for."
-    )
-
-    parallel_tool_calls: bool | None = Field(
-        True,
-        decription="Whether to enable parallel function calling during tool use.",
-    )
-
-
-class OpenAICompletionsModelTrainingFormat(BaseModel):
-    prompt: str = Field(
-        description="The input prompt for this training example."
-    )
-
-    completion: str = Field(
-        description="The desired completion for this training example."
-    )

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

@@ -1,9 +0,0 @@
-from .image_edit_models import OpenAIImageEditRequestBody
-from .image_models import OpenAIImageRequestBody
-from .image_variation_models import OpenAIImageVariationRequestBody
-
-__all__ = [
-    "OpenAIImageRequestBody",
-    "OpenAIImageEditRequestBody",
-    "OpenAIImageVariationRequestBody",
-]

lionagi/integrations/openai_/api_endpoints/images/image_edit_models.py

@@ -1,69 +0,0 @@
-from enum import Enum
-from typing import IO, Literal, Optional
-
-from pydantic import ConfigDict, Field, field_validator
-
-from ..data_models import OpenAIEndpointRequestBody
-
-
-class Size(str, Enum):
-    S256 = "256x256"
-    S512 = "512x512"
-    S1024 = "1024x1024"
-
-
-class OpenAIImageEditRequestBody(OpenAIEndpointRequestBody):
-    image: str | IO = Field(
-        description="The image to edit. Must be a valid PNG file, less than 4MB, and square.",
-    )
-
-    prompt: str = Field(
-        description="A text description of the desired image(s).",
-        max_length=1000,
-    )
-
-    mask: str | IO | None = Field(
-        None,
-        description="An additional image whose fully transparent areas indicate where image should be edited.",
-    )
-
-    model: Literal["dall-e-2"] | None = Field(
-        "dall-e-2", description="The model to use for image generation."
-    )
-
-    n: int | None = Field(
-        1, description="The number of images to generate.", ge=1, le=10
-    )
-
-    size: Size | None = Field(
-        "1024x1024", description="The size of the generated images."
-    )
-
-    response_format: Literal["url", "b64_json"] | None = Field(
-        "url",
-        description="The format in which the generated images are returned.",
-    )
-
-    user: str | None = Field(
-        None,
-        description="A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse.",
-    )
-
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-
-    @field_validator("image", "mask")
-    @classmethod
-    def get_file_object(cls, value):
-        if isinstance(value, str):
-            value = open(value, "rb")
-        return value
-
-    def __del__(self):
-        # Ensure files are closed when the object is destroyed
-        image_obj = self.__dict__.get("image")
-        if image_obj and not image_obj.closed:
-            image_obj.close()
-
-        mask_obj = self.__dict__.get("mask")
-        if mask_obj and not mask_obj.closed:
-            mask_obj.close()

lionagi/integrations/openai_/api_endpoints/images/image_models.py

@@ -1,56 +0,0 @@
-from enum import Enum
-from typing import Literal, Optional
-
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointRequestBody
-
-
-class Size(str, Enum):
-    S256 = "256x256"
-    S512 = "512x512"
-    S1024 = "1024x1024"
-    S1792_1024 = "1792x1024"
-    S1024_1792 = "1024x1792"
-
-
-class Style(str, Enum):
-    VIVID = "vivid"
-    NATURAL = "natural"
-
-
-class OpenAIImageRequestBody(OpenAIEndpointRequestBody):
-    prompt: str = Field(
-        description="A text description of the desired image(s)."
-    )
-
-    model: str = Field(
-        "dall-e-2", description="The model to use for image generation."
-    )
-
-    n: int | None = Field(
-        1, description="The number of images to generate.", ge=1, le=10
-    )
-
-    quality: Literal["standard", "hd"] | None = Field(
-        "standard",
-        description="The quality of the image that will be generated.",
-    )
-
-    response_format: Literal["url", "b64_json"] | None = Field(
-        "url",
-        description="The format in which the generated images are returned.",
-    )
-
-    size: Size | None = Field(
-        "1024x1024", description="The size of the generated images."
-    )
-
-    style: Style | None = Field(
-        "vivid", description="The style of the generated images."
-    )
-
-    user: str | None = Field(
-        None,
-        description="A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse.",
-    )

lionagi/integrations/openai_/api_endpoints/images/image_variation_models.py

@@ -1,56 +0,0 @@
-from enum import Enum
-from typing import IO, Literal, Optional
-
-from pydantic import ConfigDict, Field, field_validator
-
-from ..data_models import OpenAIEndpointRequestBody
-
-
-class Size(str, Enum):
-    S256 = "256x256"
-    S512 = "512x512"
-    S1024 = "1024x1024"
-
-
-class OpenAIImageVariationRequestBody(OpenAIEndpointRequestBody):
-    image: str | IO = Field(
-        description="The image to use as the basis for the variation(s). "
-        "Must be a valid PNG file, less than 4MB, and square.",
-    )
-
-    model: Literal["dall-e-2"] | None = Field(
-        "dall-e-2", description="The model to use for image generation."
-    )
-
-    n: int | None = Field(
-        1, description="The number of images to generate.", ge=1, le=10
-    )
-
-    response_format: Literal["url", "b64_json"] | None = Field(
-        "url",
-        description="The format in which the generated images are returned.",
-    )
-
-    size: Size | None = Field(
-        "1024x1024", description="The size of the generated images."
-    )
-
-    user: str | None = Field(
-        None,
-        description="A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse.",
-    )
-
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-
-    @field_validator("image")
-    @classmethod
-    def get_file_object(cls, value):
-        if isinstance(value, str):
-            value = open(value, "rb")
-        return value
-
-    def __del__(self):
-        # Ensure files are closed when the object is destroyed
-        image_obj = self.__dict__.get("image")
-        if image_obj and not image_obj.closed:
-            image_obj.close()

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

@@ -1,30 +0,0 @@
-from typing import List, Optional
-
-from pydantic import BaseModel, Field
-
-from ..data_models import OpenAIEndpointRequestBody
-
-
-class Image(BaseModel):
-    b64_json: str | None = Field(
-        None,
-        description="The base64-encoded JSON of the generated image, if response_format is b64_json.",
-    )
-
-    url: str | None = Field(
-        None,
-        description="The URL of the generated image, if response_format is url (default).",
-    )
-
-    revised_prompt: str | None = Field(
-        None,
-        description="The prompt that was used to generate the image, if there was any revision to the prompt.",
-    )
-
-
-class OpenAIImageResponseBody(OpenAIEndpointRequestBody):
-    created: int = Field(
-        description="The Unix timestamp of when the image was created."
-    )
-
-    data: list[Image] = Field(description="The list of generated images.")

lionagi/integrations/openai_/api_endpoints/match_data_model.py

@@ -1,197 +0,0 @@
-def match_data_model(task_name):
-    if task_name == "create_speech":
-        from lionagi.integrations.openai_.api_endpoints.audio.speech_models import (
-            OpenAISpeechRequestBody,
-        )
-
-        return {"request_body": OpenAISpeechRequestBody}
-    elif task_name == "create_transcription":
-        from lionagi.integrations.openai_.api_endpoints.audio.transcription_models import (
-            OpenAITranscriptionRequestBody,
-        )
-
-        return {"request_body": OpenAITranscriptionRequestBody}
-    elif task_name == "create_translation":
-        from lionagi.integrations.openai_.api_endpoints.audio.translation_models import (
-            OpenAITranslationRequestBody,
-        )
-
-        return {"request_body": OpenAITranslationRequestBody}
-    elif task_name == "create_chat_completion":
-        from lionagi.integrations.openai_.api_endpoints.chat_completions.request.request_body import (
-            OpenAIChatCompletionRequestBody,
-        )
-
-        return {"request_body": OpenAIChatCompletionRequestBody}
-    elif task_name == "create_embeddings":
-        from lionagi.integrations.openai_.api_endpoints.embeddings.request_body import (
-            OpenAIEmbeddingRequestBody,
-        )
-
-        return {"request_body": OpenAIEmbeddingRequestBody}
-    elif task_name == "create_fine_tuning_job":
-        from lionagi.integrations.openai_.api_endpoints.fine_tuning.create_jobs import (
-            OpenAICreateFineTuningJobRequestBody,
-        )
-
-        return {"json_data": OpenAICreateFineTuningJobRequestBody}
-    elif task_name == "list_fine_tuning_jobs":
-        from lionagi.integrations.openai_.api_endpoints.fine_tuning.list_fine_tuning_jobs import (
-            OpenAIListFineTuningJobsQueryParam,
-        )
-
-        return {"params": OpenAIListFineTuningJobsQueryParam}
-    elif task_name == "list_fine_tuning_events":
-        from lionagi.integrations.openai_.api_endpoints.fine_tuning.list_fine_tuning_events import (
-            OpenAIListFineTuningEventsPathParam,
-            OpenAIListFineTuningEventsQueryParam,
-        )
-
-        return {
-            "params": OpenAIListFineTuningEventsQueryParam,
-            "path_param": OpenAIListFineTuningEventsPathParam,
-        }
-    elif task_name == "list_fine_tuning_checkpoints":
-        from lionagi.integrations.openai_.api_endpoints.fine_tuning.list_fine_tuning_checkpoints import (
-            OpenAIListFineTuningCheckpointsPathParam,
-            OpenAIListFineTuningCheckpointsQueryParam,
-        )
-
-        return {
-            "params": OpenAIListFineTuningCheckpointsQueryParam,
-            "path_param": OpenAIListFineTuningCheckpointsPathParam,
-        }
-    elif task_name == "retrieve_fine_tuning_job":
-        from lionagi.integrations.openai_.api_endpoints.fine_tuning.retrieve_jobs import (
-            OpenAIRetrieveFineTuningJobPathParam,
-        )
-
-        return {"path_param": OpenAIRetrieveFineTuningJobPathParam}
-    elif task_name == "cancel_fine_tuning":
-        from lionagi.integrations.openai_.api_endpoints.fine_tuning.cancel_jobs import (
-            OpenAICancelFineTuningPathParam,
-        )
-
-        return {"path_param": OpenAICancelFineTuningPathParam}
-    elif task_name == "create_batch":
-        from lionagi.integrations.openai_.api_endpoints.batch.create_batch import (
-            OpenAIBatchRequestBody,
-        )
-
-        return {"json_data": OpenAIBatchRequestBody}
-    elif task_name == "retrieve_batch":
-        from lionagi.integrations.openai_.api_endpoints.batch.retrieve_batch import (
-            OpenAIRetrieveBatchPathParam,
-        )
-
-        return {"path_param": OpenAIRetrieveBatchPathParam}
-    elif task_name == "cancel_batch":
-        from lionagi.integrations.openai_.api_endpoints.batch.cancel_batch import (
-            OpenAICancelBatchPathParam,
-        )
-
-        return {"path_param": OpenAICancelBatchPathParam}
-    elif task_name == "list_batch":
-        from lionagi.integrations.openai_.api_endpoints.batch.list_batch import (
-            OpenAIListBatchQueryParam,
-        )
-
-        return {"params": OpenAIListBatchQueryParam}
-    elif task_name == "upload_file":
-        from lionagi.integrations.openai_.api_endpoints.files.upload_file import (
-            OpenAIUploadFileRequestBody,
-        )
-
-        return {"form_data": OpenAIUploadFileRequestBody}
-    elif task_name == "list_files":
-        from lionagi.integrations.openai_.api_endpoints.files.list_files import (
-            OpenAIListFilesQueryParam,
-        )
-
-        return {"params": OpenAIListFilesQueryParam}
-    elif task_name == "retrieve_file" or task_name == "retrieve_file_content":
-        from lionagi.integrations.openai_.api_endpoints.files.retrieve_file import (
-            OpenAIRetrieveFilePathParam,
-        )
-
-        return {"path_param": OpenAIRetrieveFilePathParam}
-    elif task_name == "delete_file":
-        from lionagi.integrations.openai_.api_endpoints.files.delete_file import (
-            OpenAIDeleteFilePathParam,
-        )
-
-        return {"path_param": OpenAIDeleteFilePathParam}
-    elif task_name == "create_upload":
-        from lionagi.integrations.openai_.api_endpoints.uploads.create_upload import (
-            OpenAIUploadRequestBody,
-        )
-
-        return {"json_data": OpenAIUploadRequestBody}
-    elif task_name == "add_upload_part":
-        from lionagi.integrations.openai_.api_endpoints.uploads.add_upload_part import (
-            OpenAIUploadPartPathParam,
-            OpenAIUploadPartRequestBody,
-        )
-
-        return {
-            "path_param": OpenAIUploadPartPathParam,
-            "form_data": OpenAIUploadPartRequestBody,
-        }
-    elif task_name == "complete_upload":
-        from lionagi.integrations.openai_.api_endpoints.uploads.complete_upload import (
-            OpenAICompleteUploadPathParam,
-            OpenAICompleteUploadRequestBody,
-        )
-
-        return {
-            "path_param": OpenAICompleteUploadPathParam,
-            "json_data": OpenAICompleteUploadRequestBody,
-        }
-    elif task_name == "cancel_upload":
-        from lionagi.integrations.openai_.api_endpoints.uploads.cancel_upload import (
-            OpenAICancelUploadPathParam,
-        )
-
-        return {"path_param": OpenAICancelUploadPathParam}
-    elif task_name == "create_image":
-        from lionagi.integrations.openai_.api_endpoints.images.image_models import (
-            OpenAIImageRequestBody,
-        )
-
-        return {"json_data": OpenAIImageRequestBody}
-    elif task_name == "create_image_edit":
-        from lionagi.integrations.openai_.api_endpoints.images.image_edit_models import (
-            OpenAIImageEditRequestBody,
-        )
-
-        return {"form_data": OpenAIImageEditRequestBody}
-    elif task_name == "create_image_variation":
-        from lionagi.integrations.openai_.api_endpoints.images.image_variation_models import (
-            OpenAIImageVariationRequestBody,
-        )
-
-        return {"form_data": OpenAIImageVariationRequestBody}
-    elif task_name == "list_models":
-        return {}
-    elif task_name == "retrieve_model":
-        from lionagi.integrations.openai_.api_endpoints.models.retrieve_model import (
-            OpenAIRetrieveModelPathParam,
-        )
-
-        return {"path_param": OpenAIRetrieveModelPathParam}
-    elif task_name == "delete_fine_tuned_model":
-        from lionagi.integrations.openai_.api_endpoints.models.delete_fine_tuned_model import (
-            OpenAIDeleteFineTunedModelPathParam,
-        )
-
-        return {"path_param": OpenAIDeleteFineTunedModelPathParam}
-    elif task_name == "create_moderation":
-        from lionagi.integrations.openai_.api_endpoints.moderations.request_body import (
-            OpenAIModerationRequestBody,
-        )
-
-        return {"json_data": OpenAIModerationRequestBody}
-    else:
-        raise ValueError(
-            f"Invalid task: {task_name}. Not supported in the service."
-        )