tinyflow-llm 0.1.0__py3-none-any.whl

tinyflow_llm-0.1.0.dist-info/METADATA

@@ -0,0 +1,243 @@
+Metadata-Version: 2.4
+Name: tinyflow-llm
+Version: 0.1.0
+Summary: A lightweight, extensible, and provider-agnostic Python framework for building LLM agents.
+Project-URL: Homepage, https://github.com/your-username/tinyflow
+Project-URL: Repository, https://github.com/your-username/tinyflow
+Project-URL: Issues, https://github.com/your-username/tinyflow/issues
+Author-email: Your Name <your.email@example.com>
+License: MIT
+License-File: LICENSE
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.12
+Requires-Dist: anthropic>=0.78.0
+Requires-Dist: google-genai>=1.62.0
+Requires-Dist: google-generativeai>=0.8.6
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: openai>=2.16.0
+Requires-Dist: pydantic-settings>=2.12.0
+Requires-Dist: pydantic>=2.12.5
+Requires-Dist: python-dotenv>=1.2.1
+Requires-Dist: rich>=14.3.2
+Requires-Dist: tavily>=1.1.0
+Requires-Dist: tenacity>=9.1.3
+Provides-Extra: dev
+Requires-Dist: basedpyright>=1.37.4; extra == 'dev'
+Requires-Dist: black>=26.1.0; extra == 'dev'
+Requires-Dist: pyright>=1.1.408; extra == 'dev'
+Requires-Dist: pytest-asyncio>=1.3.0; extra == 'dev'
+Requires-Dist: pytest>=9.0.2; extra == 'dev'
+Requires-Dist: ruff>=0.15.0; extra == 'dev'
+Provides-Extra: local
+Requires-Dist: onnxruntime<1.24.1; extra == 'local'
+Requires-Dist: sentence-transformers<3.0,>=2.2.0; extra == 'local'
+Requires-Dist: torch==2.2.2; extra == 'local'
+Provides-Extra: vector
+Requires-Dist: chromadb>=1.4.1; extra == 'vector'
+Requires-Dist: qdrant-client>=1.16.2; extra == 'vector'
+Description-Content-Type: text/markdown
+
+# TinyFlow
+
+**TinyFlow** is a lightweight, extensible, and provider-agnostic Python framework for building LLM agents. It provides a clean abstraction layer over major LLM providers (OpenAI, Anthropic, Gemini), embedding models, and vector databases, enabling you to build complex agentic workflows with ease.
+
+## 🚀 Key Features
+
+- **Provider Agnostic**: Switch seamlessly between OpenAI, Anthropic, and Google Gemini models using a unified `LLMFactory`.
+- **Agentic Workflow**: Built-in "Thinking -> Acting -> Observing" loop for autonomous task execution.
+- **Tool Support**: Easy-to-use `@tool` decorator to give your agents capabilities (web search, code execution, etc.).
+- **Memory & RAG**: Integrated Vector Database support (ChromaDB, Qdrant) and Embedding abstraction (OpenAI, SentenceTransformers).
+- **Unified Configuration**: flexible configuration system supporting parameters, environment variables, and settings files with clear precedence.
+- **Streaming UI**: Rich streaming support for building interactive chat interfaces, including reasoning steps and tool execution visibility.
+- **Modern Stack**: Built with Python 3.12+, Pydantic, Asyncio, and managed with `uv`.
+
+## 📦 Installation
+
+You can install TinyFlow using `pip` or `uv`.
+
+### Using pip
+
+```bash
+pip install tinyflow-llm
+```
+
+### Using uv (Recommended)
+
+```bash
+uv add tinyflow-llm
+```
+
+### Installation with Extras
+
+TinyFlow supports optional dependencies for specific features:
+
+- **Local Embeddings**: `pip install "tinyflow-llm[local]"`
+- **Vector Databases**: `pip install "tinyflow-llm[vector]"` (includes ChromaDB and Qdrant)
+
+## ⚙️ Configuration
+
+TinyFlow uses a unified configuration system. You can configure providers via:
+
+1.  **Explicit Parameters** (passed to Factory `create` methods) - _Highest Priority_
+2.  **Environment Variables** - _Medium Priority_
+3.  **Settings / Defaults** - _Lowest Priority_
+
+### Common Environment Variables
+
+| Category       | Variable             | Description                                      |
+| -------------- | -------------------- | ------------------------------------------------ |
+| **LLM**        | `LLM_PROVIDER`       | `openai`, `anthropic`, or `gemini`               |
+|                | `LLM_API_KEY`        | API Key for the selected provider                |
+|                | `LLM_MODEL`          | Model name (e.g., `gpt-4o`, `claude-3-5-sonnet`) |
+|                | `LLM_BASE_URL`       | Optional custom base URL (e.g., for proxies)     |
+| **Embeddings** | `EMBEDDING_PROVIDER` | `openai`, `local`, or `sentence-transformers`    |
+|                | `EMBEDDING_API_KEY`  | API Key for embedding provider                   |
+|                | `EMBEDDING_MODEL`    | Model name (e.g., `text-embedding-3-small`)      |
+| **Vector DB**  | `VECTOR_DB_PROVIDER` | `chroma` or `qdrant`                             |
+|                | `VECTOR_DB_PATH`     | Path for local ChromaDB persistence              |
+|                | `VECTOR_DB_URL`      | URL for remote Vector DB (e.g., Qdrant Cloud)    |
+|                | `VECTOR_DB_API_KEY`  | API Key for remote Vector DB                     |
+
+## 💡 Usage
+
+### 1. Basic LLM Usage
+
+Use the `LLMFactory` to create a provider instance. It automatically handles configuration.
+
+```python
+import asyncio
+from tinyflow.providers.base.factory import LLMFactory
+from tinyflow.core.types import Message
+
+async def main():
+    # Automatically loads config from env vars
+    llm = LLMFactory.create()
+
+    # Or specify explicitly
+    # llm = LLMFactory.create(provider="anthropic", model="claude-3-opus-20240229")
+
+    messages = [
+        Message(role="system", content="You are a helpful assistant."),
+        Message(role="user", content="Explain quantum computing in one sentence.")
+    ]
+
+    response = await llm.generate(messages)
+    print(response.content)
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+### 2. Building an Agent with Tools
+
+Create an `Agent` equipped with custom tools.
+
+```python
+import asyncio
+from tinyflow.core.agent import Agent
+from tinyflow.providers.base.factory import LLMFactory
+from tinyflow.core.tools import tool
+
+# Define a tool
+@tool
+def get_weather(location: str, unit: str = "celsius") -> str:
+    """Get the current weather for a location."""
+    # In a real app, call a weather API here
+    return f"The weather in {location} is 25°{unit.upper()} and sunny."
+
+async def main():
+    # 1. Initialize LLM
+    llm = LLMFactory.create()
+
+    # 2. Create Agent with tools
+    agent = Agent(
+        llm=llm,
+        tools=[get_weather],
+        system_prompt="You are a helpful assistant with access to weather tools."
+    )
+
+    # 3. Run Agent
+    response = await agent.run("What's the weather like in Paris?")
+    print(response)
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+### 3. Using Vector Memory
+
+Integrate RAG (Retrieval-Augmented Generation) capabilities.
+
+```python
+from tinyflow.vector.factory import VectorDBFactory
+from tinyflow.embeddings.factory import EmbeddingFactory
+from tinyflow.memory.vector import VectorMemory
+
+# Initialize components
+embedding_model = EmbeddingFactory.create()
+vector_db = VectorDBFactory.create()
+
+# Create memory interface
+memory = VectorMemory(
+    vector_db=vector_db,
+    embedding_model=embedding_model
+)
+
+# Use in Agent
+agent = Agent(
+    llm=llm,
+    memory=memory,
+    system_prompt="Use your memory to answer questions."
+)
+```
+
+## 🏗️ Project Structure
+
+```
+tinyflow/
+├── tinyflow/
+│   ├── config/       # Configuration and helper utilities
+│   ├── core/         # Core abstractions (Agent, Tools, Types)
+│   ├── providers/    # LLM Provider implementations (OpenAI, Anthropic, Gemini)
+│   ├── embeddings/   # Embedding models (OpenAI, Local)
+│   ├── vector/       # Vector Database adapters (Chroma, Qdrant)
+│   └── memory/       # Memory implementations
+├── tests/            # Unit and integration tests
+├── main.py           # Entry point example
+└── pyproject.toml    # Project dependencies and config
+```
+
+## 🧪 Development
+
+### Running Tests
+
+TinyFlow uses `pytest` for testing.
+
+```bash
+# Run all tests
+uv run pytest
+
+# Run specific test file
+uv run pytest tests/test_factories.py -v
+```
+
+### Code Style
+
+The project uses `ruff` for linting and formatting.
+
+```bash
+# Lint
+uv run ruff check .
+
+# Format
+uv run ruff format .
+```
+
+## 📄 License
+
+[MIT License](LICENSE)

tinyflow_llm-0.1.0.dist-info/RECORD

@@ -0,0 +1,43 @@
+tinyflow/__init__.py,sha256=JjvZNmhlebmCdwcaKEZUKU_hivWFFTXQowvZaTqLEFo,1315
+tinyflow/config/helpers.py,sha256=Q9vGX6LrWqrORWsFgQVhpiCJ9Op0McrL7uglpetotjQ,2022
+tinyflow/config/settings.py,sha256=VyVFhqMDYP1iglaJoAav2wuZjbn6qXlqt7g3EOdrh5U,2902
+tinyflow/core/__init__.py,sha256=ppLaYPNv9NNNCbhn30HgsONF6i79reFyFHqvb9R5r7M,667
+tinyflow/core/agent.py,sha256=XIOf1c69XrbAp3szSnLlOW4JX22S3nskIJGq5p9CxDI,18845
+tinyflow/core/exceptions.py,sha256=ff_KYeCNlfCgBUXkTAZo2jipUvKur2boboImu5OzmMc,1469
+tinyflow/core/logger.py,sha256=DaXcv7_0xJv7Af1YUQBuc9Y4w9y0nzCa6gyiWafuDlI,408
+tinyflow/core/message.py,sha256=VvJFlV5dXk-cysGlFcIICltOanwzDb5dczOK7YZOcqY,241
+tinyflow/core/protocol.py,sha256=6LHOZbdTMojIiQIlRZXNk2uI43s7EkMJktUYP1k76r0,3502
+tinyflow/core/tools.py,sha256=JYk1IfmhXvuJDYqPNOkQUsR94JtREiW4vn5S9Yp6aPg,7434
+tinyflow/core/types.py,sha256=oODX-N8OpyIdtJA0MxVbphLUGcct3oikfzDiAlb8fE8,7049
+tinyflow/embeddings/__init__.py,sha256=OI_uoOZ_IloSsAPiqZBs3JE7q_Ei4TKQV_05uL5A4KE,77
+tinyflow/embeddings/base.py,sha256=at7_uvr2GuZlXsCMKNieiBR6rh0YxUjtquHhWCkz5w4,643
+tinyflow/embeddings/factory.py,sha256=tzGIkx0vHV-3BdAVNbTmQdm0LgyyodbGZV73h0BQoKA,5098
+tinyflow/embeddings/local_embedding.py,sha256=xeQIl0UdeEFvWJFjdzihSRz5bqSmm8eB8EOk_2ZPmAA,2079
+tinyflow/embeddings/openai_embedding.py,sha256=5Io3sz5wMlUEo1bRi_uy5HB5YT8lpza46f-FUb6sjjA,2202
+tinyflow/memory/__init__.py,sha256=PtT0bM8wF4Jpoq7pIX0g27fbSI2AoMv6D3mn_3sM6Vk,153
+tinyflow/memory/base.py,sha256=9IgmOjlQb06iY17l66pnAlIatEKvISJZQ8hQxvYl-vs,309
+tinyflow/memory/simple.py,sha256=fTmI3qXtISQhS00rY6P8fbLIhSoxxs8VkTnT_MkhnNk,969
+tinyflow/memory/vector.py,sha256=gReV9WeA9a64rogI5266g7BZ8jZIPDfs9Xupl7ZoAkA,756
+tinyflow/providers/anthropic_llm.py,sha256=kzMxz7FljBCxZBIdke4FzPe4zSPDlgN3yXDUSjt2ARk,4153
+tinyflow/providers/gemini_llm.py,sha256=1IJf97cN6xilyQmmJF5X_ojYnWVPUSBl6SrDs_1O7K8,4694
+tinyflow/providers/openai_llm.py,sha256=WXD9eKzOmpAbuDFIdPRxYhRacP1lAR84IcSYTVMKCoI,7297
+tinyflow/providers/base/factory.py,sha256=C-g826n3Vr5PPNxQNVSaQa01t0UENVvfV4POSI1o5M4,3022
+tinyflow/providers/base/llm.py,sha256=TKOjsMl8iQFveW3dR0qfMfWDjmhMWx5nLpKZ1Gg84zI,1314
+tinyflow/tools/builtin/__init__.py,sha256=UPAhiAW5VQHhx7NQ4v6Gbo8WqcW6Djrt723FUfsY4Qk,946
+tinyflow/tools/builtin/code_execution.py,sha256=vhM602O0SGLSPt_w2sb7DHv4OhMFtSJskk07yP0dWks,4534
+tinyflow/tools/builtin/search.py,sha256=0NhHcKB5ba252MWK3Xt6hEJuXQVPlG31wwKMSmMuIMM,4796
+tinyflow/tools/builtin/web_reader.py,sha256=3A-ammXOf4MCcRLmeRVfewifubG55Sz3YfJ_TDmQNao,2777
+tinyflow/vector/__init__.py,sha256=Vi8ViT7r8lR--vxVZ1IG2icHclKGg4Wdcfv3AQuBpAg,83
+tinyflow/vector/base.py,sha256=R6zJgrOmTO3a6oh1c9-VC-7Yfe6mM5OMXli_bkMbV2o,1620
+tinyflow/vector/chroma_db.py,sha256=N7UjEIIpVWaQ5w1RUrRZ1q2X7D-fsKTSrMG5y7i7Skw,4528
+tinyflow/vector/factory.py,sha256=8MECybeXBZa_UfPdxV6hv5871quYiNL7NBMYVNsDhH0,2720
+tinyflow/vector/qdrant_db.py,sha256=6OMElLG06H9zjhuLFTF7_2fGNYgaV0MgjWaZE_LIfY0,6190
+tinyflow/workflow/__init__.py,sha256=h4v14_ebYxawVcPXQNiJQ_Tw6uq5EX-1O_IqbOk2vwc,457
+tinyflow/workflow/executor.py,sha256=v4Sdj7751-cKaJBQ8A6tBkDMhM5IncIGACuxxY9GGV4,10516
+tinyflow/workflow/hooks.py,sha256=xh06rCQZMBLvasHI2dKvbyjPPpSxYeTDn7q_1Ve2LAk,5773
+tinyflow/workflow/state.py,sha256=WQK_jYTbZfo4jnT0U1LVgv9svUa2sqNXNcsvRuH8vqw,4822
+tinyflow/workflow/step.py,sha256=J4DrMNQgUEjU0sFlDuPE6YdebXPmbmDmpuvTsaKlLwk,2574
+tinyflow_llm-0.1.0.dist-info/METADATA,sha256=fHPuT60gj6_KX8Z4MMHHLjrJo2LJ5J8YRI-ojg8dtJs,8279
+tinyflow_llm-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+tinyflow_llm-0.1.0.dist-info/licenses/LICENSE,sha256=v2spsd7N1pKFFh2G8wGP_45iwe5S0DYiJzG4im8Rupc,1066
+tinyflow_llm-0.1.0.dist-info/RECORD,,

tinyflow_llm-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

tinyflow_llm-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Your Name
+
+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.