agent.py

# agent.py
import os
from langchain_huggingface import HuggingFaceEndpoint
from dotenv import load_dotenv

load_dotenv()
HF_TOKEN = os.getenv('HF_TOKEN')

class Agent:
    def __init__(self, name: str):
        self._name = name
        self._persona = ""
        self._model = HuggingFaceEndpoint(
            repo_id="mistralai/Mistral-7B-Instruct-v0.3",
            huggingfacehub_api_token=HF_TOKEN,
            temperature=0.7,
            max_new_tokens=150,
        )

    @property
    def name(self):
        return self._name

    @property
    def persona(self):
        return self._persona

    @persona.setter
    def persona(self, value: str):
        self._persona = value

    def execute(self, task: str) -> str:
        try:
            response = self._model.invoke(f"{self._persona}\n\nUser: {task}\nAssistant:")
            return response
        except Exception as e:
            return f"An error occurred: {str(e)}"