Metadata-Version: 2.1
Name: dria_agent
Version: 0.1.53
Summary: speedy, performant and small tool calling agents
License: MIT
Author: andthattoo
Author-email: omer@firstbatch.xyz
Requires-Python: >=3.12,<4.0
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.12
Provides-Extra: huggingface
Provides-Extra: mlx
Provides-Extra: tools
Requires-Dist: docker (>=7.1.0,<8.0.0) ; extra == "tools" or extra == "tools"
Requires-Dist: duckduckgo-search (>=7.3.2,<8.0.0) ; extra == "tools"
Requires-Dist: google-api-python-client (>=2.160.0,<3.0.0) ; extra == "tools"
Requires-Dist: google-auth-oauthlib (>=1.2.1,<2.0.0) ; extra == "tools"
Requires-Dist: markdownify (>=0.14.1,<0.15.0) ; extra == "tools"
Requires-Dist: mlx (>=0.22.1,<0.23.0) ; extra == "mlx"
Requires-Dist: mlx-lm (>=0.21.4,<0.22.0) ; extra == "mlx"
Requires-Dist: numpy (>=2.2.2,<3.0.0)
Requires-Dist: ollama (>=0.4.7,<0.5.0)
Requires-Dist: openai (>=1.61.1,<2.0.0)
Requires-Dist: python-telegram (>=0.19.0,<0.20.0) ; extra == "tools"
Requires-Dist: requests (>=2.32.3,<3.0.0)
Requires-Dist: rich (>=13.9.4,<14.0.0)
Requires-Dist: scikit-learn (>=1.6.1,<2.0.0) ; extra == "tools"
Requires-Dist: sentence-transformers (>=3.4.1,<4.0.0) ; extra == "huggingface" or extra == "mlx"
Requires-Dist: slack-sdk (>=3.34.0,<4.0.0) ; extra == "tools"
Requires-Dist: smolagents (>=1.8.1,<2.0.0) ; extra == "tools"
Requires-Dist: transformers (>=4.48.3,<5.0.0) ; extra == "huggingface"
Requires-Dist: urllib3 (>=2.3.0,<3.0.0)
Description-Content-Type: text/markdown

# tiny-agent-α


### Installation

To install the package run: (Best for CPU inference)
```bash
pip install dria_agent
```
To use MLX as backend for macOS. 
```bash
pip install 'dria_agent[mlx]'
```
For HuggingFace backend for GPU.
```bash
pip install 'dria_agent[huggingface]'
```

In order to use extra tooling select a backend (e.g. mlx) and run:
```bash
pip install 'dria_agent[mlx, tools]'
```

### Models

| Model                  | Description                                | HF Download Link                                                                                                         | Ollama Tag                         | Size   |
|------------------------|--------------------------------------------|--------------------------------------------------------------------------------------------------------------------------|-------------------------------------|--------|
| Tiny-Agent-a-3B (8bit) | High performance and reasoning             | [Download](https://huggingface.co/driaforall/Tiny-Agent-a-3B/resolve/main/dria-agent-a-3b.Q8_0.gguf?download=true)       | driaforall/tiny-agent-a:3B-q8_0  | 3.3 GB |
| Tiny-Agent-a-3B (4bit) | Tradeoff 3B quality for memory             | [Download](https://huggingface.co/driaforall/Tiny-Agent-a-3B/resolve/main/dria-agent-a-3b.Q4_K_M.gguf?download=true)     | driaforall/tiny-agent-a:3B-q4_K_M | 1.9 GB |
| Tiny-Agent-a-1.5B (8bit) | Balanced performance and speed             | [Download](https://huggingface.co/driaforall/Tiny-Agent-a-1.5B/resolve/main/dria-agent-a-1.5b.Q8_0.gguf?download=true)   | driaforall/tiny-agent-a:1.5B-q8_0 | 1.6 GB |
| Tiny-Agent-a-1.5B (4bit) | Faster CPU inference, performance tradeoff | [Download](https://huggingface.co/driaforall/Tiny-Agent-a-1.5B/resolve/main/dria-agent-a-1.5b.Q8_0.gguf?download=true)   | driaforall/tiny-agent-a:1.5B-q4_K_M | 986 MB |
| Tiny-Agent-a-0.5B (8bit) | Ultra-light                                | [Download](https://huggingface.co/driaforall/Tiny-Agent-a-1.5B/resolve/main/dria-agent-a-1.5b.Q4_K_M.gguf?download=true) | driaforall/tiny-agent-a:0.5B-q8_0 | 531 MB |

a fast and powerful tool calling model designed to run on edge devices.

3B 8bit high performance and quality [download](https://huggingface.co/driaforall/Tiny-Agent-a-3B/resolve/main/dria-agent-a-3b.Q8_0.gguf?download=true)

3B 4bit balanced performance and speed [download](https://huggingface.co/driaforall/Tiny-Agent-a-3B/resolve/main/dria-agent-a-3b.Q4_K_M.gguf?download=true)

1.5B 8bit fast CPU inference [download](https://huggingface.co/driaforall/Tiny-Agent-a-1.5B/resolve/main/dria-agent-a-1.5b.Q8_0.gguf?download=true)

1.5B 4bit faster CPU inference [download](https://huggingface.co/driaforall/Tiny-Agent-a-1.5B/resolve/main/dria-agent-a-1.5b.Q8_0.gguf?download=true)

0.5B 8bit ultra-light [download](https://huggingface.co/driaforall/Tiny-Agent-a-1.5B/resolve/main/dria-agent-a-1.5b.Q4_K_M.gguf?download=true)



