janito 2.19.0__tar.gz → 2.20.0__tar.gz

{janito-2.19.0 → janito-2.20.0}/CHANGELOG.md

@@ -2,6 +2,25 @@
 
 All notable changes to this project will be documented in this file.
 
+## [2.20.0] - 2025-08-10
+
+### Added
+
+- **Mistral AI Provider**: Added complete support for Mistral AI models
+  - **General Purpose Models**:
+    - `mistral-large-latest`: 128k context, most capable model (default)
+    - `mistral-medium-latest`: 32k context, balanced performance
+    - `mistral-small-latest`: 32k context, compact and efficient
+  - **Code-Focused Models**:
+    - `codestral-latest`: 256k context, specialized for code generation
+    - `codestral-2405`: Previous version of code-focused model
+  - **Development-Focused Models**:
+    - `devstral-small-latest`: 128k context, optimized for agentic tool use
+    - `devstral-medium-latest`: 128k context, enhanced agentic capabilities
+- **Mistral Setup Guide**: Comprehensive documentation for Mistral AI integration
+- **Tool Support**: Full tool/function calling support across all Mistral models
+- **API Integration**: Uses OpenAI-compatible API format via Mistral's La Plateforme
+
 ## [2.19.0] - 2025-08-08
 
 ### Added

{janito-2.19.0 → janito-2.20.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: janito
-Version: 2.19.0
+Version: 2.20.0
 Summary: A new Python package called janito.
 Author-email: João Pinto <janito@ikignosis.org>
 Project-URL: Homepage, https://github.com/ikignosis/janito

janito-2.20.0/docs/index.md

@@ -0,0 +1,54 @@
+# Janito Documentation
+
+Welcome to the Janito documentation! This site contains comprehensive guides, tutorials, and reference materials for using and extending Janito.
+
+## What is Janito?
+
+Janito is a powerful AI assistant designed for developers, providing intelligent code assistance, terminal integration, and extensible tooling capabilities.
+
+## Quick Start
+
+- [**Installation Guide**](guides/installation.md) - Get Janito up and running
+- [**Quickstart & Basic Usage**](guides/using.md) - Learn the basics
+- [**Configuration**](guides/configuration.md) - Set up Janito for your needs
+
+## Key Features
+
+- **Terminal Integration**: Seamless shell integration with intelligent command assistance
+- **Code Intelligence**: Advanced code analysis and generation capabilities
+- **Extensible Tools**: Custom tool development and integration
+- **Multi-Provider Support**: Works with OpenAI, DeepSeek, MoonshotAI, and more
+- **Human-Guided AI**: Collaborative AI that works with you, not instead of you
+
+## Documentation Sections
+
+### Using Janito
+Learn how to use Janito effectively:
+- [Terminal Shell](guides/terminal-shell.md) - Interactive terminal sessions
+- [Single-Shot Commands](guides/single-shot-terminal.md) - One-off command execution
+- [Profiles](guides/profiles.md) - Custom configurations for different workflows
+
+### Development & Extension
+For developers looking to extend Janito:
+- [Developer Guide](guides/developing.md) - Core development concepts
+- [Tools Developer Guide](guides/tools-developer-guide.md) - Creating custom tools
+- [Developer Toolchain](meta/developer-toolchain.md) - Development setup and workflows
+
+### Supported Providers
+Janito supports multiple AI providers:
+- [OpenAI](openai-setup.md)
+- [DeepSeek](deepseek-setup.md)
+- [MoonshotAI](moonshotai-setup.md)
+- [Mistral](mistral-setup.md)
+- [Z.ai](z-ai-setup.md)
+- [Cerebras](cerebras-setup.md)
+
+## Getting Help
+
+- Check our [troubleshooting guides](guides/configuration.md)
+- Review [common issues](guides/using.md)
+- Explore [alternative solutions](alternatives.md)
+
+---
+
+*This documentation is automatically deployed from the latest commit and will always reflect the most current version of Janito.*

janito-2.20.0/docs/mistral-setup.md

@@ -0,0 +1,180 @@
+# Mistral Setup Guide
+
+This guide walks you through setting up Mistral AI models with Janito.
+
+## Overview
+
+Mistral AI provides a comprehensive suite of models including:
+- **General-purpose models**: Mistral Small, Medium, and Large
+- **Code-focused models**: Codestral for code generation and completion
+- **Development-focused models**: Devstral for agentic software development
+
+## Prerequisites
+
+1. **Python 3.8+** installed
+2. **Janito** installed (`pip install janito`)
+3. **Mistral API Key** from [La Plateforme](https://console.mistral.ai/)
+
+## Quick Setup
+
+### 1. Get Your API Key
+
+1. Visit [Mistral AI La Plateforme](https://console.mistral.ai/)
+2. Sign up or log in to your account
+3. Navigate to **API Keys** section
+4. Create a new API key
+5. Copy the key for use in Janito
+
+### 2. Configure Janito
+
+Set your Mistral API key using one of these methods:
+
+#### Method 1: Environment Variable (Recommended)
+
+```bash
+# Linux/macOS
+export MISTRAL_API_KEY="your-api-key-here"
+
+# Windows (PowerShell)
+$env:MISTRAL_API_KEY="your-api-key-here"
+
+# Windows (Command Prompt)
+set MISTRAL_API_KEY=your-api-key-here
+```
+
+#### Method 2: Janito CLI
+
+```bash
+janito set-api-key mistral your-api-key-here
+```
+
+#### Method 3: Interactive Setup
+
+```bash
+janito chat
+# Then use the !model command to select mistral and follow prompts
+```
+
+## Available Models
+
+### General Purpose Models
+- **mistral-large-latest** (default) - Most capable with 128k context
+- **mistral-medium-latest** - Balanced performance with 32k context
+- **mistral-small-latest** - Compact and efficient with 32k context
+
+### Code-Focused Models
+- **codestral-latest** - Specialized for code generation with 256k context
+- **codestral-2405** - Previous version of code-focused model
+
+### Development-Focused Models
+- **devstral-small-latest** - Optimized for agentic tool use
+- **devstral-medium-latest** - Enhanced agentic capabilities
+
+## Usage Examples
+
+### Basic Chat
+
+```bash
+# Start chat with default model (mistral-large-latest)
+janito chat --provider mistral
+
+# Use specific model
+janito chat --provider mistral --model codestral-latest
+```
+
+### Single-Shot Mode
+
+```bash
+# Quick question
+janito "Explain quantum computing" --provider mistral
+
+# Code generation
+janito "Write a Python function for fibonacci" --provider mistral --model codestral-latest
+```
+
+### Configuration File
+
+You can also configure Mistral in your Janito configuration:
+
+```yaml
+# ~/.janito/config.yaml
+providers:
+  mistral:
+    api_key: "your-api-key-here"
+    model: "mistral-large-latest"
+```
+
+## Model Selection Tips
+
+- **General conversations**: Use `mistral-large-latest` for best results
+- **Code generation**: Use `codestral-latest` for programming tasks
+- **Development workflows**: Use `devstral-*` models for agentic development
+- **Cost optimization**: Use `mistral-small-latest` for simpler tasks
+
+## Troubleshooting
+
+### Common Issues
+
+1. **"Invalid API key" error**
+   - Verify your API key is correct
+   - Check if the key has proper permissions
+   - Ensure no extra spaces in the key
+
+2. **"Model not found" error**
+   - Check available models with: `janito list-models --provider mistral`
+   - Ensure you're using the exact model name
+
+3. **Rate limiting**
+   - Mistral has rate limits based on your plan
+   - Consider upgrading your plan or implementing retry logic
+
+### Debug Mode
+
+Enable debug logging to see API requests:
+
+```bash
+# Enable HTTP debug for Mistral
+export MISTRAL_DEBUG_HTTP=1
+janito chat --provider mistral --verbose-api
+```
+
+### Testing Connection
+
+Test your setup:
+
+```bash
+# Ping the provider
+janito ping-providers
+
+# List available models
+janito list-models --provider mistral
+
+# Test with a simple query
+janito "Hello, Mistral!" --provider mistral
+```
+
+## Advanced Configuration
+
+### Custom Base URL
+
+If you need to use a different endpoint (e.g., for enterprise deployments):
+
+```bash
+janito set-config mistral.base_url "https://your-custom-endpoint.com/v1"
+```
+
+### Model Parameters
+
+You can set default parameters for Mistral models:
+
+```bash
+janito set-config mistral.temperature 0.5
+janito set-config mistral.max_tokens 2000
+```
+
+## Support
+
+For additional help:
+- Check the [Mistral AI documentation](https://docs.mistral.ai/)
+- Review [Janito troubleshooting guides](guides/configuration.md)
+- Join our [Discord community](https://discord.gg/janito)

janito-2.20.0/docs/overrides/partials/copyright.html

@@ -0,0 +1,6 @@
+<div class="md-copyright" style="text-align: center; width: 100%;">
+  <div class="md-copyright__highlight" style="display: inline-block;">
+    Generated by Janito • 
+    <span style="color: var(--md-accent-fg-color);">{{ config.extra.commit_hash or 'unknown' }}</span>
+  </div>
+</div>

janito-2.20.0/docs/provider-platform-access.md

@@ -0,0 +1,26 @@
+# Provider Platform and Documentation Access
+
+This table shows the accessibility status of each provider's platform/dashboard and API documentation.
+
+| Provider | Platform Access | Documentation | API Docs | Status |
+|----------|-----------------|---------------|----------|--------|
+| **Alibaba** | [Alibaba Cloud Console](https://account.alibabacloud.com) | [Model Studio Help](https://www.alibabacloud.com/help/en/model-studio) | [API Reference](https://www.alibabacloud.com/help/en/model-studio/api-reference) | ✅ Full Access |
+| **Anthropic** | [Console](https://console.anthropic.com) | [API Docs](https://docs.anthropic.com/en/api/getting-started) | [API Reference](https://docs.anthropic.com/en/api/messages) | ⚠️ Console Blocked |
+| **Cerebras** | [API Dashboard](https://api.cerebras.ai) | [Inference Docs](https://cerebras.ai/inference) | [API Docs](https://cerebras.ai/inference) | ✅ Full Access |
+| **DeepSeek** | [Platform](https://platform.deepseek.com) | [API Docs](https://platform.deepseek.com/api-docs) | [API Reference](https://platform.deepseek.com/api-docs) | ❌ All Blocked |
+| **Google** | [AI Studio](https://aistudio.google.com) | [Gemini API Docs](https://ai.google.dev/gemini-api/docs) | [API Reference](https://ai.google.dev/gemini-api/docs) | ✅ Full Access |
+| **MoonshotAI** | [Platform](https://platform.moonshot.ai) | [API Docs](https://platform.moonshot.ai/docs) | [API Reference](https://platform.moonshot.ai/docs) | ✅ Full Access |
+| **OpenAI** | [Platform](https://platform.openai.com) | [API Docs](https://platform.openai.com/docs) | [API Reference](https://platform.openai.com/docs/api-reference) | ❌ All Blocked |
+| **Z.ai** | [API Management](https://z.ai/manage-apikey/apikey-list) | [Model API Docs](https://z.ai/model-api) | [API Reference](https://z.ai/model-api) | ✅ Full Access |
+
+## Status Legend
+
+- ✅ **Full Access**: Both platform and documentation are publicly accessible
+- ⚠️ **Partial Access**: Documentation available but platform/dashboard blocked
+- ❌ **Blocked**: Both platform and documentation are inaccessible (403 errors)
+
+## Notes
+
+- **API Endpoints**: All API base URLs return 404 when accessed directly without authentication (expected behavior)
+- **Authentication**: All providers require valid API keys for actual API usage
+- **Regional Restrictions**: Some providers may have additional access restrictions based on geographic location

janito-2.20.0/docs/supported-providers-models.md

@@ -0,0 +1,110 @@
+# Supported Providers and Models
+
+This page lists the supported providers and their available models, organized by openness and sorted alphabetically within each category.
+
+> 📋 **Platform & Documentation Access**: For information about which providers have publicly accessible platforms and documentation vs. blocked access, see [Platform & Documentation Access](provider-platform-access.md).
+
+## Open-Source / Open-Weight Models
+
+### Alibaba
+
+- **Qwen3 235B A22B Instruct** (default) - Latest 1M context model
+- **Qwen3 235B A22B Thinking** - Reasoning-focused version
+- **Qwen3 30B A3B Instruct** - Compact 1M context model
+- **Qwen3 30B A3B Thinking** - Compact reasoning version
+- **Qwen3 Coder Plus** - Specialized for programming tasks
+- **Qwen3 Coder 480B** - Large-scale coding model
+- **Qwen Turbo** - High-speed general purpose
+- **Qwen Plus** - Balanced performance
+- **Qwen Max** - Maximum capability
+
+### Cerebras
+
+**Production Models:**
+
+- **Qwen-3 32B** - General instruction following
+
+**Preview Models:**
+
+- **Qwen-3 Coder 480B** - Programming-focused with 32k context
+- **Qwen-3 235B A22B Instruct** - Large-scale instruction model
+- **Qwen-3 235B A22B Thinking** - Reasoning-focused version
+- **GPT-OSS 120B** - Open-source model
+
+**Notes:**
+
+- All Cerebras models support 128k context window
+- Models are optimized for low-latency inference
+- Pricing varies by model size and capability
+
+### DeepSeek
+
+- **DeepSeek Chat** (default) - General purpose chat model
+- **DeepSeek Reasoner** - Specialized for complex reasoning tasks
+
+### Mistral
+
+- **Mistral Large Latest** (default) - Most capable Mistral model with 128k context
+- **Mistral Medium Latest** - Balanced performance with 32k context
+- **Mistral Small Latest** - Compact and efficient with 32k context
+- **Codestral Latest** - Specialized for code generation with 256k context
+- **Codestral 2405** - Previous version of code-focused model
+- **Devstral Small Latest** - Optimized for agentic tool use in software development
+- **Devstral Medium Latest** - Enhanced agentic capabilities for development tasks
+
+Mistral provides both general-purpose and specialized models, with Codestral specifically designed for code generation and Devstral for agentic software development.
+
+For setup instructions, see the [Mistral Setup Guide](mistral-setup.md).
+
+### MoonshotAI
+
+- **Kimi K2 Turbo Preview** (default) - Latest generation with enhanced performance
+- **Kimi K2 0711 Preview** - Previous preview version
+
+MoonshotAI provides open-source Kimi models with competitive performance.
+
+### Z.AI
+
+- **GLM-4.5** (default) - Advanced reasoning and conversation
+- **GLM-4.5 Air** - Compact and efficient version
+
+## Proprietary Models
+
+### Anthropic
+
+- **Claude 3.7 Sonnet** (default) - Latest Claude model with enhanced reasoning
+- **Claude 4 Opus** - Most capable Claude model (preview)
+- **Claude 4 Sonnet** - Balanced performance and capability (preview)
+- **Claude 3.5 Sonnet** - Previous generation, still highly capable
+- **Claude 3.5 Haiku** - Fast and efficient
+- **Claude 3 Haiku** - Compact and cost-effective
+
+### Google
+
+- **Gemini 2.5 Flash** (default) - Fast and efficient
+- **Gemini 2.5 Pro** - Advanced capabilities with extended context
+- **Gemini 2.5 Flash-Lite Preview** - Lightweight preview version
+
+### OpenAI
+
+- **GPT-5** (default) - Latest generation with advanced reasoning
+- **GPT-5 Mini** - Compact version of GPT-5
+- **GPT-5 Nano** - Ultra-compact version
+- **GPT-4.1** - Enhanced GPT-4 with improved capabilities
+- **GPT-4.1-mini** - Balanced performance and efficiency
+- **GPT-4.1-nano** - Lightweight version
+- **GPT-4 Turbo** - High-performance GPT-4 variant
+- **GPT-4o** - Multimodal GPT-4 optimized for chat
+- **GPT-4o-mini** - Compact multimodal version
+- **o3** - Advanced reasoning model
+- **o3-mini** - Compact reasoning model
+- **o4-mini** - Latest mini reasoning model
+- **GPT-3.5 Turbo** - Cost-effective general purpose
+
+For setup instructions, see the [OpenAI Setup Guide](openai-setup.md).
+
+### Azure OpenAI
+
+- **Azure OpenAI Deployment** - Custom Azure OpenAI deployments
+
+For setup instructions, see [Using Azure OpenAI with Janito](reference/azure-openai.md).

{janito-2.19.0 → janito-2.20.0}/janito/cli/cli_commands/list_models.py

@@ -2,8 +2,8 @@
 CLI Command: List models for the specified/current provider
 """
 
-from janito.cli.cli_commands.model_utils import _print_models_table
 import sys
+from janito.cli.cli_commands.model_utils import _print_models_table
 
 _provider_instance = None
 
@@ -36,8 +36,6 @@ def handle_list_models(args, provider_instance):
             
             for m in models:
                 table.add_row(str(m))
-                
-            import sys
             if sys.stdout.isatty():
                 shared_console.print(table)
             else:

{janito-2.19.0 → janito-2.20.0}/janito/drivers/openai/driver.py

@@ -64,14 +64,20 @@ class OpenAIModelDriver(LLMDriver):
         if config.model:
             api_kwargs["model"] = config.model
         # Prefer max_completion_tokens if present, else fallback to max_tokens (for backward compatibility)
-        if (
-            hasattr(config, "max_completion_tokens")
-            and config.max_completion_tokens is not None
-        ):
-            api_kwargs["max_completion_tokens"] = int(config.max_completion_tokens)
+        # Skip max_completion_tokens for Mistral as their API doesn't support it
+        is_mistral = config.base_url and "mistral.ai" in str(config.base_url)
+        if not is_mistral:
+            if (
+                hasattr(config, "max_completion_tokens")
+                and config.max_completion_tokens is not None
+            ):
+                api_kwargs["max_completion_tokens"] = int(config.max_completion_tokens)
+            elif hasattr(config, "max_tokens") and config.max_tokens is not None:
+                # For models that do not support 'max_tokens', map to 'max_completion_tokens'
+                api_kwargs["max_completion_tokens"] = int(config.max_tokens)
         elif hasattr(config, "max_tokens") and config.max_tokens is not None:
-            # For models that do not support 'max_tokens', map to 'max_completion_tokens'
-            api_kwargs["max_completion_tokens"] = int(config.max_tokens)
+            # For Mistral, use max_tokens directly
+            api_kwargs["max_tokens"] = int(config.max_tokens)
         for p in (
             "temperature",
             "top_p",

janito-2.20.0/janito/i18n/it.py

@@ -0,0 +1,46 @@
+# pragma: allowlist secret
+translations = {
+    "36107ed78ab25f6fb12ad8ce13018cd1ce6735d1": "Avvio del server web...",
+    "70a0d194687568a47aa617fd85036ace1e69a982": "Vuoi davvero uscire? (s/n): ",
+    "5c9ebcbbd7632ecb328bd52958b17158afaa32c6": "F12 = Azione Rapida (segue l'azione raccomandata)",
+    "fe21121e2934234b68d19b2757532117d440c1e3": "Chiave API non trovata. Si prega di configurare 'api_key' nel file di configurazione.",
+    "c9e3759b1756eba35b381ce2b72cd659e132b01f": "Ciao, {name}!",
+    "ca1fee2f55baabdc2e4b0e9529c89ee024e62079": "Nessun prompt fornito nei messaggi",
+    "f7449d23d0c500ae2a0b31e04f92b47a4d8ae845": "max_tokens deve essere un intero, ricevuto: {resolved_max_tokens!r}",
+    "70a9ed8edb6da12e208431a31aa16ba54419b26f": "Risposta non valida/malformed da OpenAI (tentativo {attempt}/{max_retries}). Riprovo tra {wait_time} secondi...",
+    "a873085e3b06184fb5d27e842f97b06b6190976d": "Numero massimo di tentativi per risposta non valida raggiunto. Generazione errore.",
+    "66a34568bbe846bb1bde3619eb4d6dfa10211104": "L'API non supporta l'uso degli strumenti.",
+    "09b81476b75586da4116b83f8be70d77b174cec3": "Limite di richieste API OpenAI (429) (tentativo {attempt}/{max_retries}): {e}. Riprovo tra {wait_time} secondi...",
+    "5717a35dd2a1533fb7e15edc8c9329cb69f3410b": "Errore server API OpenAI (tentativo {attempt}/{max_retries}): {e}. Riprovo tra {wait_time} secondi...",
+    "02e760ba15ed863176c1290ac8a9b923963103cd": "Errore client API OpenAI {status_code}: {e}. Nessun nuovo tentativo.",
+    "2e52b0bbc8f16226b70e3e20f95c9245d2bcdb47": "Errore API OpenAI (tentativo {attempt}/{max_retries}): {e}. Riprovo tra {wait_time} secondi...",
+    "012cc970e039fdd79c452fc676202c814ffc76ae": "Numero massimo di tentativi per errore API OpenAI raggiunto. Generazione errore.",
+    "d0438e45667d31e0022b2497b5901cd4300f084b": "QueuedMessageHandler.handle_message si aspetta un dizionario con 'type' e 'message', ricevuto {msg_type}: {msg!r}",
+    "9d3460187ffa19c7c8a4020157072b1087e1bd2f": "[QueuedMessageHandler] {msg_type}: {msg}",
+    "3813833343430e8afa8fce33385c5e39fb24dd60": "[QueuedMessageHandler] {msg_type}: {message}",
+    "0be9a22226e16a40797010d23a0f581542dca40e": "[ToolExecutor] {tool_name} chiamato con argomenti: {args}",
+    "42c68edcb25442f518b1af77c6a2ddc07461aae0": "[ToolExecutor] Motivo chiamata: {tool_call_reason}",
+    "002ff598115d84595ffeee6219cb5c03d3a1d4a6": "Domanda",
+    "35747d13dcd91e8e8790c7f767d5ed764f541b9e": "procedi",
+    "33dde3a1afbc418768a69fa53168d9b0638fe1aa": "avanti",
+    "eee0bbba4ff92adbeb038a77df0466d660f15716": "continua",
+    "edee9402d198b04ac77dcf5dc9cc3dac44573782": "prossimo",
+    "8fdb7e2fa84f4faf0d9b92f466df424ec47a165b": "ok",
+    "5f8f924671cda79b5205a6bf1b776f347c4a7a07": "Opzioni di configurazione disponibili:\n",
+    "cef780a309cd234750764d42697882c24168ddab": "{key:15} {desc} (predefinito: {default})",
+    "68c2cc7f0ceaa3e499ecb4db331feb4debbbcc23": "Modello",
+    "c3f104d1365744b538bfde9f4adb6a6df4b80355": "Funzione",
+    "99a0efc6cfd85d8ff2732a6718140f822cb90472": "Stile",
+    "f1702b4686278becffc88baabe6f4b7a8355532c": "Messaggi",
+    "c38c6c1f3a2743f8626703abb302e403d20ff81c": "Token",
+    "a817d7eb8e0f1dab755ab5203a082e5c3c094fce": "Prompt",
+    "2ff255684a2277f806fcebf3fe338ed27857f350": "Completamento",
+    "b25928c69902557b0ef0a628490a3a1768d7b82f": "Totale",
+    "76e63d65c883ed50df40ac3aeef0c2d6a1c4ad60": "Azione Rapida",
+    "5397e0583f14f6c88de06b1ef28f460a1fb5b0ae": "Sì",
+    "816c52fd2bdd94a63cd0944823a6c0aa9384c103": "No",
+    "c47ae15370cfe1ed2781eedc1dc2547d12d9e972": "Aiuto",
+    "cc3dbd47e1cf9003a55d3366b3adbcd72275e525": "Nuovo Task",
+    "efa5a8b84e1afe65c81ecfce28c398c48f19ddc2": "Benvenuto su Janito{version_str}! Modalità chat attiva. Digita /exit per uscire.",
+    "b314d6e1460f86e0f21abc5aceb7935a2a0667e8": "Benvenuto su Janito{version_str} in modalità [white on magenta]VANILLA[/white on magenta]! Strumenti, prompt di sistema e temperatura sono disattivati, a meno che non siano sovrascritti.",
+}

{janito-2.19.0 → janito-2.20.0}/janito/provider_registry.py

@@ -39,7 +39,17 @@ class ProviderRegistry:
             if len(info) == 4 and info[3]:
                 continue  # skip providers flagged as not implemented
             rows.append(info[:3])
-        rows.sort(key=self._maintainer_sort_key)
+        
+        # Group providers by openness (open-source first, then proprietary)
+        open_providers = {'cerebras', 'deepseek', 'alibaba', 'moonshotai', 'zai'}
+        
+        def sort_key(row):
+            provider_name = row[0]
+            is_open = provider_name in open_providers
+            # Sort open providers alphabetically first, then proprietary alphabetically
+            return (not is_open, provider_name)
+        
+        rows.sort(key=sort_key)
         return rows
 
     def _add_rows_to_table(self, table, rows):

{janito-2.19.0 → janito-2.20.0}/janito/providers/__init__.py

@@ -8,3 +8,4 @@ import janito.providers.moonshotai.provider
 import janito.providers.alibaba.provider
 import janito.providers.zai.provider
 import janito.providers.cerebras.provider
+import janito.providers.mistral.provider

janito-2.20.0/janito/providers/mistral/__init__.py

@@ -0,0 +1 @@
+# Codestral provider module

janito-2.20.0/janito/providers/mistral/model_info.py

@@ -0,0 +1,81 @@
+from janito.llm.model import LLMModelInfo
+
+MODEL_SPECS = {
+    "codestral-latest": LLMModelInfo(
+        name="codestral-latest",
+        context=256000,
+        max_input=250000,
+        max_cot="N/A",
+        max_response=4096,
+        thinking_supported=False,
+        default_temp=0.2,
+        open="mistral",
+        driver="OpenAIModelDriver",
+    ),
+    "codestral-2405": LLMModelInfo(
+        name="codestral-2405",
+        context=256000,
+        max_input=250000,
+        max_cot="N/A",
+        max_response=4096,
+        thinking_supported=False,
+        default_temp=0.2,
+        open="mistral",
+        driver="OpenAIModelDriver",
+    ),
+    "mistral-small-latest": LLMModelInfo(
+        name="mistral-small-latest",
+        context=32000,
+        max_input=28000,
+        max_cot="N/A",
+        max_response=4096,
+        thinking_supported=False,
+        default_temp=0.7,
+        open="mistral",
+        driver="OpenAIModelDriver",
+    ),
+    "mistral-medium-latest": LLMModelInfo(
+        name="mistral-medium-latest",
+        context=32000,
+        max_input=28000,
+        max_cot="N/A",
+        max_response=4096,
+        thinking_supported=False,
+        default_temp=0.7,
+        open="mistral",
+        driver="OpenAIModelDriver",
+    ),
+    "mistral-large-latest": LLMModelInfo(
+        name="mistral-large-latest",
+        context=128000,
+        max_input=120000,
+        max_cot="N/A",
+        max_response=4096,
+        thinking_supported=False,
+        default_temp=0.7,
+        open="mistral",
+        driver="OpenAIModelDriver",
+    ),
+    "devstral-small-latest": LLMModelInfo(
+        name="devstral-small-latest",
+        context=128000,
+        max_input=120000,
+        max_cot="N/A",
+        max_response=4096,
+        thinking_supported=False,
+        default_temp=0.7,
+        open="mistral",
+        driver="OpenAIModelDriver",
+    ),
+    "devstral-medium-latest": LLMModelInfo(
+        name="devstral-medium-latest",
+        context=128000,
+        max_input=120000,
+        max_cot="N/A",
+        max_response=4096,
+        thinking_supported=False,
+        default_temp=0.7,
+        open="mistral",
+        driver="OpenAIModelDriver",
+    ),
+}