refactor: Lint agents-api (CI)

julep-ai · Oct 4, 2024 · c3afa8f · c3afa8f
1 parent 9decd1f
commit c3afa8f
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 7 deletions.
diff --git a/agents-api/agents_api/dependencies/query_filter.py b/agents-api/agents_api/dependencies/query_filter.py
@@ -20,6 +20,7 @@ def convert_value(value: str) -> Any:
             continue
     return value
 
+
 def create_filter_extractor(prefix: str = "filter") -> Callable[[Request], FilterModel]:
     """
     Creates a dependency function to extract filter parameters with a given prefix.
@@ -47,7 +48,7 @@ def extract_filters(request: Request) -> FilterModel:
         filters: dict[str, Any] = {}
         for key, value in request.query_params.items():
             if key.startswith(prefix):
-                filter_key = key[len(prefix):]
+                filter_key = key[len(prefix) :]
                 filters[filter_key] = convert_value(value)
 
         return FilterModel(**filters)

diff --git a/agents-api/agents_api/routers/agents/list_agents.py b/agents-api/agents_api/routers/agents/list_agents.py
@@ -17,15 +17,13 @@ async def list_agents(
     # Example:
     # > ?metadata_filter.name=John&metadata_filter.age=30
     metadata_filter: Annotated[
-        FilterModel | None, 
-        Depends(create_filter_extractor("metadata_filter"))
+        FilterModel | None, Depends(create_filter_extractor("metadata_filter"))
     ],
     limit: int = 100,
     offset: int = 0,
     sort_by: Literal["created_at", "updated_at"] = "created_at",
     direction: Literal["asc", "desc"] = "desc",
 ) -> ListResponse[Agent]:
-
     agents = list_agents_query(
         developer_id=x_developer_id,
         limit=limit,

diff --git a/agents-api/agents_api/routers/docs/list_docs.py b/agents-api/agents_api/routers/docs/list_docs.py
@@ -16,7 +16,9 @@
 async def list_user_docs(
     x_developer_id: Annotated[UUID, Depends(get_developer_id)],
     user_id: UUID,
-    metadata_filter: Annotated[FilterModel | None, Depends(create_filter_extractor("metadata_filter"))],
+    metadata_filter: Annotated[
+        FilterModel | None, Depends(create_filter_extractor("metadata_filter"))
+    ],
     limit: int = 100,
     offset: int = 0,
     sort_by: Literal["created_at", "updated_at"] = "created_at",
@@ -40,7 +42,9 @@ async def list_user_docs(
 async def list_agent_docs(
     x_developer_id: Annotated[UUID, Depends(get_developer_id)],
     agent_id: UUID,
-    metadata_filter: Annotated[FilterModel | None, Depends(create_filter_extractor("metadata_filter"))],
+    metadata_filter: Annotated[
+        FilterModel | None, Depends(create_filter_extractor("metadata_filter"))
+    ],
     limit: int = 100,
     offset: int = 0,
     sort_by: Literal["created_at", "updated_at"] = "created_at",

diff --git a/agents-api/agents_api/routers/sessions/list_sessions.py b/agents-api/agents_api/routers/sessions/list_sessions.py
@@ -15,7 +15,9 @@
 @router.get("/sessions", tags=["sessions"])
 async def list_sessions(
     x_developer_id: Annotated[UUID, Depends(get_developer_id)],
-    metadata_filter: Annotated[FilterModel | None, Depends(create_filter_extractor("metadata_filter"))],
+    metadata_filter: Annotated[
+        FilterModel | None, Depends(create_filter_extractor("metadata_filter"))
+    ],
     limit: int = 100,
     offset: int = 0,
     sort_by: Literal["created_at", "updated_at"] = "created_at",