fix: add mistralé

Author: mpJunot

copyme-ai/mistral/README.md

@@ -0,0 +1,84 @@
+# MistralRephraser
+
+A simple Python class to rephrase sentences using the Mistral AI API. Perfect for generating dynamic feedback
+
+## Quick Setup
+
+1. **Add your API key** in a `.env` file at your project root:
+
+```
+MISTRAL_API_KEY=your_mistral_api_key
+```
+
+## Usage
+
+### 1. Import and create the rephraser
+
+```python
+from mistral import MistralRephraser
+
+rephraser = MistralRephraser()  # Uses key from .env
+# or
+# rephraser = MistralRephraser(api_key="your_mistral_api_key")
+```
+
+### 2. Rephrase a sentence
+
+```python
+sentence = "The cat is on the mat."
+instruction = "Rephrase this sentence in a more formal way."
+result = rephraser.rephrase(sentence, instruction)
+print(result)
+```
+or with json file
+
+```python
+# Example usage:
+rephraser = MistralRephraser()
+feedback_json = {
+    "phase": "shot_release",
+    "corrections": [
+        {"joint": "elbow", "correction": 20, "unit": "deg"},
+        {"joint": "knee", "correction": -10, "unit": "deg"}
+    ]
+}
+instruction = "Rephrase this feedback in a motivating and clear way for the user."
+print(rephraser.rephrase(feedback_json, instruction))
+```
+
+### 3. Change the model (optional)
+
+```python
+rephraser.set_model("mistral-large-latest")
+```
+
+### 4. Check if your API key is valid
+
+```python
+if not rephraser.is_api_key_valid():
+    print("Invalid API key!")
+```
+
+## Example: FastAPI Endpoint
+
+```python
+from fastapi import FastAPI, HTTPException
+from mistral import MistralRephraser
+
+app = FastAPI()
+rephraser = MistralRephraser()
+
+@app.post("/rephrase")
+def rephrase_endpoint(data: dict):
+    sentence = data["sentence"]
+    instruction = data["instruction"]
+    try:
+        result = rephraser.rephrase(sentence, instruction)
+        return {"result": result}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+```
+
+---
+
+**Ressource:** [https://github.com/mistralai/client-python](https://github.com/mistralai/client-python)

copyme-ai/mistral/metadata.json

@@ -0,0 +1,15 @@
+{
+  "phase": "shot_release",
+  "corrections": [
+    {
+      "joint": "elbow",
+      "correction": 20,
+      "unit": "deg"
+    },
+    {
+      "joint": "knee",
+      "correction": -10,
+      "unit": "deg"
+    }
+  ]
+}

copyme-ai/mistral/mistral.py

@@ -0,0 +1,58 @@
+from mistralai import Mistral, models
+import os
+from dotenv import load_dotenv
+import json
+
+load_dotenv()
+
+class MistralRephraser:
+    def __init__(self, api_key: str = None, model: str = "mistral-small-latest"):
+        self.api_key = api_key or os.getenv("MISTRAL_API_KEY")
+        if not self.api_key:
+            raise ValueError("MISTRAL_API_KEY is not set. Please add it to your .env file.")
+        self.model = model
+        self.client = Mistral(api_key=self.api_key)
+
+    def rephrase(self, original_sentence, instruction: str) -> str:
+        """
+        Rephrase a sentence or a JSON object according to the instruction.
+        :param original_sentence: str or dict (JSON-like)
+        :param instruction: str
+        :return: str
+        """
+        if isinstance(original_sentence, dict):
+            original_sentence_str = json.dumps(original_sentence, ensure_ascii=False, indent=2)
+        else:
+            original_sentence_str = str(original_sentence)
+
+        messages = [
+            {"role": "system", "content": instruction},
+            {"role": "user", "content": original_sentence_str}
+        ]
+        try:
+            response = self.client.chat.complete(
+                model=self.model,
+                messages=messages
+            )
+            return response.choices[0].message.content.strip()
+        except models.SDKError as e:
+            raise RuntimeError(f"Mistral API error: {e.message}") from e
+        except Exception as e:
+            raise RuntimeError(f"Unexpected error: {str(e)}") from e
+
+    def set_model(self, model: str):
+        """
+        Change the model used (e.g., 'mistral-large-latest').
+        """
+        self.model = model
+
+    def is_api_key_valid(self) -> bool:
+        """
+        Check if the API key is valid (by listing the models).
+        """
+        try:
+            self.client.models.list()
+            return True
+        except Exception:
+            return False
+