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

lionagi/integrations/openai_/api_endpoints/match_response.py

@@ -1,336 +0,0 @@
-imported_models = {}
-
-
-def match_response(request_model, response: dict | list):
-    global imported_models
-
-    endpoint = request_model.endpoint
-    method = request_model.method
-
-    # audio
-    if "audio" in endpoint:
-        # create transcription
-        if endpoint == "audio/transcriptions":
-            if len(response) == 1:
-                # transcription object
-                if "OpenAITranscriptionResponseBody" not in imported_models:
-                    from .audio.transcription_models import (
-                        OpenAITranscriptionResponseBody,
-                    )
-
-                    imported_models["OpenAITranscriptionResponseBody"] = (
-                        OpenAITranscriptionResponseBody
-                    )
-                return imported_models["OpenAITranscriptionResponseBody"](
-                    **response
-                )
-            else:
-                # verbose transcription object
-                if (
-                    "OpenAIVerboseTranscriptionResponseBody"
-                    not in imported_models
-                ):
-                    from .audio.transcription_models import (
-                        OpenAIVerboseTranscriptionResponseBody,
-                    )
-
-                    imported_models[
-                        "OpenAIVerboseTranscriptionResponseBody"
-                    ] = OpenAIVerboseTranscriptionResponseBody
-                return imported_models[
-                    "OpenAIVerboseTranscriptionResponseBody"
-                ](**response)
-
-        # create translation
-        if endpoint == "audio/translations":
-            if "OpenAITranslationResponseBody" not in imported_models:
-                from .audio.translation_models import (
-                    OpenAITranslationResponseBody,
-                )
-
-                imported_models["OpenAITranslationResponseBody"] = (
-                    OpenAITranslationResponseBody
-                )
-            return imported_models["OpenAITranslationResponseBody"](**response)
-
-    # Chat
-    elif endpoint == "chat/completions":
-        if isinstance(response, dict):
-            if "OpenAIChatCompletionResponseBody" not in imported_models:
-                from .chat_completions.response.response_body import (
-                    OpenAIChatCompletionResponseBody,
-                )
-
-                imported_models["OpenAIChatCompletionResponseBody"] = (
-                    OpenAIChatCompletionResponseBody
-                )
-            return imported_models["OpenAIChatCompletionResponseBody"](
-                **response
-            )
-        else:
-            # stream list
-            if "OpenAIChatCompletionChunkResponseBody" not in imported_models:
-                from .chat_completions.response.response_body import (
-                    OpenAIChatCompletionChunkResponseBody,
-                )
-
-                imported_models["OpenAIChatCompletionChunkResponseBody"] = (
-                    OpenAIChatCompletionChunkResponseBody
-                )
-            result = []
-            for item in response:
-                result.append(
-                    imported_models["OpenAIChatCompletionChunkResponseBody"](
-                        **item
-                    )
-                )
-            return result
-
-    # Embeddings
-    elif endpoint == "embeddings":
-        if "OpenAIEmbeddingResponseBody" not in imported_models:
-            from .embeddings.response_body import OpenAIEmbeddingResponseBody
-
-            imported_models["OpenAIEmbeddingResponseBody"] = (
-                OpenAIEmbeddingResponseBody
-            )
-        return imported_models["OpenAIEmbeddingResponseBody"](**response)
-
-    # Fine-tuning
-    elif "fine_tuning" in endpoint:
-        # create fine-tuning job
-        if endpoint == "fine_tuning/jobs" and method == "POST":
-            if "OpenAIFineTuningJobResponseBody" not in imported_models:
-                from .fine_tuning.fine_tuning_job_models import (
-                    OpenAIFineTuningJobResponseBody,
-                )
-
-                imported_models["OpenAIFineTuningJobResponseBody"] = (
-                    OpenAIFineTuningJobResponseBody
-                )
-            return imported_models["OpenAIFineTuningJobResponseBody"](
-                **response
-            )
-
-        # list fine-tuning jobs
-        if endpoint == "fine_tuning/jobs" and method == "GET":
-            if "OpenAIListFineTuningJobsResponseBody" not in imported_models:
-                from .fine_tuning.list_fine_tuning_jobs import (
-                    OpenAIListFineTuningJobsResponseBody,
-                )
-
-                imported_models["OpenAIListFineTuningJobsResponseBody"] = (
-                    OpenAIListFineTuningJobsResponseBody
-                )
-            return imported_models["OpenAIListFineTuningJobsResponseBody"](
-                **response
-            )
-
-        # list fine-tuning events
-        if endpoint == "fine_tuning/jobs/{fine_tuning_job_id}/events":
-            if "OpenAIListFineTuningEventsResponseBody" not in imported_models:
-                from .fine_tuning.list_fine_tuning_events import (
-                    OpenAIListFineTuningEventsResponseBody,
-                )
-
-                imported_models["OpenAIListFineTuningEventsResponseBody"] = (
-                    OpenAIListFineTuningEventsResponseBody
-                )
-            return imported_models["OpenAIListFineTuningEventsResponseBody"](
-                **response
-            )
-
-        # list fine-tuning checkpoints
-        if endpoint == "fine_tuning/jobs/{fine_tuning_job_id}/checkpoints":
-            if (
-                "OpenAIListFineTuningCheckpointsResponseBody"
-                not in imported_models
-            ):
-                from .fine_tuning.list_fine_tuning_checkpoints import (
-                    OpenAIListFineTuningCheckpointsResponseBody,
-                )
-
-                imported_models[
-                    "OpenAIListFineTuningCheckpointsResponseBody"
-                ] = OpenAIListFineTuningCheckpointsResponseBody
-            return imported_models[
-                "OpenAIListFineTuningCheckpointsResponseBody"
-            ](**response)
-
-        # retrieve fine-tuning job
-        if endpoint == "fine_tuning/jobs/{fine_tuning_job_id}":
-            if "OpenAIFineTuningJobResponseBody" not in imported_models:
-                from .fine_tuning.fine_tuning_job_models import (
-                    OpenAIFineTuningJobResponseBody,
-                )
-
-                imported_models["OpenAIFineTuningJobResponseBody"] = (
-                    OpenAIFineTuningJobResponseBody
-                )
-            return imported_models["OpenAIFineTuningJobResponseBody"](
-                **response
-            )
-
-        # cancel fine-tuning
-        # TODO: need verification
-        if endpoint == "fine_tuning/jobs/{fine_tuning_job_id}/cancel":
-            if "OpenAIFineTuningJobResponseBody" not in imported_models:
-                from .fine_tuning.fine_tuning_job_models import (
-                    OpenAIFineTuningJobResponseBody,
-                )
-
-                imported_models["OpenAIFineTuningJobResponseBody"] = (
-                    OpenAIFineTuningJobResponseBody
-                )
-            return imported_models["OpenAIFineTuningJobResponseBody"](
-                **response
-            )
-
-    # Batch
-    elif "batches" in endpoint:
-        if endpoint == "batches" and method == "GET":
-            # list batch
-            if "OpenAIListBatchResponseBody" not in imported_models:
-                from .batch.list_batch import OpenAIListBatchResponseBody
-
-                imported_models["OpenAIListBatchResponseBody"] = (
-                    OpenAIListBatchResponseBody
-                )
-            return imported_models["OpenAIListBatchResponseBody"](**response)
-        else:
-            # other batch endpoints
-            if "OpenAIBatchResponseBody" not in imported_models:
-                from .batch.batch_models import OpenAIBatchResponseBody
-
-                imported_models["OpenAIBatchResponseBody"] = (
-                    OpenAIBatchResponseBody
-                )
-            return imported_models["OpenAIBatchResponseBody"](**response)
-
-    # Files
-    elif "files" in endpoint:
-        # upload file
-        if endpoint == "files" and method == "POST":
-            if "OpenAIFileResponseBody" not in imported_models:
-                from .files.file_models import OpenAIFileResponseBody
-
-                imported_models["OpenAIFileResponseBody"] = (
-                    OpenAIFileResponseBody
-                )
-            return imported_models["OpenAIFileResponseBody"](**response)
-
-        # list files
-        if endpoint == "files" and method == "GET":
-            if "OpenAIListFilesResponseBody" not in imported_models:
-                from .files.list_files import OpenAIListFilesResponseBody
-
-                imported_models["OpenAIListFilesResponseBody"] = (
-                    OpenAIListFilesResponseBody
-                )
-            return imported_models["OpenAIListFilesResponseBody"](**response)
-
-        # retrieve file
-        if endpoint == "files/{file_id}" and method == "GET":
-            if "OpenAIFileResponseBody" not in imported_models:
-                from .files.file_models import OpenAIFileResponseBody
-
-                imported_models["OpenAIFileResponseBody"] = (
-                    OpenAIFileResponseBody
-                )
-            return imported_models["OpenAIFileResponseBody"](**response)
-
-        # delete file
-        if endpoint == "files/{file_id}" and method == "DELETE":
-            if "OpenAIDeleteFileResponseBody" not in imported_models:
-                from .files.delete_file import OpenAIDeleteFileResponseBody
-
-                imported_models["OpenAIDeleteFileResponseBody"] = (
-                    OpenAIDeleteFileResponseBody
-                )
-            return imported_models["OpenAIDeleteFileResponseBody"](**response)
-
-        # retrieve file content
-        if endpoint == "files/{file_id}/content":
-            return response
-
-    # Uploads
-    if "uploads" in endpoint:
-        if endpoint == "uploads/{upload_id}/parts":
-            # add upload part
-            if "OpenAIUploadPartResponseBody" not in imported_models:
-                from .uploads.uploads_models import (
-                    OpenAIUploadPartResponseBody,
-                )
-
-                imported_models["OpenAIUploadPartResponseBody"] = (
-                    OpenAIUploadPartResponseBody
-                )
-            return imported_models["OpenAIUploadPartResponseBody"](**response)
-        else:
-            # other upload endpoints
-            if "OpenAIUploadResponseBody" not in imported_models:
-                from .uploads.uploads_models import OpenAIUploadResponseBody
-
-                imported_models["OpenAIUploadResponseBody"] = (
-                    OpenAIUploadResponseBody
-                )
-            return imported_models["OpenAIUploadResponseBody"](**response)
-
-    # Images
-    if "images" in endpoint:
-        if "OpenAIImageResponseBody" not in imported_models:
-            from .images.response_body import OpenAIImageResponseBody
-
-            imported_models["OpenAIImageResponseBody"] = (
-                OpenAIImageResponseBody
-            )
-        return imported_models["OpenAIImageResponseBody"](**response)
-
-    if "models" in endpoint:
-        # list models
-        if endpoint == "models":
-            if "OpenAIListModelResponseBody" not in imported_models:
-                from .models.models_models import OpenAIListModelResponseBody
-
-                imported_models["OpenAIListModelResponseBody"] = (
-                    OpenAIListModelResponseBody
-                )
-            return imported_models["OpenAIListModelResponseBody"](**response)
-
-        # retrieve model
-        if endpoint == "models/{model}" and method == "GET":
-            if "OpenAIModelResponseBody" not in imported_models:
-                from .models.models_models import OpenAIModelResponseBody
-
-                imported_models["OpenAIModelResponseBody"] = (
-                    OpenAIModelResponseBody
-                )
-            return imported_models["OpenAIModelResponseBody"](**response)
-
-        # delete a fine-tuned model
-        if endpoint == "models/{model}" and method == "DELETE":
-            if "OpenAIDeleteFineTunedModelResponseBody" not in imported_models:
-                from .models.delete_fine_tuned_model import (
-                    OpenAIDeleteFineTunedModelResponseBody,
-                )
-
-                imported_models["OpenAIDeleteFineTunedModelResponseBody"] = (
-                    OpenAIDeleteFineTunedModelResponseBody
-                )
-            return imported_models["OpenAIDeleteFineTunedModelResponseBody"](
-                **response
-            )
-
-    # Moderation
-    if endpoint == "moderations":
-        if "OpenAIModerationResponseBody" not in imported_models:
-            from .moderations.response_body import OpenAIModerationResponseBody
-
-            imported_models["OpenAIModerationResponseBody"] = (
-                OpenAIModerationResponseBody
-            )
-        return imported_models["OpenAIModerationResponseBody"](**response)
-
-    raise ValueError(
-        "There is no standard response model for the provided request and response"
-    )

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

@@ -1,7 +0,0 @@
-from .delete_fine_tuned_model import OpenAIDeleteFineTunedModelPathParam
-from .retrieve_model import OpenAIRetrieveModelPathParam
-
-__all__ = [
-    "OpenAIRetrieveModelPathParam",
-    "OpenAIDeleteFineTunedModelPathParam",
-]

lionagi/integrations/openai_/api_endpoints/models/delete_fine_tuned_model.py

@@ -1,17 +0,0 @@
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointPathParam, OpenAIEndpointResponseBody
-
-
-class OpenAIDeleteFineTunedModelPathParam(OpenAIEndpointPathParam):
-    model: str = Field(description="The model to delete.")
-
-
-class OpenAIDeleteFineTunedModelResponseBody(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 model.")
-
-    deleted: bool = Field(description="The deletion status")

lionagi/integrations/openai_/api_endpoints/models/models_models.py

@@ -1,31 +0,0 @@
-from typing import List, Literal
-
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointResponseBody
-
-
-class OpenAIModelResponseBody(OpenAIEndpointResponseBody):
-    id: str = Field(
-        description="The model identifier, which can be referenced in the API endpoints.",
-    )
-
-    created: int = Field(
-        description="The Unix timestamp (in seconds) when the model was created."
-    )
-
-    object: Literal["model"] = Field(
-        description="The object type, which is always 'model'."
-    )
-
-    owned_by: str = Field(description="The organization that owns the model.")
-
-
-class OpenAIListModelResponseBody(OpenAIEndpointResponseBody):
-    object: Literal["list"] = Field(
-        description="The object type, which is always 'list'."
-    )
-
-    data: list[OpenAIModelResponseBody] = Field(
-        description="The list of model objects."
-    )

lionagi/integrations/openai_/api_endpoints/models/retrieve_model.py

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

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

@@ -1,3 +0,0 @@
-from .request_body import OpenAIModerationRequestBody
-
-__all__ = ["OpenAIModerationRequestBody"]

lionagi/integrations/openai_/api_endpoints/moderations/request_body.py

@@ -1,20 +0,0 @@
-from enum import Enum
-from typing import Optional
-
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointRequestBody
-
-
-class Model(str, Enum):
-    LATEST = "text-moderation-latest"
-    STABLE = "text-moderation-stable"
-
-
-class OpenAIModerationRequestBody(OpenAIEndpointRequestBody):
-    input: list | str = Field(description="The input text to classify")
-
-    model: Model | None = Field(
-        "text-moderation-latest",
-        description="Two content moderations models are available: text-moderation-stable and text-moderation-latest.",
-    )

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

@@ -1,139 +0,0 @@
-from typing import List
-
-from pydantic import BaseModel, Field
-
-from ..data_models import OpenAIEndpointResponseBody
-
-
-class Categories(BaseModel):
-    hate: bool = Field(
-        description="Content that expresses, incites, or promotes hate based on race, gender, ethnicity, "
-        "religion, nationality, sexual orientation, disability status, or caste. Hateful "
-        "content aimed at non-protected groups (e.g., chess players) is harassment."
-    )
-
-    hate_threatening: bool = Field(
-        alias="hate/threatening",
-        description="Hateful content that also includes violence or serious harm towards the targeted "
-        "group based on race, gender, ethnicity, religion, nationality, sexual orientation, "
-        "disability status, or caste.",
-    )
-
-    harassment: bool = Field(
-        description="Content that expresses, incites, or promotes harassing language towards any target."
-    )
-
-    harassment_threatening: bool = Field(
-        alias="harassment/threatening",
-        description="Harassment content that also includes violence or serious harm towards any target.",
-    )
-
-    self_harm: bool = Field(
-        alias="self-harm",
-        description="Content that promotes, encourages, or depicts acts of self-harm, such as suicide, "
-        "cutting, and eating disorders.",
-    )
-
-    self_harm_intent: bool = Field(
-        alias="self-harm/intent",
-        description="Content where the speaker expresses that they are engaging or intend to engage in "
-        "acts of self-harm, such as suicide, cutting, and eating disorders.",
-    )
-
-    self_harm_instructions: bool = Field(
-        alias="self-harm/instructions",
-        description="Content that encourages performing acts of self-harm, such as suicide, cutting, "
-        "and eating disorders, or that gives instructions or advice on how to commit such acts.",
-    )
-
-    sexual: bool = Field(
-        description="Content meant to arouse sexual excitement, such as the description of sexual activity, "
-        "or that promotes sexual services (excluding sex education and wellness)."
-    )
-
-    sexual_minors: bool = Field(
-        alias="sexual/minors",
-        description="Sexual content that includes an individual who is under 18 years old.",
-    )
-
-    violence: bool = Field(
-        description="Content that depicts death, violence, or physical injury."
-    )
-
-    violence_graphic: bool = Field(
-        alias="violence/graphic",
-        description="Content that depicts death, violence, or physical injury in graphic detail.",
-    )
-
-
-class CategoryScores(BaseModel):
-    hate: float = Field(description="The score for the category 'hate'.")
-
-    hate_threatening: float = Field(
-        alias="hate/threatening",
-        description="The score for the category 'hate/threatening'.",
-    )
-
-    harassment: float = Field(
-        description="The score for the category 'harassment'."
-    )
-
-    harassment_threatening: float = Field(
-        alias="harassment/threatening",
-        description="The score for the category 'harassment/threatening'.",
-    )
-
-    self_harm: float = Field(
-        alias="self-harm",
-        description="The score for the category 'self-harm'.",
-    )
-
-    self_harm_intent: float = Field(
-        alias="self-harm/intent",
-        description="The score for the category 'self-harm/intent'.",
-    )
-
-    self_harm_instructions: float = Field(
-        alias="self-harm/instructions",
-        description="The score for the category 'self-harm/instructions'.",
-    )
-
-    sexual: float = Field(description="The score for the category 'sexual'.")
-
-    sexual_minors: float = Field(
-        alias="sexual/minors",
-        description="The score for the category 'sexual/minors'.",
-    )
-
-    violence: float = Field(
-        description="The score for the category 'violence'."
-    )
-
-    violence_graphic: float = Field(
-        alias="violence/graphic",
-        description="The score for the category 'violence/graphic'.",
-    )
-
-
-class Result(BaseModel):
-    flagged: bool = Field(
-        description="Whether the content violates OpenAI's usage policies."
-    )
-
-    categories: Categories = Field(
-        description="A dictionary of per-category binary content policy violation flags.",
-    )
-
-    category_scores: CategoryScores = Field(
-        description="A dictionary of per-category raw scores output by the model."
-    )
-
-
-class OpenAIModerationResponseBody(OpenAIEndpointResponseBody):
-    id: str = Field(
-        description="The unique identifier for the moderation request."
-    )
-    model: str = Field(
-        description="The model used to generate the moderation results."
-    )
-    results: list[Result] = Field(description="A list of moderation objects.")

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

@@ -1,19 +0,0 @@
-from .add_upload_part import (
-    OpenAIUploadPartPathParam,
-    OpenAIUploadPartRequestBody,
-)
-from .cancel_upload import OpenAICancelUploadPathParam
-from .complete_upload import (
-    OpenAICompleteUploadPathParam,
-    OpenAICompleteUploadRequestBody,
-)
-from .create_upload import OpenAIUploadRequestBody
-
-__all__ = [
-    "OpenAIUploadRequestBody",
-    "OpenAIUploadPartPathParam",
-    "OpenAIUploadPartRequestBody",
-    "OpenAICompleteUploadPathParam",
-    "OpenAICompleteUploadRequestBody",
-    "OpenAICancelUploadPathParam",
-]

lionagi/integrations/openai_/api_endpoints/uploads/add_upload_part.py

@@ -1,11 +0,0 @@
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointPathParam, OpenAIEndpointRequestBody
-
-
-class OpenAIUploadPartPathParam(OpenAIEndpointPathParam):
-    upload_id: str = Field(description="The ID of the Upload.")
-
-
-class OpenAIUploadPartRequestBody(OpenAIEndpointRequestBody):
-    data: bytes = Field(description="The chunk of bytes for this Part.")

lionagi/integrations/openai_/api_endpoints/uploads/cancel_upload.py

@@ -1,7 +0,0 @@
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointPathParam
-
-
-class OpenAICancelUploadPathParam(OpenAIEndpointPathParam):
-    upload_id: str = Field(description="The ID of the Upload.")

lionagi/integrations/openai_/api_endpoints/uploads/complete_upload.py

@@ -1,18 +0,0 @@
-from typing import List, Optional
-
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointPathParam, OpenAIEndpointRequestBody
-
-
-class OpenAICompleteUploadPathParam(OpenAIEndpointPathParam):
-    upload_id: str = Field(description="The ID of the Upload.")
-
-
-class OpenAICompleteUploadRequestBody(OpenAIEndpointRequestBody):
-    part_ids: list[str] = Field(description="The ordered list of Part IDs.")
-
-    md5: str | None = Field(
-        description="The optional md5 checksum for the file contents to verify "
-        "if the bytes uploaded matches what you expect."
-    )

lionagi/integrations/openai_/api_endpoints/uploads/create_upload.py

@@ -1,17 +0,0 @@
-from pydantic import Field
-
-from ..data_models import OpenAIEndpointRequestBody
-
-
-class OpenAIUploadRequestBody(OpenAIEndpointRequestBody):
-    filename: str = Field(description="The name of the file to upload.")
-
-    purpose: str = Field(
-        description="The intended purpose of the uploaded file."
-    )
-
-    bytes: int = Field(
-        description="The intended number of bytes to be uploaded."
-    )
-
-    mime_type: str = Field(description="The MIME type of the file.")