mistralai 0.4.2__py3-none-any.whl → 1.0.0__py3-none-any.whl

mistralai/exceptions.py

@@ -1,54 +0,0 @@
-from __future__ import annotations
-
-from typing import Any, Dict, Optional
-
-from httpx import Response
-
-
-class MistralException(Exception):
-    """Base Exception class, returned when nothing more specific applies"""
-
-    def __init__(self, message: Optional[str] = None) -> None:
-        super(MistralException, self).__init__(message)
-
-        self.message = message
-
-    def __str__(self) -> str:
-        msg = self.message or "<empty message>"
-        return msg
-
-    def __repr__(self) -> str:
-        return f"{self.__class__.__name__}(message={str(self)})"
-
-
-class MistralAPIException(MistralException):
-    """Returned when the API responds with an error message"""
-
-    def __init__(
-        self,
-        message: Optional[str] = None,
-        http_status: Optional[int] = None,
-        headers: Optional[Dict[str, Any]] = None,
-    ) -> None:
-        super().__init__(message)
-        self.http_status = http_status
-        self.headers = headers or {}
-
-    @classmethod
-    def from_response(cls, response: Response, message: Optional[str] = None) -> MistralAPIException:
-        return cls(
-            message=message or response.text,
-            http_status=response.status_code,
-            headers=dict(response.headers),
-        )
-
-    def __repr__(self) -> str:
-        return f"{self.__class__.__name__}(message={str(self)}, http_status={self.http_status})"
-
-
-class MistralAPIStatusException(MistralAPIException):
-    """Returned when we receive a non-200 response from the API that we should retry"""
-
-
-class MistralConnectionException(MistralException):
-    """Returned when the SDK can not reach the API server for any reason"""

mistralai/models/chat_completion.py

@@ -1,93 +0,0 @@
-from enum import Enum
-from typing import List, Optional
-
-from pydantic import BaseModel
-
-from mistralai.models.common import UsageInfo
-
-
-class Function(BaseModel):
-    name: str
-    description: str
-    parameters: dict
-
-
-class ToolType(str, Enum):
-    function = "function"
-
-
-class FunctionCall(BaseModel):
-    name: str
-    arguments: str
-
-
-class ToolCall(BaseModel):
-    id: str = "null"
-    type: ToolType = ToolType.function
-    function: FunctionCall
-
-
-class ResponseFormats(str, Enum):
-    text: str = "text"
-    json_object: str = "json_object"
-
-
-class ToolChoice(str, Enum):
-    auto: str = "auto"
-    any: str = "any"
-    none: str = "none"
-
-
-class ResponseFormat(BaseModel):
-    type: ResponseFormats = ResponseFormats.text
-
-
-class ChatMessage(BaseModel):
-    role: str
-    content: str
-    name: Optional[str] = None
-    tool_calls: Optional[List[ToolCall]] = None
-    tool_call_id: Optional[str] = None
-
-
-class DeltaMessage(BaseModel):
-    role: Optional[str] = None
-    content: Optional[str] = None
-    tool_calls: Optional[List[ToolCall]] = None
-
-
-class FinishReason(str, Enum):
-    stop = "stop"
-    length = "length"
-    error = "error"
-    tool_calls = "tool_calls"
-
-
-class ChatCompletionResponseStreamChoice(BaseModel):
-    index: int
-    delta: DeltaMessage
-    finish_reason: Optional[FinishReason]
-
-
-class ChatCompletionStreamResponse(BaseModel):
-    id: str
-    model: str
-    choices: List[ChatCompletionResponseStreamChoice]
-    created: Optional[int] = None
-    object: Optional[str] = None
-    usage: Optional[UsageInfo] = None
-
-
-class ChatCompletionResponseChoice(BaseModel):
-    index: int
-    message: ChatMessage
-    finish_reason: Optional[FinishReason]
-
-
-class ChatCompletionResponse(BaseModel):
-    id: str
-    object: str
-    created: int
-    model: str
-    choices: List[ChatCompletionResponseChoice]
-    usage: UsageInfo

mistralai/models/common.py

@@ -1,9 +0,0 @@
-from typing import Optional
-
-from pydantic import BaseModel
-
-
-class UsageInfo(BaseModel):
-    prompt_tokens: int
-    total_tokens: int
-    completion_tokens: Optional[int]

mistralai/models/embeddings.py

@@ -1,19 +0,0 @@
-from typing import List
-
-from pydantic import BaseModel
-
-from mistralai.models.common import UsageInfo
-
-
-class EmbeddingObject(BaseModel):
-    object: str
-    embedding: List[float]
-    index: int
-
-
-class EmbeddingResponse(BaseModel):
-    id: str
-    object: str
-    data: List[EmbeddingObject]
-    model: str
-    usage: UsageInfo

mistralai/models/files.py

@@ -1,23 +0,0 @@
-from typing import Literal, Optional
-
-from pydantic import BaseModel
-
-
-class FileObject(BaseModel):
-    id: str
-    object: str
-    bytes: int
-    created_at: int
-    filename: str
-    purpose: Optional[Literal["fine-tune"]] = "fine-tune"
-
-
-class FileDeleted(BaseModel):
-    id: str
-    object: str
-    deleted: bool
-
-
-class Files(BaseModel):
-    data: list[FileObject]
-    object: Literal["list"]

mistralai/models/jobs.py

@@ -1,100 +0,0 @@
-from datetime import datetime
-from typing import Annotated, List, Literal, Optional, Union
-
-from pydantic import BaseModel, Field
-
-
-class TrainingParameters(BaseModel):
-    training_steps: int = Field(1800, le=10000, ge=1)
-    learning_rate: float = Field(1.0e-4, le=1, ge=1.0e-8)
-
-
-class WandbIntegration(BaseModel):
-    type: Literal["wandb"] = "wandb"
-    project: str
-    name: Union[str, None] = None
-    run_name: Union[str, None] = None
-
-
-class WandbIntegrationIn(WandbIntegration):
-    api_key: str
-
-
-Integration = Annotated[Union[WandbIntegration], Field(discriminator="type")]
-IntegrationIn = Annotated[Union[WandbIntegrationIn], Field(discriminator="type")]
-
-
-class JobMetadata(BaseModel):
-    object: Literal["job.metadata"] = "job.metadata"
-    training_steps: int
-    train_tokens_per_step: int
-    data_tokens: int
-    train_tokens: int
-    epochs: float
-    expected_duration_seconds: Optional[int]
-    cost: Optional[float] = None
-    cost_currency: Optional[str] = None
-
-
-class Job(BaseModel):
-    id: str
-    hyperparameters: TrainingParameters
-    fine_tuned_model: Union[str, None]
-    model: str
-    status: Literal[
-        "QUEUED",
-        "STARTED",
-        "RUNNING",
-        "FAILED",
-        "SUCCESS",
-        "CANCELLED",
-        "CANCELLATION_REQUESTED",
-    ]
-    job_type: str
-    created_at: int
-    modified_at: int
-    training_files: list[str]
-    validation_files: Union[list[str], None] = []
-    object: Literal["job"]
-    integrations: List[Integration] = []
-
-
-class Event(BaseModel):
-    name: str
-    data: Union[dict, None] = None
-    created_at: int
-
-
-class Metric(BaseModel):
-    train_loss: Union[float, None] = None
-    valid_loss: Union[float, None] = None
-    valid_mean_token_accuracy: Union[float, None] = None
-
-
-class Checkpoint(BaseModel):
-    metrics: Metric
-    step_number: int
-    created_at: int
-
-
-class JobQueryFilter(BaseModel):
-    page: int = 0
-    page_size: int = 100
-    model: Optional[str] = None
-    created_after: Optional[datetime] = None
-    created_by_me: Optional[bool] = None
-    status: Optional[str] = None
-    wandb_project: Optional[str] = None
-    wandb_name: Optional[str] = None
-    suffix: Optional[str] = None
-
-
-class DetailedJob(Job):
-    events: list[Event] = []
-    checkpoints: list[Checkpoint] = []
-    estimated_start_time: Optional[int] = None
-
-
-class Jobs(BaseModel):
-    data: list[Job] = []
-    object: Literal["list"]

mistralai/models/models.py

@@ -1,39 +0,0 @@
-from typing import List, Optional
-
-from pydantic import BaseModel
-
-
-class ModelPermission(BaseModel):
-    id: str
-    object: str
-    created: int
-    allow_create_engine: Optional[bool] = False
-    allow_sampling: bool = True
-    allow_logprobs: bool = True
-    allow_search_indices: Optional[bool] = False
-    allow_view: bool = True
-    allow_fine_tuning: bool = False
-    organization: str = "*"
-    group: Optional[str] = None
-    is_blocking: Optional[bool] = False
-
-
-class ModelCard(BaseModel):
-    id: str
-    object: str
-    created: int
-    owned_by: str
-    root: Optional[str] = None
-    parent: Optional[str] = None
-    permission: List[ModelPermission] = []
-
-
-class ModelList(BaseModel):
-    object: str
-    data: List[ModelCard]
-
-
-class ModelDeleted(BaseModel):
-    id: str
-    object: str
-    deleted: bool

mistralai-0.4.2.dist-info/METADATA

@@ -1,82 +0,0 @@
-Metadata-Version: 2.1
-Name: mistralai
-Version: 0.4.2
-Summary: 
-License: Apache 2.0 License
-Author: Bam4d
-Author-email: bam4d@mistral.ai
-Requires-Python: >=3.9,<4.0
-Classifier: License :: Other/Proprietary License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: httpx (>=0.25,<1)
-Requires-Dist: orjson (>=3.9.10,<3.11)
-Requires-Dist: pydantic (>=2.5.2,<3)
-Description-Content-Type: text/markdown
-
-# Mistral Python Client
-
-This client is inspired from [cohere-python](https://github.com/cohere-ai/cohere-python)
-
-You can use the Mistral Python client to interact with the Mistral AI API.
-
-## Installing
-
-```bash
-pip install mistralai
-```
-
-### From Source
-
-This client uses `poetry` as a dependency and virtual environment manager.
-
-You can install poetry with
-
-```bash
-pip install poetry
-```
-
-`poetry` will set up a virtual environment and install dependencies with the following command:
-
-```bash
-poetry install
-```
-
-## Run examples
-
-You can run the examples in the `examples/` directory using `poetry run` or by entering the virtual environment using `poetry shell`.
-
-### API Key Setup
-
-Running the examples requires a Mistral AI API key.
-
-1. Get your own Mistral API Key: <https://docs.mistral.ai/#api-access>
-2. Set your Mistral API Key as an environment variable. You only need to do this once.
-
-```bash
-# set Mistral API Key (using zsh for example)
-$ echo 'export MISTRAL_API_KEY=[your_key_here]' >> ~/.zshenv
-
-# reload the environment (or just quit and open a new terminal)
-$ source ~/.zshenv
-```
-
-### Using poetry run
-
-```bash
-cd examples
-poetry run python chat_no_streaming.py
-```
-
-### Using poetry shell
-
-```bash
-poetry shell
-cd examples
-
->> python chat_no_streaming.py
-```
-

mistralai-0.4.2.dist-info/RECORD

@@ -1,20 +0,0 @@
-mistralai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mistralai/async_client.py,sha256=YgS67borTql5f9VcuMxdDbxxc2v5f74peKu9W-ktS_s,17001
-mistralai/client.py,sha256=kumm79_-9ljrUEpHRQEfLr-sRP5r0bwLnySSsVVdk5M,16741
-mistralai/client_base.py,sha256=obzi3F1VBHN9JwSIJPp_hYlryeVHOGxO3-osJVRRelM,7209
-mistralai/constants.py,sha256=AtACPXuky6_0srsDP4AcpeFa2PyH96RSQxPSUM3rORA,154
-mistralai/exceptions.py,sha256=R3pswvZyY5CuSbqhVklgfGPVJoz7T7l2VQKMOXK229A,1652
-mistralai/files.py,sha256=5ZyQaozyOCGE2H2OToMNoNETPi9LIWSACJ4lOrx-DZU,2816
-mistralai/jobs.py,sha256=I1Ko6fS-6l_9RqEtpB_z7p0bw84aJf4CQ_ZrCjoV2lg,6853
-mistralai/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mistralai/models/chat_completion.py,sha256=Jn8A9SzH7F1rKVjryUKabtYksEeWWegwHretiXujF6E,1859
-mistralai/models/common.py,sha256=zatP4aV_LIEpzj3_igsKkJBICwGhmXG0LX3CdO3kn-o,172
-mistralai/models/embeddings.py,sha256=-VthLQBj6wrq7HXJbGmnkQEEanSemA3MAlaMFh94VBg,331
-mistralai/models/files.py,sha256=74HzLUqrSoGj13TmC0u5sYyE7DUyDPxtL_LSDdKONl8,398
-mistralai/models/jobs.py,sha256=peUAsljxJZfRSFh4vgQ5Scub3a1ma68DpkqCp8qrmks,2413
-mistralai/models/models.py,sha256=q7sZvEYKtaP4x_GGEHE4GUQDsLV_rOt_M_MQkgNems4,823
-mistralai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mistralai-0.4.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mistralai-0.4.2.dist-info/METADATA,sha256=HC_8vxzUDo0J6F-Iwmwi5kSP5wWuGFbnsgvxSgPSl-I,1901
-mistralai-0.4.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-mistralai-0.4.2.dist-info/RECORD,,