fix: apply ruff format to composio_cognee.py

Author: Patel230

src/hawk/composio_cognee.py

@@ -20,6 +20,7 @@
 
 # --- Composio Types ---
 
+
 @dataclass
 class ComposioTool:
     name: str
@@ -50,6 +51,7 @@ class ComposioCredential:
 
 # --- Cognee Entry Types ---
 
+
 @dataclass
 class CogneeQAEntry:
     question: str
@@ -114,6 +116,7 @@ class CogneeImproveResult:
 
 # --- Composio Methods ---
 
+
 def search_composio_tools(
     base_url: str = DEFAULT_BASE_URL,
     query: str = "",
@@ -191,6 +194,7 @@ def _do_list() -> list[ComposioCredential]:
 
 # --- Cognee Memory Methods ---
 
+
 def remember_qa(
     base_url: str = DEFAULT_BASE_URL,
     qa: CogneeQAEntry | None = None,
@@ -244,18 +248,21 @@ def remember_trace(
 
     def _do_remember() -> str:
         with httpx.Client(timeout=timeout) as client:
-            resp = client.post(f"{base_url}/cognee/trace", json={
-                "origin_function": trace.origin_function,
-                "status": trace.status,
-                "method_params": trace.method_params,
-                "method_return_value": trace.method_return_value,
-                "memory_query": trace.memory_query,
-                "memory_context": trace.memory_context,
-                "error_message": trace.error_message,
-                "session_id": trace.session_id,
-                "project": trace.project,
-                "source_agent": trace.source_agent,
-            })
+            resp = client.post(
+                f"{base_url}/cognee/trace",
+                json={
+                    "origin_function": trace.origin_function,
+                    "status": trace.status,
+                    "method_params": trace.method_params,
+                    "method_return_value": trace.method_return_value,
+                    "memory_query": trace.memory_query,
+                    "memory_context": trace.memory_context,
+                    "error_message": trace.error_message,
+                    "session_id": trace.session_id,
+                    "project": trace.project,
+                    "source_agent": trace.source_agent,
+                },
+            )
             if resp.status_code != 200:
                 raise parse_error(resp)
             return str(resp.json().get("id", ""))
@@ -279,14 +286,17 @@ def remember_feedback(
 
     def _do_remember() -> str:
         with httpx.Client(timeout=timeout) as client:
-            resp = client.post(f"{base_url}/cognee/feedback", json={
-                "target_node_id": feedback.target_node_id,
-                "feedback_text": feedback.feedback_text,
-                "score": feedback.score,
-                "project": feedback.project,
-                "session_id": feedback.session_id,
-                "source_agent": feedback.source_agent,
-            })
+            resp = client.post(
+                f"{base_url}/cognee/feedback",
+                json={
+                    "target_node_id": feedback.target_node_id,
+                    "feedback_text": feedback.feedback_text,
+                    "score": feedback.score,
+                    "project": feedback.project,
+                    "session_id": feedback.session_id,
+                    "source_agent": feedback.source_agent,
+                },
+            )
             if resp.status_code != 200:
                 raise parse_error(resp)
             return str(resp.json().get("id", ""))
@@ -310,18 +320,21 @@ def remember_skill_run(
 
     def _do_remember() -> str:
         with httpx.Client(timeout=timeout) as client:
-            resp = client.post(f"{base_url}/cognee/skill_run", json={
-                "skill_name": skill_run.skill_name,
-                "skill_version": skill_run.skill_version,
-                "params": skill_run.params,
-                "result": skill_run.result,
-                "status": skill_run.status,
-                "duration_ms": skill_run.duration_ms,
-                "error": skill_run.error,
-                "project": skill_run.project,
-                "session_id": skill_run.session_id,
-                "source_agent": skill_run.source_agent,
-            })
+            resp = client.post(
+                f"{base_url}/cognee/skill_run",
+                json={
+                    "skill_name": skill_run.skill_name,
+                    "skill_version": skill_run.skill_version,
+                    "params": skill_run.params,
+                    "result": skill_run.result,
+                    "status": skill_run.status,
+                    "duration_ms": skill_run.duration_ms,
+                    "error": skill_run.error,
+                    "project": skill_run.project,
+                    "session_id": skill_run.session_id,
+                    "source_agent": skill_run.source_agent,
+                },
+            )
             if resp.status_code != 200:
                 raise parse_error(resp)
             return str(resp.json().get("id", ""))
@@ -349,15 +362,18 @@ def improve(
 
     def _do_improve() -> CogneeImproveResult:
         with httpx.Client(timeout=timeout) as client:
-            resp = client.post(f"{base_url}/cognee/improve", json={
-                "project": project,
-                "min_confidence": min_confidence,
-                "min_access_count": min_access_count,
-                "consolidate_duplicates": consolidate_duplicates,
-                "regenerate_summaries": regenerate_summaries,
-                "refresh_embeddings": refresh_embeddings,
-                "limit": limit,
-            })
+            resp = client.post(
+                f"{base_url}/cognee/improve",
+                json={
+                    "project": project,
+                    "min_confidence": min_confidence,
+                    "min_access_count": min_access_count,
+                    "consolidate_duplicates": consolidate_duplicates,
+                    "regenerate_summaries": regenerate_summaries,
+                    "refresh_embeddings": refresh_embeddings,
+                    "limit": limit,
+                },
+            )
             if resp.status_code != 200:
                 raise parse_error(resp)
             return CogneeImproveResult(**resp.json())
@@ -407,12 +423,15 @@ def recall_with_session(
 
     def _do_recall() -> str:
         with httpx.Client(timeout=timeout) as client:
-            resp = client.post(f"{base_url}/cognee/recall_session", json={
-                "query": query,
-                "session_id": session_id,
-                "project": project,
-                "limit": limit,
-            })
+            resp = client.post(
+                f"{base_url}/cognee/recall_session",
+                json={
+                    "query": query,
+                    "session_id": session_id,
+                    "project": project,
+                    "limit": limit,
+                },
+            )
             if resp.status_code != 200:
                 raise parse_error(resp)
             return str(resp.json().get("context", ""))