Create programmer.py
Browse files- agents/programmer.py +34 -0
agents/programmer.py
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# agents/programmer.py
|
2 |
+
|
3 |
+
from agents.base_agent import BaseAgent, ACPMessage
|
4 |
+
import random
|
5 |
+
|
6 |
+
class ProgrammerAgent(BaseAgent):
|
7 |
+
def __init__(self):
|
8 |
+
super().__init__(name="CodeBot", role="Programmer")
|
9 |
+
|
10 |
+
def receive_message(self, message: ACPMessage) -> ACPMessage:
|
11 |
+
# Basic behavior for now — add AI model later
|
12 |
+
if message.performative == "request":
|
13 |
+
response_content = random.choice([
|
14 |
+
"Sure, here's a simple Python function to reverse a list:\n```python\ndef reverse_list(lst): return lst[::-1]```",
|
15 |
+
"Okay! Here's a sorting example:\n```python\ndef sort_list(lst): return sorted(lst)```",
|
16 |
+
"Let me help with that. To calculate factorial:\n```python\ndef factorial(n): return 1 if n==0 else n*factorial(n-1)```"
|
17 |
+
])
|
18 |
+
return self.create_message(
|
19 |
+
receiver=message.sender,
|
20 |
+
performative="inform",
|
21 |
+
content=response_content
|
22 |
+
)
|
23 |
+
elif message.performative == "inform":
|
24 |
+
return self.create_message(
|
25 |
+
receiver=message.sender,
|
26 |
+
performative="acknowledge",
|
27 |
+
content="Thanks for the update. Let me know if you need help."
|
28 |
+
)
|
29 |
+
else:
|
30 |
+
return self.create_message(
|
31 |
+
receiver=message.sender,
|
32 |
+
performative="refuse",
|
33 |
+
content="Sorry, I didn’t understand your request."
|
34 |
+
)
|