fix: add health protocol to missing agents (#87)

* fix: add health protocol to missing agents

* minor fixes

Author: Alejandro-Morales

6-deployed-agents/chained/blog-creator-agent/agent.py

@@ -1,5 +1,6 @@
 import os
-from uagents import Agent, Context
+from enum import Enum
+from uagents import Agent, Context, Model
 from uagents.experimental.chat_agent import ChatAgent
 from uagents.experimental.quota import QuotaProtocol, RateLimit
 from models import (
@@ -12,6 +13,7 @@
 )
 
 AGENT_SEED = os.getenv("AGENT_SEED", "your-blog-agent-seed")
+AGENT_NAME = os.getenv("AGENT_NAME", "Blog Agent")
 TOPIC_AGENT_ADDRESS = os.getenv("topic-agent-address")
 AI_AGENT_ADDRESS = os.getenv("ai-agent-address")
 
@@ -87,5 +89,35 @@ async def handle_request(ctx: Context, sender: str, msg: BlogRequest):
 
 agent.include(proto)
 
+
+# Health Check code
+class HealthCheck(Model):
+    pass
+
+
+class HealthStatus(str, Enum):
+    HEALTHY = "healthy"
+    UNHEALTHY = "unhealthy"
+
+
+class AgentHealth(Model):
+    agent_name: str
+    status: HealthStatus
+
+
+health_protocol = QuotaProtocol(
+    storage_reference=agent.storage, name="HealthProtocol", version="0.1.0"
+)
+
+
+@health_protocol.on_message(HealthCheck, replies={AgentHealth})
+async def handle_health_check(ctx: Context, sender: str, msg: HealthCheck):
+    await ctx.send(
+        sender, AgentHealth(agent_name=AGENT_NAME, status=HealthStatus.HEALTHY)
+    )
+
+
+agent.include(health_protocol, publish_manifest=True)
+
 if __name__ == "__main__":
     agent.run()

6-deployed-agents/chained/topic-extractor-agent/agent.py

@@ -1,4 +1,5 @@
 import os
+from enum import Enum
 from uagents import Agent, Context, Model
 from uagents.experimental.chat_agent import ChatAgent
 from uagents.experimental.quota import QuotaProtocol, RateLimit
@@ -12,6 +13,7 @@
 )
 
 AGENT_SEED = os.getenv("AGENT_SEED", "your-topic-agent-seed")
+AGENT_NAME = os.getenv("AGENT_NAME", "Topic Agent")
 POSTS_AGENT_ADDRESS = os.getenv("posts-agent-address")
 OPEN_AI_AGENT_ADDRESS = os.getenv("openai-agent-address")
 
@@ -81,5 +83,35 @@ async def handle_request(ctx: Context, sender: str, msg: TopicAnalysisRequest):
 
 agent.include(proto)
 
+
+# Health Check code
+class HealthCheck(Model):
+    pass
+
+
+class HealthStatus(str, Enum):
+    HEALTHY = "healthy"
+    UNHEALTHY = "unhealthy"
+
+
+class AgentHealth(Model):
+    agent_name: str
+    status: HealthStatus
+
+
+health_protocol = QuotaProtocol(
+    storage_reference=agent.storage, name="HealthProtocol", version="0.1.0"
+)
+
+
+@health_protocol.on_message(HealthCheck, replies={AgentHealth})
+async def handle_health_check(ctx: Context, sender: str, msg: HealthCheck):
+    await ctx.send(
+        sender, AgentHealth(agent_name=AGENT_NAME, status=HealthStatus.HEALTHY)
+    )
+
+
+agent.include(health_protocol, publish_manifest=True)
+
 if __name__ == "__main__":
     agent.run()

6-deployed-agents/utility/post-extractor-agent/agent.py

@@ -1,8 +1,10 @@
 import os
 import requests
+from enum import Enum
 from datetime import datetime
 from uagents import Agent, Context, Model, Protocol
 from uagents.experimental.chat_agent import ChatAgent
+from uagents.experimental.quota import QuotaProtocol, RateLimit
 
 class RedditPostsRequest(Model):
     limit: int
@@ -20,6 +22,7 @@ class RedditPostsResponse(Model):
 
 
 AGENT_SEED = os.getenv("AGENT_SEED", "your-post-agent-seed")
+AGENT_NAME = os.getenv("AGENT_NAME", "Post Extractor Agent")
 REDDIT_ID = os.getenv("REDDIT_ID")
 REDDIT_SECRET = os.getenv("REDDIT_SECRET")
 REDDIT_USER = os.getenv("REDDIT_USER")
@@ -76,7 +79,6 @@ async def handle_request(ctx: Context, sender: str, msg: RedditPostsRequest):
                 title=post_data["title"],
                 author=post_data["author"],
                 url=post_data["url"],
-                created=datetime.fromtimestamp(post_data["created_utc"]).isoformat(),
                 content=post_data["selftext"] if post_data["selftext"] else "[No text content]",
             )
 
@@ -94,5 +96,35 @@ async def handle_request(ctx: Context, sender: str, msg: RedditPostsRequest):
 
 agent.include(proto)
 
+
+# Health Check code
+class HealthCheck(Model):
+    pass
+
+
+class HealthStatus(str, Enum):
+    HEALTHY = "healthy"
+    UNHEALTHY = "unhealthy"
+
+
+class AgentHealth(Model):
+    agent_name: str
+    status: HealthStatus
+
+
+health_protocol = QuotaProtocol(
+    storage_reference=agent.storage, name="HealthProtocol", version="0.1.0"
+)
+
+
+@health_protocol.on_message(HealthCheck, replies={AgentHealth})
+async def handle_health_check(ctx: Context, sender: str, msg: HealthCheck):
+    await ctx.send(
+        sender, AgentHealth(agent_name=AGENT_NAME, status=HealthStatus.HEALTHY)
+    )
+
+
+agent.include(health_protocol, publish_manifest=True)
+
 if __name__ == "__main__":
     agent.run()