draive 0.9.1__tar.gz

draive-0.9.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Miquido
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

draive-0.9.1/PKG-INFO

@@ -0,0 +1,76 @@
+Metadata-Version: 2.1
+Name: draive
+Version: 0.9.1
+Maintainer-email: Kacper Kaliński <kacper.kalinski@miquido.com>
+License: MIT License
+        
+        Copyright (c) 2024 Miquido
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+Project-URL: Homepage, https://miquido.com
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Topic :: Software Development
+Classifier: Typing :: Typed
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: openai~=1.14
+Requires-Dist: mistralai~=0.1.3
+Requires-Dist: numpy~=1.26
+Requires-Dist: tiktoken~=0.6.0
+Requires-Dist: pydantic~=2.6
+Requires-Dist: httpx~=0.25.0
+Provides-Extra: dev
+Requires-Dist: ruff~=0.2.0; extra == "dev"
+Requires-Dist: pyright~=1.1.0; extra == "dev"
+Requires-Dist: bandit~=1.7.0; extra == "dev"
+Requires-Dist: pytest~=7.4.0; extra == "dev"
+Requires-Dist: pytest-cov~=4.1.0; extra == "dev"
+Requires-Dist: pytest-asyncio~=0.23.0; extra == "dev"
+Requires-Dist: python-dotenv~=1.0.0; extra == "dev"
+
+# draive
+
+Framework for building AI oriented applications.
+
+## License
+
+MIT License
+
+Copyright (c) 2024 Miquido
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

draive-0.9.1/README.md

@@ -0,0 +1,27 @@
+# draive
+
+Framework for building AI oriented applications.
+
+## License
+
+MIT License
+
+Copyright (c) 2024 Miquido
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

draive-0.9.1/pyproject.toml

@@ -0,0 +1,70 @@
+[build-system]
+requires = ["setuptools"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "draive"
+version = "0.9.1"
+readme = "README.md"
+maintainers = [
+  {name = "Kacper Kaliński", email = "kacper.kalinski@miquido.com"}
+]
+requires-python = ">=3.11"
+classifiers = [
+    "Intended Audience :: Developers",
+    "Programming Language :: Python",
+    "License :: OSI Approved :: MIT License",
+    "Topic :: Software Development",
+    "Typing :: Typed",
+]
+license = {file = "LICENSE"}
+dependencies = [
+    "openai~=1.14",
+    "mistralai~=0.1.3",
+    "numpy~=1.26",
+    "tiktoken~=0.6.0",
+    "pydantic~=2.6",
+    "httpx~=0.25.0",
+]
+
+[project.urls]
+Homepage = "https://miquido.com"
+
+[project.optional-dependencies]
+dev = [
+    "ruff~=0.2.0",
+    "pyright~=1.1.0",
+    "bandit~=1.7.0",
+    "pytest~=7.4.0",
+    "pytest-cov~=4.1.0",
+    "pytest-asyncio~=0.23.0",
+    "python-dotenv~=1.0.0",
+]
+
+[tool.ruff]
+target-version = "py311"
+line-length = 100
+extend-exclude = [".venv", ".git", ".cache"]
+lint.select = ["E", "F", "A", "I", "B", "PL", "W", "C", "RUF", "UP"]
+lint.ignore = []
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["F401", "E402"]
+"./tests/*.py" = ["PLR2004"]
+
+[tool.ruff.lint.flake8-bugbear]
+extend-immutable-calls = ["Argument", "draive.Argument"]
+
+[tool.pyright]
+pythonVersion = "3.11"
+venvPath = "./.venv"
+include = [
+  "./src",
+]
+exclude = [
+  "**/__pycache__",
+]
+ignore = []
+reportMissingImports = true
+reportMissingTypeStubs = true
+typeCheckingMode = "strict"

draive-0.9.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

draive-0.9.1/src/draive/__init__.py

@@ -0,0 +1,186 @@
+from draive.conversation import (
+    Conversation,
+    ConversationCompletion,
+    ConversationCompletionStream,
+    ConversationMessage,
+    ConversationMessageContent,
+    conversation_completion,
+    lmm_conversation_completion,
+)
+from draive.embedding import Embedded, Embedder, Embedding, embed_text
+from draive.generation import (
+    ImageGeneration,
+    ImageGenerator,
+    ModelGeneration,
+    ModelGenerator,
+    TextGeneration,
+    TextGenerator,
+    generate_image,
+    generate_model,
+    generate_text,
+)
+from draive.helpers import (
+    getenv_bool,
+    getenv_float,
+    getenv_int,
+    getenv_str,
+    load_env,
+    setup_logging,
+    split_sequence,
+)
+from draive.lmm import (
+    LMM,
+    LMMCompletion,
+    LMMCompletionContent,
+    LMMCompletionMessage,
+    LMMCompletionStream,
+    LMMCompletionStreamingUpdate,
+    lmm_completion,
+)
+from draive.mistral import (
+    MistralChatConfig,
+    MistralClient,
+    MistralException,
+    mistral_lmm_completion,
+)
+from draive.openai import (
+    OpenAIChatConfig,
+    OpenAIClient,
+    OpenAIEmbeddingConfig,
+    OpenAIException,
+    OpenAIImageGenerationConfig,
+    openai_embed_text,
+    openai_generate_image,
+    openai_lmm_completion,
+    openai_tokenize_text,
+)
+from draive.scope import (
+    DependenciesScope,
+    MetricsScope,
+    ScopeDependency,
+    ScopeMetric,
+    StateScope,
+    TokenUsage,
+    ctx,
+)
+from draive.similarity import mmr_similarity, similarity
+from draive.splitters import split_text
+from draive.tokenization import TextTokenizer, Tokenization, count_text_tokens, tokenize_text
+from draive.tools import (
+    Tool,
+    Toolbox,
+    ToolCallContext,
+    ToolCallStatus,
+    ToolCallUpdate,
+    ToolException,
+    ToolsUpdatesContext,
+    tool,
+)
+from draive.types import (
+    MISSING,
+    Argument,
+    Field,
+    ImageBase64Content,
+    ImageContent,
+    ImageURLContent,
+    Memory,
+    MissingValue,
+    Model,
+    MultimodalContent,
+    ReadOnlyMemory,
+    State,
+    UpdateSend,
+)
+from draive.utils import allowing_early_exit, auto_retry, cache, with_early_exit
+
+__all__ = [
+    "Conversation",
+    "ImageContent",
+    "ImageBase64Content",
+    "ImageURLContent",
+    "UpdateSend",
+    "Embedded",
+    "Embedder",
+    "Embedding",
+    "Model",
+    "Memory",
+    "ModelGeneration",
+    "ModelGenerator",
+    "ReadOnlyMemory",
+    "DependenciesScope",
+    "ScopeDependency",
+    "ScopeMetric",
+    "MetricsScope",
+    "StateScope",
+    "State",
+    "Field",
+    "TextGeneration",
+    "TextGenerator",
+    "ImageGeneration",
+    "ImageGenerator",
+    "generate_image",
+    "TextTokenizer",
+    "Tokenization",
+    "TokenUsage",
+    "auto_retry",
+    "cache",
+    "conversation_completion",
+    "tokenize_text",
+    "count_text_tokens",
+    "ctx",
+    "embed_text",
+    "generate_model",
+    "generate_text",
+    "load_env",
+    "getenv_bool",
+    "getenv_float",
+    "getenv_int",
+    "getenv_str",
+    "mmr_similarity",
+    "similarity",
+    "split_sequence",
+    "split_text",
+    "tool",
+    "Argument",
+    "allowing_early_exit",
+    "with_early_exit",
+    "setup_logging",
+    "MissingValue",
+    "MISSING",
+    "MultimodalContent",
+    "Tool",
+    "ToolException",
+    "Toolbox",
+    "ToolCallContext",
+    "ToolCallStatus",
+    "ToolCallUpdate",
+    "ToolException",
+    "ToolsUpdatesContext",
+    "LMM",
+    "LMMCompletion",
+    "LMMCompletionContent",
+    "LMMCompletionMessage",
+    "LMMCompletionStreamingUpdate",
+    "LMMCompletionStream",
+    "lmm_completion",
+    "ConversationMessage",
+    "ConversationMessageContent",
+    "Conversation",
+    "ConversationCompletionStream",
+    "ConversationCompletion",
+    "conversation_completion",
+    "lmm_conversation_completion",
+    "OpenAIException",
+    "OpenAIChatConfig",
+    "OpenAIClient",
+    "OpenAIEmbeddingConfig",
+    "OpenAIImageGenerationConfig",
+    "openai_tokenize_text",
+    "openai_embed_text",
+    "openai_lmm_completion",
+    "openai_generate_image",
+    "MistralException",
+    "MistralClient",
+    "MistralChatConfig",
+    "mistral_lmm_completion",
+]

draive-0.9.1/src/draive/conversation/__init__.py

@@ -0,0 +1,18 @@
+from draive.conversation.call import conversation_completion
+from draive.conversation.completion import ConversationCompletion, ConversationCompletionStream
+from draive.conversation.lmm import lmm_conversation_completion
+from draive.conversation.message import (
+    ConversationMessage,
+    ConversationMessageContent,
+)
+from draive.conversation.state import Conversation
+
+__all__ = [
+    "ConversationMessage",
+    "ConversationMessageContent",
+    "Conversation",
+    "ConversationCompletionStream",
+    "ConversationCompletion",
+    "conversation_completion",
+    "lmm_conversation_completion",
+]

draive-0.9.1/src/draive/conversation/call.py

@@ -0,0 +1,87 @@
+from typing import Literal, overload
+
+from draive.conversation.completion import ConversationCompletionStream
+from draive.conversation.message import (
+    ConversationMessage,
+    ConversationMessageContent,
+    ConversationStreamingUpdate,
+)
+from draive.conversation.state import Conversation
+from draive.scope import ctx
+from draive.tools import Toolbox
+from draive.types import Memory, UpdateSend
+
+__all__ = [
+    "conversation_completion",
+]
+
+
+@overload
+async def conversation_completion(
+    *,
+    instruction: str,
+    input: ConversationMessage | ConversationMessageContent,  # noqa: A002
+    memory: Memory[ConversationMessage] | None = None,
+    tools: Toolbox | None = None,
+    stream: Literal[True],
+) -> ConversationCompletionStream:
+    ...
+
+
+@overload
+async def conversation_completion(
+    *,
+    instruction: str,
+    input: ConversationMessage | ConversationMessageContent,  # noqa: A002
+    memory: Memory[ConversationMessage] | None = None,
+    tools: Toolbox | None = None,
+    stream: UpdateSend[ConversationStreamingUpdate],
+) -> ConversationMessage:
+    ...
+
+
+@overload
+async def conversation_completion(
+    *,
+    instruction: str,
+    input: ConversationMessage | ConversationMessageContent,  # noqa: A002
+    memory: Memory[ConversationMessage] | None = None,
+    tools: Toolbox | None = None,
+) -> ConversationMessage:
+    ...
+
+
+async def conversation_completion(
+    *,
+    instruction: str,
+    input: ConversationMessage | ConversationMessageContent,  # noqa: A002
+    memory: Memory[ConversationMessage] | None = None,
+    tools: Toolbox | None = None,
+    stream: UpdateSend[ConversationStreamingUpdate] | bool = False,
+) -> ConversationCompletionStream | ConversationMessage:
+    conversation: Conversation = ctx.state(Conversation)
+
+    match stream:
+        case False:
+            return await conversation.completion(
+                instruction=instruction,
+                input=input,
+                memory=memory or conversation.memory,
+                tools=tools or conversation.tools,
+            )
+        case True:
+            return await conversation.completion(
+                instruction=instruction,
+                input=input,
+                memory=memory or conversation.memory,
+                tools=tools or conversation.tools,
+                stream=True,
+            )
+        case progress:
+            return await conversation.completion(
+                instruction=instruction,
+                input=input,
+                memory=memory or conversation.memory,
+                tools=tools or conversation.tools,
+                stream=progress,
+            )

draive-0.9.1/src/draive/conversation/completion.py

@@ -0,0 +1,67 @@
+from typing import Literal, Protocol, overload, runtime_checkable
+
+from draive.conversation.message import (
+    ConversationMessage,
+    ConversationMessageContent,
+    ConversationStreamingUpdate,
+)
+from draive.lmm import LMMCompletionStream
+from draive.tools import Toolbox
+from draive.types import Memory, UpdateSend
+
+__all__ = [
+    "ConversationCompletionStream",
+    "ConversationCompletion",
+]
+
+
+ConversationCompletionStream = LMMCompletionStream
+
+
+@runtime_checkable
+class ConversationCompletion(Protocol):
+    @overload
+    async def __call__(
+        self,
+        *,
+        instruction: str,
+        input: ConversationMessage | ConversationMessageContent,  # noqa: A002
+        memory: Memory[ConversationMessage] | None = None,
+        tools: Toolbox | None = None,
+        stream: Literal[True],
+    ) -> ConversationCompletionStream:
+        ...
+
+    @overload
+    async def __call__(
+        self,
+        *,
+        instruction: str,
+        input: ConversationMessage | ConversationMessageContent,  # noqa: A002
+        memory: Memory[ConversationMessage] | None = None,
+        tools: Toolbox | None = None,
+        stream: UpdateSend[ConversationStreamingUpdate],
+    ) -> ConversationMessage:
+        ...
+
+    @overload
+    async def __call__(
+        self,
+        *,
+        instruction: str,
+        input: ConversationMessage | ConversationMessageContent,  # noqa: A002
+        memory: Memory[ConversationMessage] | None = None,
+        tools: Toolbox | None = None,
+    ) -> ConversationMessage:
+        ...
+
+    async def __call__(  # noqa: PLR0913
+        self,
+        *,
+        instruction: str,
+        input: ConversationMessage | ConversationMessageContent,  # noqa: A002
+        memory: Memory[ConversationMessage] | None = None,
+        tools: Toolbox | None = None,
+        stream: UpdateSend[ConversationStreamingUpdate] | bool = False,
+    ) -> ConversationCompletionStream | ConversationMessage:
+        ...