import ollama
import os

class OllamaClient():
    def __init__(self):
        self.client = ollama.Client(host=os.getenv("ENDPOINT_OLLAMA", "https://ollamamodel.matitos.org"))
    
    def _get_default_model(self):
        return "llama3.2:3b"

    def get_models(self):
        models = sorted([m.model for m in self.client.list().models])
        if (self._get_default_model() in models):
            return [self._get_default_model()] + [m for m in models if m != self._get_default_model()]
        else:
            return models
    
    def get_prompt(self):
        return ("Rewrite the text below into a clear and concise summary of one paragraph maximum, presenting the key points as if they are newly written insights. "
                "Do not mention or reference the original text, its source, or any phrases like 'According to' or 'The text states'. "
                "Write in a natural, standalone format that feels like an original explanation. "
                "Keep it brief, engaging, informative, in the style of a news article: \n"
        )