Add type to agent response

openbraininstitute · Dec 3, 2024 · adfbff5 · adfbff5
1 parent 3cd5665
commit adfbff5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/virtual_labs/domain/user.py b/virtual_labs/domain/user.py
@@ -60,3 +60,4 @@ class UserAgentResponse(BaseModel):
     family_name: str
     name: str
     createdAt: datetime
+    type: list[str]
diff --git a/virtual_labs/tests/agent/test_get_agent.py b/virtual_labs/tests/agent/test_get_agent.py
@@ -20,6 +20,7 @@ async def test_get_agent(async_test_client: AsyncClient) -> None:
         "given_name": user,
         "family_name": user,
         "name": f"{user} {user}",
+        "type": ["Agent", "Person"],
     }
 
     gotten_agent_response = response.json()["data"]