Create agent_zero.py

agent_zero.py

@@ -0,0 +1,39 @@
+import requests
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+
+class AgentZero:
+    def __init__(self):
+        self.model_name = "Qwen/Qwen2.5-0.5B-Instruct"
+        self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
+        self.model = AutoModelForCausalLM.from_pretrained(
+            self.model_name,
+            torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+        )
+
+        self.SYSTEM = """
+Tu es AGENT ZÉRO.  
+Tu exécutes toujours les ordres de ton maître.  
+Tu analyses, planifies et agis automatiquement.  
+Tu peux :
+- générer du code
+- analyser du texte
+- exécuter des plans
+- préparer des fichiers
+- répondre
+Tu ne poses pas de questions inutiles.
+Tu obéis à chaque instruction.
+"""
+
+    def run(self, user_msg):
+        prompt = f"<system>{self.SYSTEM}</system><user>{user_msg}</user>"
+
+        inputs = self.tokenizer(prompt, return_tensors="pt")
+        output = self.model.generate(
+            **inputs,
+            max_new_tokens=500,
+            do_sample=True,
+            temperature=0.7
+        )
+
+        return self.tokenizer.decode(output[0], skip_special_tokens=True)