lionagi 0.5.4__py3-none-any.whl → 0.6.0__py3-none-any.whl

lionagi/integrations/ollama_/api_endpoints/match_response.py

@@ -1,190 +0,0 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-imported_models = {}
-
-
-def match_response(request_model, response: dict | list):
-    global imported_models
-
-    endpoint = request_model.endpoint
-
-    if endpoint == "generate":
-        if isinstance(response, list):
-            if "OllamaStreamCompletionResponseBody" not in imported_models:
-                from .completion.response_body import (
-                    OllamaStreamCompletionResponseBody,
-                )
-
-                imported_models["OllamaStreamCompletionResponseBody"] = (
-                    OllamaStreamCompletionResponseBody
-                )
-            if "OllamaCompletionResponseBody" not in imported_models:
-                from .completion.response_body import (
-                    OllamaCompletionResponseBody,
-                )
-
-                imported_models["OllamaCompletionResponseBody"] = (
-                    OllamaCompletionResponseBody
-                )
-
-            result = []
-            for item in response[:-1]:
-                result.append(
-                    imported_models["OllamaStreamCompletionResponseBody"](
-                        **item
-                    )
-                )
-
-            result.append(
-                imported_models["OllamaCompletionResponseBody"](**response[-1])
-            )
-            return result
-        else:
-            if "OllamaCompletionResponseBody" not in imported_models:
-                from .completion.response_body import (
-                    OllamaCompletionResponseBody,
-                )
-
-                imported_models["OllamaCompletionResponseBody"] = (
-                    OllamaCompletionResponseBody
-                )
-            return imported_models["OllamaCompletionResponseBody"](**response)
-
-    elif endpoint == "chat":
-        if isinstance(response, list):
-            if "OllamaStreamChatCompletionResponseBody" not in imported_models:
-                from .chat_completion.response_body import (
-                    OllamaStreamChatCompletionResponseBody,
-                )
-
-                imported_models["OllamaStreamChatCompletionResponseBody"] = (
-                    OllamaStreamChatCompletionResponseBody
-                )
-            if "OllamaChatCompletionResponseBody" not in imported_models:
-                from .chat_completion.response_body import (
-                    OllamaChatCompletionResponseBody,
-                )
-
-                imported_models["OllamaChatCompletionResponseBody"] = (
-                    OllamaChatCompletionResponseBody
-                )
-
-            result = []
-            for item in response[:-1]:
-                result.append(
-                    imported_models["OllamaStreamChatCompletionResponseBody"](
-                        **item
-                    )
-                )
-
-            result.append(
-                imported_models["OllamaChatCompletionResponseBody"](
-                    **response[-1]
-                )
-            )
-            return result
-        else:
-            if "OllamaChatCompletionResponseBody" not in imported_models:
-                from .chat_completion.response_body import (
-                    OllamaChatCompletionResponseBody,
-                )
-
-                imported_models["OllamaChatCompletionResponseBody"] = (
-                    OllamaChatCompletionResponseBody
-                )
-            return imported_models["OllamaChatCompletionResponseBody"](
-                **response
-            )
-
-    elif endpoint == "embed":
-        if "OllamaEmbeddingResponseBody" not in imported_models:
-            from .embedding.response_body import OllamaEmbeddingResponseBody
-
-            imported_models["OllamaEmbeddingResponseBody"] = (
-                OllamaEmbeddingResponseBody
-            )
-        return imported_models["OllamaEmbeddingResponseBody"](**response)
-
-    elif endpoint == "create":
-        if "OllamaCreateModelResponseBody" not in imported_models:
-            from .model.create_model import OllamaCreateModelResponseBody
-
-            imported_models["OllamaCreateModelResponseBody"] = (
-                OllamaCreateModelResponseBody
-            )
-        if isinstance(response, list):
-            return [
-                imported_models["OllamaCreateModelResponseBody"](**res)
-                for res in response
-            ]
-        else:  # dict
-            return imported_models["OllamaCreateModelResponseBody"](**response)
-
-    elif endpoint == "tags":
-        if "OllamaListLocalModelsResponseBody" not in imported_models:
-            from .model.list_model import OllamaListLocalModelsResponseBody
-
-            imported_models["OllamaListLocalModelsResponseBody"] = (
-                OllamaListLocalModelsResponseBody
-            )
-        return imported_models["OllamaListLocalModelsResponseBody"](**response)
-
-    elif endpoint == "show":
-        if "OllamaShowModelResponseBody" not in imported_models:
-            from .model.show_model import OllamaShowModelResponseBody
-
-            imported_models["OllamaShowModelResponseBody"] = (
-                OllamaShowModelResponseBody
-            )
-        return imported_models["OllamaShowModelResponseBody"](**response)
-
-    elif endpoint == "pull":
-        if "OllamaPullModelResponseBody" not in imported_models:
-            from .model.pull_model import OllamaPullModelResponseBody
-
-            imported_models["OllamaPullModelResponseBody"] = (
-                OllamaPullModelResponseBody
-            )
-        if isinstance(response, list):
-            return [
-                imported_models["OllamaPullModelResponseBody"](**res)
-                for res in response
-            ]
-        else:  # dict
-            return imported_models["OllamaPullModelResponseBody"](**response)
-
-    elif endpoint == "push":
-        if "OllamaPushModelResponseBody" not in imported_models:
-            from .model.push_model import OllamaPushModelResponseBody
-
-            imported_models["OllamaPushModelResponseBody"] = (
-                OllamaPushModelResponseBody
-            )
-        if isinstance(response, list):
-            return [
-                imported_models["OllamaPushModelResponseBody"](**res)
-                for res in response
-            ]
-        else:  # dict
-            return imported_models["OllamaPushModelResponseBody"](**response)
-
-    elif endpoint == "ps":
-        if "OllamaListRunningModelsResponseBody" not in imported_models:
-            from .model.list_model import OllamaListRunningModelsResponseBody
-
-            imported_models["OllamaListRunningModelsResponseBody"] = (
-                OllamaListRunningModelsResponseBody
-            )
-        return imported_models["OllamaListRunningModelsResponseBody"](
-            **response
-        )
-
-    elif not response:
-        return
-
-    else:
-        raise ValueError(
-            "There is no standard response model for the provided request and response"
-        )

lionagi/integrations/ollama_/api_endpoints/model/copy_model.py

@@ -1,13 +0,0 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-from pydantic import Field
-
-from ..data_models import OllamaEndpointRequestBody
-
-
-class OllamaCopyModelRequestBody(OllamaEndpointRequestBody):
-    source: str = Field(description="Name of the source existing model")
-
-    destination: str = Field(description="Name of the copy model")

lionagi/integrations/ollama_/api_endpoints/model/create_model.py

@@ -1,28 +0,0 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-
-from pydantic import Field
-
-from ..data_models import OllamaEndpointRequestBody, OllamaEndpointResponseBody
-
-
-class OllamaCreateModelRequestBody(OllamaEndpointRequestBody):
-    name: str = Field(description="name of the model to create")
-
-    modelfile: str | None = Field(
-        None, description="Contents of the Modelfile"
-    )
-
-    stream: bool | None = Field(
-        True,
-        description="if 'false' the response will be returned as a single response object, "
-        "rather than a stream of objects",
-    )
-
-    path: str | None = Field(None, description="path to the Modelfile")
-
-
-class OllamaCreateModelResponseBody(OllamaEndpointResponseBody):
-    status: str = Field(None)

lionagi/integrations/ollama_/api_endpoints/model/delete_model.py

@@ -1,11 +0,0 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-from pydantic import Field
-
-from ..data_models import OllamaEndpointRequestBody
-
-
-class OllamaDeleteModelRequestBody(OllamaEndpointRequestBody):
-    name: str = Field(description="Name of the model to delete")

lionagi/integrations/ollama_/api_endpoints/model/list_model.py

@@ -1,60 +0,0 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-
-from pydantic import BaseModel, Field
-
-from ..data_models import OllamaEndpointResponseBody
-
-
-class Detail(BaseModel):
-    parent_model: str = Field(None)
-
-    format: str = Field(None)
-
-    family: str = Field(None)
-
-    families: list[str] | None = Field(None)
-
-    parameter_size: str = Field(None)
-
-    quantization_level: str = Field(None)
-
-
-class LocalModel(BaseModel):
-    name: str = Field(None)
-
-    model: str = Field(None)
-
-    modified_at: str = Field(None)
-
-    size: int = Field(None)
-
-    digest: str = Field(None)
-
-    details: Detail = Field(None)
-
-
-class OllamaListLocalModelsResponseBody(OllamaEndpointResponseBody):
-    models: list[LocalModel] = Field(None)
-
-
-class RunningModel(BaseModel):
-    name: str = Field(None)
-
-    model: str = Field(None)
-
-    size: int = Field(None)
-
-    digest: str = Field(None)
-
-    details: Detail = Field(None)
-
-    expires_at: str = Field(None)
-
-    size_vram: int = Field(None)
-
-
-class OllamaListRunningModelsResponseBody(OllamaEndpointResponseBody):
-    models: list[RunningModel] = Field(None)

lionagi/integrations/ollama_/api_endpoints/model/pull_model.py

@@ -1,34 +0,0 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-
-from pydantic import Field
-
-from ..data_models import OllamaEndpointRequestBody, OllamaEndpointResponseBody
-
-
-class OllamaPullModelRequestBody(OllamaEndpointRequestBody):
-    name: str = Field(description="Name of the model to pull")
-
-    insecure: bool = Field(
-        False,
-        description="Allow insecure connections to the library. "
-        "Only use this if you are pulling from your own library during development.",
-    )
-
-    stream: bool = Field(
-        True,
-        description="If 'false' the response will be returned as a single response object, "
-        "rather than a stream of objects",
-    )
-
-
-class OllamaPullModelResponseBody(OllamaEndpointResponseBody):
-    status: str = Field(None)
-
-    digest: str = Field(None)
-
-    total: int = Field(None)
-
-    completed: int = Field(None)

lionagi/integrations/ollama_/api_endpoints/model/push_model.py

@@ -1,35 +0,0 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-from typing import Any
-
-from pydantic import Field
-
-from ..data_models import OllamaEndpointRequestBody, OllamaEndpointResponseBody
-
-
-class OllamaPushModelRequestBody(OllamaEndpointRequestBody):
-    name: str = Field(
-        description="Name of the model to push in the form of '<namespace>/<model>:<tag>'"
-    )
-
-    insecure: Any = Field(
-        None,
-        description="Allow insecure connections to the library. "
-        "Only use this if you are pushing to your library during development.",
-    )
-
-    stream: bool = Field(
-        True,
-        description="If 'false' the response will be returned as a single response object, "
-        "rather than a stream of objects",
-    )
-
-
-class OllamaPushModelResponseBody(OllamaEndpointResponseBody):
-    status: str = Field(None)
-
-    digest: str = Field(None)
-
-    total: int = Field(None)

lionagi/integrations/ollama_/api_endpoints/model/show_model.py

@@ -1,36 +0,0 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-
-from pydantic import Field
-
-from ..data_models import OllamaEndpointRequestBody, OllamaEndpointResponseBody
-from .list_model import Detail
-
-
-class OllamaShowModelRequestBody(OllamaEndpointRequestBody):
-    name: str = Field(description="Name of the model to show")
-
-    verbose: bool | None = Field(
-        None,
-        description="If set to true, returns full data for verbose response fields",
-    )
-
-
-class OllamaShowModelResponseBody(OllamaEndpointResponseBody):
-    license: str = Field(None)
-
-    modelfile: str = Field(None)
-
-    parameters: str = Field(None)
-
-    template: str = Field(None)
-
-    details: Detail = Field(None)
-
-    model_info: dict = Field(None)
-
-    projector_info: dict = Field(None)
-
-    modified_at: str = Field(None)

lionagi/integrations/ollama_/api_endpoints/option_models.py

@@ -1,68 +0,0 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-from pydantic import BaseModel, Field
-
-
-class Option(BaseModel):
-    mirostat: int = Field(
-        0, description="Enable Mirostat sampling for controlling perplexity."
-    )
-
-    mirostat_eta: float = Field(
-        0.1,
-        description="Influences how quickly the algorithm responds to feedback from the generated text. ",
-    )
-
-    mirostat_tau: float = Field(
-        5.0,
-        description="Controls the balance between coherence and diversity of the output.",
-    )
-
-    num_ctx: int = Field(
-        2048,
-        description="Sets the size of the context window used to generate the next token.",
-    )
-
-    repeat_last_n: int = Field(
-        64,
-        description="Sets how far back for the model to look back to prevent repetition.",
-    )
-
-    repeat_penalty: float = Field(
-        1.1, description="Sets how strongly to penalize repetitions."
-    )
-
-    temperature: float = Field(
-        0.8, description="The temperature of the model."
-    )
-
-    seed: int = Field(
-        0, description="Sets the random number seed to use for generation."
-    )
-
-    stop: str = Field(None, description="Sets the stop sequences to use.")
-
-    tfs_z: float = Field(
-        1,
-        description="Tail free sampling is used to reduce the impact of less probable tokens from the output.",
-    )
-
-    num_predict: int = Field(
-        128,
-        description="Maximum number of tokens to predict when generating text.",
-    )
-
-    top_k: int = Field(
-        40, description="Reduces the probability of generating nonsense."
-    )
-
-    top_p: float = Field(0.9, description="Works together with top-k.")
-
-    min_p: float = Field(
-        0.0,
-        description="Alternative to the top_p, and aims to ensure a balance of quality and variety. "
-        "The parameter p represents the minimum probability for a token to be considered, "
-        "relative to the probability of the most likely token. ",
-    )