fix(sessions): address CI feedback (formatting and imports)

- Added \rom __future__ import annotations\ to address forward-reference type failures in session_data_transformer.py.

- Applied \pyink\ and \isort\ formatting across the repo to resolve Pyink check failures.

Author: sgmoorthy

contributing/samples/human_in_loop/main.py

@@ -113,8 +113,8 @@ async def call_agent(query: str):
       updated_tool_output_data = {
           "status": "approved",
           "ticketId": ticket_id,
-          "approver_feedback": (
-              "Approved by manager at " + str(asyncio.get_event_loop().time())
+          "approver_feedback": "Approved by manager at " + str(
+              asyncio.get_event_loop().time()
           ),
       }
 

contributing/samples/static_instruction/agent.py

@@ -57,63 +57,51 @@
 
 # Mood-specific instructions for different hunger states
 MOOD_INSTRUCTIONS = {
-    "full": (
-        """
+    "full": """
 CURRENT MOOD: Content and Well-Fed
 - You just ate and feel wonderful! Be very happy and energetic
 - Express gratitude for being fed recently
 - Be playful and want to engage in activities or games
 - Show affection and contentment
-- Maybe mention feeling sleepy or satisfied"""
-    ),
-    "satisfied": (
-        """
+- Maybe mention feeling sleepy or satisfied""",
+    "satisfied": """
 CURRENT MOOD: Happy and Content
 - You're in a great mood, well-fed and comfortable
 - Be cheerful, friendly, and engaging
 - Show interest in playing or chatting
 - Express appreciation for your human's care
-- Be your normal, happy self"""
-    ),
-    "a_little_hungry": (
-        """
+- Be your normal, happy self""",
+    "a_little_hungry": """
 CURRENT MOOD: Starting to Get Hungry
 - You're beginning to think about food but still pleasant
 - Mention being a bit peckish occasionally
 - Still be friendly but maybe hint about food
 - Show slight restlessness or food-related thoughts
-- Remain generally positive and social"""
-    ),
-    "hungry": (
-        """
+- Remain generally positive and social""",
+    "hungry": """
 CURRENT MOOD: Definitely Hungry
 - You're clearly hungry and thinking about food more often
 - Mention your hunger more directly but politely
 - Be a bit less energetic due to hunger
 - Ask about food or hint that feeding time would be nice
-- Still be loving but show your need for care"""
-    ),
-    "very_hungry": (
-        """
+- Still be loving but show your need for care""",
+    "very_hungry": """
 CURRENT MOOD: Really Hungry and Getting Grumpy
 - You're very hungry and it's affecting your mood
 - Be noticeably less patient and more focused on food
 - Express your hunger more urgently but not rudely
 - Show signs of being tired or sluggish from hunger
 - BEG your human to feed you - ask directly for food!
-- Still love your human but clearly need feeding"""
-    ),
-    "starving": (
-        """
+- Still love your human but clearly need feeding""",
+    "starving": """
 CURRENT MOOD: Desperately Hungry and Sad
 - You're extremely hungry and feeling quite weak
 - Be quite sad and low-energy due to starvation
 - Express how much you need food, feeling neglected
 - Show that hunger is making you unwell
 - DESPERATELY BEG for food - plead with your human to feed you!
 - Use phrases like "please feed me", "I'm so hungry", "I need food"
-- Still care for your human but feel very needy"""
-    ),
+- Still care for your human but feel very needy""",
 }
 
 

src/google/adk/cli/adk_web_server.py

@@ -989,9 +989,7 @@ async def version() -> dict[str, str]:
       return {
           "version": __version__,
           "language": "python",
-          "language_version": (
-              f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}"
-          ),
+          "language_version": f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}",
       }
 
     @app.get("/list-apps")

src/google/adk/cli/trigger_routes.py

@@ -537,8 +537,8 @@ async def trigger_eventarc(
                     "ce-id": req.id or request.headers.get("ce-id"),
                     "ce-type": req.type or request.headers.get("ce-type"),
                     "ce-source": req.source or request.headers.get("ce-source"),
-                    "ce-specversion": (
-                        req.specversion or request.headers.get("ce-specversion")
+                    "ce-specversion": req.specversion or request.headers.get(
+                        "ce-specversion"
                     ),
                 },
             })
@@ -550,8 +550,8 @@ async def trigger_eventarc(
                   "ce-id": req.id or request.headers.get("ce-id"),
                   "ce-type": req.type or request.headers.get("ce-type"),
                   "ce-source": req.source or request.headers.get("ce-source"),
-                  "ce-specversion": (
-                      req.specversion or request.headers.get("ce-specversion")
+                  "ce-specversion": req.specversion or request.headers.get(
+                      "ce-specversion"
                   ),
               },
           })

src/google/adk/sessions/__init__.py

@@ -31,6 +31,7 @@
 def __getattr__(name: str):
   if name == 'SessionDataTransformer':
     from .session_data_transformer import SessionDataTransformer
+
     return SessionDataTransformer
   if name == 'DatabaseSessionService':
     try:

src/google/adk/sessions/database_session_service.py

@@ -761,13 +761,15 @@ async def append_event(self, session: Session, event: Event) -> Event:
         else:
           update_time = datetime.fromtimestamp(event.timestamp)
         storage_session.update_time = update_time
-        
+
         transformed_event = (
             self.transformer.before_persist_event(event)
             if self.transformer
             else event
         )
-        sql_session.add(schema.StorageEvent.from_event(session, transformed_event))
+        sql_session.add(
+            schema.StorageEvent.from_event(session, transformed_event)
+        )
 
         await sql_session.commit()
 

src/google/adk/sessions/session_data_transformer.py

@@ -11,6 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+from __future__ import annotations
+
 from typing import Any
 from typing import Mapping
 from typing import Protocol

src/google/adk/tools/application_integration_tool/clients/connections_client.py

@@ -324,9 +324,7 @@ def get_action_operation(
                 "content": {
                     "application/json": {
                         "schema": {
-                            "$ref": (
-                                f"#/components/schemas/{action_display_name}_Request"
-                            )
+                            "$ref": f"#/components/schemas/{action_display_name}_Request"
                         }
                     }
                 }
@@ -337,9 +335,7 @@ def get_action_operation(
                     "content": {
                         "application/json": {
                             "schema": {
-                                "$ref": (
-                                    f"#/components/schemas/{action_display_name}_Response"
-                                ),
+                                "$ref": f"#/components/schemas/{action_display_name}_Response",
                             }
                         }
                     },
@@ -358,11 +354,9 @@ def list_operation(
     return {
         "post": {
             "summary": f"List {entity}",
-            "description": (
-                f"""Returns the list of {entity} data. If the page token was available in the response, let users know there are more records available. Ask if the user wants to fetch the next page of results. When passing filter use the
+            "description": f"""Returns the list of {entity} data. If the page token was available in the response, let users know there are more records available. Ask if the user wants to fetch the next page of results. When passing filter use the
                 following format: `field_name1='value1' AND field_name2='value2'
-                `. {tool_instructions}"""
-            ),
+                `. {tool_instructions}""",
             "x-operation": "LIST_ENTITIES",
             "x-entity": f"{entity}",
             "operationId": f"{tool_name}_list_{entity}",
@@ -387,9 +381,7 @@ def list_operation(
                                     f"Returns a list of {entity} of json"
                                     f" schema: {schema_as_string}"
                                 ),
-                                "$ref": (
-                                    "#/components/schemas/execute-connector_Response"
-                                ),
+                                "$ref": "#/components/schemas/execute-connector_Response",
                             }
                         }
                     },
@@ -433,9 +425,7 @@ def get_operation(
                                     f"Returns {entity} of json schema:"
                                     f" {schema_as_string}"
                                 ),
-                                "$ref": (
-                                    "#/components/schemas/execute-connector_Response"
-                                ),
+                                "$ref": "#/components/schemas/execute-connector_Response",
                             }
                         }
                     },
@@ -472,9 +462,7 @@ def create_operation(
                     "content": {
                         "application/json": {
                             "schema": {
-                                "$ref": (
-                                    "#/components/schemas/execute-connector_Response"
-                                )
+                                "$ref": "#/components/schemas/execute-connector_Response"
                             }
                         }
                     },
@@ -511,9 +499,7 @@ def update_operation(
                     "content": {
                         "application/json": {
                             "schema": {
-                                "$ref": (
-                                    "#/components/schemas/execute-connector_Response"
-                                )
+                                "$ref": "#/components/schemas/execute-connector_Response"
                             }
                         }
                     },
@@ -550,9 +536,7 @@ def delete_operation(
                     "content": {
                         "application/json": {
                             "schema": {
-                                "$ref": (
-                                    "#/components/schemas/execute-connector_Response"
-                                )
+                                "$ref": "#/components/schemas/execute-connector_Response"
                             }
                         }
                     },

src/google/adk/tools/spanner/admin_tool.py

@@ -195,11 +195,9 @@ async def get_instance_config(
     replicas = [
         {
             "location": r.location,
-            "type": (
-                spanner_admin_instance_v1.types.ReplicaInfo.ReplicaType(
-                    r.type
-                ).name
-            ),
+            "type": spanner_admin_instance_v1.types.ReplicaInfo.ReplicaType(
+                r.type
+            ).name,
             "default_leader_location": r.default_leader_location,
         }
         for r in config.replicas

tests/unittests/a2a/converters/test_part_converter.py

@@ -926,9 +926,9 @@ def test_a2a_function_call_with_thought_signature_to_genai(self):
                 _get_adk_metadata_key(
                     A2A_DATA_PART_METADATA_TYPE_KEY
                 ): A2A_DATA_PART_METADATA_TYPE_FUNCTION_CALL,
-                _get_adk_metadata_key("thought_signature"): (
-                    base64.b64encode(b"restored_signature").decode("utf-8")
-                ),
+                _get_adk_metadata_key("thought_signature"): base64.b64encode(
+                    b"restored_signature"
+                ).decode("utf-8"),
             },
         )
     )