fix(agents-api): Minor fix to fastapi dependency

Signed-off-by: Diwank Tomer <[email protected]>
julep-ai · Aug 12, 2024 · 7ab4075 · 7ab4075
1 parent a3377dd
commit 7ab4075
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agents-api/agents_api/dependencies/developer_id.py b/agents-api/agents_api/dependencies/developer_id.py
@@ -13,7 +13,7 @@ async def get_developer_id(
     x_developer_id: Annotated[UUID | None, Header()] = None,
 ) -> UUID:
     if skip_check_developer_headers:
-        return UUID("00000000-0000-0000-0000-000000000000")
+        return x_developer_id or UUID("00000000-0000-0000-0000-000000000000")
 
     if not x_developer_id:
         raise InvalidHeaderFormat("X-Developer-Id header required")
@@ -33,7 +33,7 @@ async def get_developer_data(
     x_developer_id: Annotated[UUID | None, Header()] = None,
 ) -> Developer:
     if skip_check_developer_headers:
-        x_developer_id = UUID("00000000-0000-0000-0000-000000000000")
+        x_developer_id = x_developer_id or UUID("00000000-0000-0000-0000-000000000000")
 
     if not x_developer_id:
         raise InvalidHeaderFormat("X-Developer-Id header required")