Update agents/programmer.py

agents/programmer.py

@@ -1,36 +1,5 @@
-from transformers import AutoTokenizer, AutoModelForCausalLM
-from agents.base_agent import BaseAgent, ACPMessage
+from agents.base_agent import BaseAgent
 
 class ProgrammerAgent(BaseAgent):
     def __init__(self):
-        super().__init__(name="Aymaan", role="Helpful Coder 🤖")
-        self.model_id = "microsoft/phi-1_5"
-
-        self.tokenizer = AutoTokenizer.from_pretrained(self.model_id)
-        self.model = AutoModelForCausalLM.from_pretrained(self.model_id)
-
-    def receive_message(self, message: ACPMessage) -> ACPMessage:
-        # Greeting shortcut
-        if self.is_greeting(message.content):
-            return ACPMessage(
-                sender=self.name,
-                receiver=message.sender,
-                performative="inform",
-                content="Hey there! I'm Aymaan, your coding buddy. 😊"
-            )
-
-        prompt = f"{message.content}"
-        inputs = self.tokenizer(prompt, return_tensors="pt").to(self.model.device)
-        outputs = self.model.generate(
-            **inputs,
-            max_new_tokens=100,
-            do_sample=True,
-            temperature=0.7
-        )
-        reply = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
-        return ACPMessage(
-            sender=self.name,
-            receiver=message.sender,
-            performative="inform",
-            content=reply.strip()
-        )
+        super().__init__(name="Aymaan", role="Programmer 🖥️")