File size: 890 Bytes
2dfe763
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import json
import requests
from typing import Dict, List, Any

class EndpointHandler:
    def __init__(self, path=""):
        # URL de ton modèle sur Featherless
        self.api_url = "https://api.featherless.ai/v1/chat/completions"
        self.model_id = "RAANA-IA/Gheya-med"

    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
        inputs = data.pop("inputs", data)
        
        # Préparation de la requête pour Featherless
        payload = {
            "model": self.model_id,
            "messages": [{"role": "user", "content": inputs}],
            "max_tokens": 150
        }
        
        # Note : Si tu as une clé API, il faudra l'ajouter dans les headers ici
        response = requests.post(self.api_url, json=payload)
        result = response.json()
        
        return [{"generated_text": result['choices'][0]['message']['content']}]