feat(backend): Add subscription_id in public GET route

Libertai · Nov 26, 2024 · 3597c80 · 3597c80
1 parent 9263304
commit 3597c80
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/backend/src/interfaces/agent.py b/backend/src/interfaces/agent.py
@@ -25,12 +25,12 @@ class UpdateAgentResponse(BaseModel):
 
 class PublicAgentData(BaseModel):
     id: str
+    subscription_id: str
     vm_hash: str | None
     last_update: int
 
 
 class Agent(PublicAgentData):
-    subscription_id: str
     encrypted_secret: str
     tags: list[str]
 

diff --git a/backend/src/main.py b/backend/src/main.py
@@ -83,7 +83,10 @@ async def get_agent_public_info(agent_id: str) -> GetAgentResponse:
     agent = agents[0]
 
     return GetAgentResponse(
-        id=agent.id, vm_hash=agent.vm_hash, last_update=agent.last_update
+        id=agent.id,
+        vm_hash=agent.vm_hash,
+        last_update=agent.last_update,
+        subscription_id=agent.subscription_id,
     )