refactor: Format with nox

Author: batprem

examples/no-llm-framework/noxfile.py

@@ -37,7 +37,6 @@
 @nox.session(python=DEFAULT_PYTHON_VERSION)
 def format(session):
     """Format Python code using autoflake, pyupgrade, and ruff."""
-
     format_all = False
 
     if format_all:

examples/no-llm-framework/src/no_llm_framework/client/__main__.py

@@ -1,14 +1,20 @@
 import asyncio
-from typing import Any, Literal
+from typing import Literal
 from uuid import uuid4
 
 import asyncclick as click
 import httpx
 from a2a.client import A2ACardResolver, A2AClient
-from a2a.types import (Message, MessageSendParams, Part, Role,
-                       SendStreamingMessageRequest,
-                       SendStreamingMessageSuccessResponse,
-                       TaskStatusUpdateEvent, TextPart)
+from a2a.types import (
+    Message,
+    MessageSendParams,
+    Part,
+    Role,
+    SendStreamingMessageRequest,
+    SendStreamingMessageSuccessResponse,
+    TaskStatusUpdateEvent,
+    TextPart,
+)
 
 
 @click.command()
@@ -41,13 +47,10 @@ async def a_main(
             )
         )
 
-
         if mode == 'completion':
             raise NotImplementedError('Completion mode not implemented')
 
-        streaming_request = SendStreamingMessageRequest(
-            params=message
-        )
+        streaming_request = SendStreamingMessageRequest(params=message)
         stream_response = client.send_message_streaming(streaming_request)
         async for chunk in stream_response:
             if isinstance(
@@ -58,10 +61,8 @@ async def a_main(
                     print(message.parts[0].root.text, end='', flush=True)
 
 
-
 def main() -> None:
     """Main function to run the A2A Repo Agent client."""
-
     asyncio.run(a_main())
 
 

examples/no-llm-framework/src/no_llm_framework/client/agent.py

@@ -37,7 +37,7 @@
     agent_answer_template = Template(f.read())
 
 
-def stream_llm(prompt: str) -> Generator[str, None, None]:
+def stream_llm(prompt: str) -> Generator[str]:
     """Stream LLM response.
 
     Args:
@@ -174,7 +174,6 @@ async def stream(self, question: str):
             else:
                 print('<Answer> tag not found')
         print(agent_answers)
-        return
 
 
 if __name__ == '__main__':

examples/no-llm-framework/src/no_llm_framework/client/constant.py

@@ -1,3 +1,3 @@
 import os
 
-GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')

examples/no-llm-framework/src/no_llm_framework/server/__main__.py

@@ -2,12 +2,20 @@
 import uvicorn
 from a2a.server.agent_execution import AgentExecutor
 from a2a.server.apps.starlette_app import A2AStarletteApplication
-from a2a.server.request_handlers.default_request_handler import \
-    DefaultRequestHandler
+from a2a.server.request_handlers.default_request_handler import (
+    DefaultRequestHandler,
+)
 from a2a.server.tasks import InMemoryTaskStore
-from a2a.types import (AgentAuthentication, AgentCapabilities, AgentCard,
-                       AgentSkill, GetTaskRequest, GetTaskResponse,
-                       SendMessageRequest, SendMessageResponse)
+from a2a.types import (
+    AgentAuthentication,
+    AgentCapabilities,
+    AgentCard,
+    AgentSkill,
+    GetTaskRequest,
+    GetTaskResponse,
+    SendMessageRequest,
+    SendMessageResponse,
+)
 
 from no_llm_framework.server.agent_executor import HelloWorldAgentExecutor
 

examples/no-llm-framework/src/no_llm_framework/server/agent.py

@@ -106,7 +106,7 @@ async def call_tool(self, tools: list[dict]) -> list[CallToolResult]:
             ]
         )
 
-    async def stream(self, question: str) -> AsyncGenerator[str, None]:
+    async def stream(self, question: str) -> AsyncGenerator[str]:
         called_tools = []
         for i in range(10):
             yield {

examples/no-llm-framework/src/no_llm_framework/server/agent_executor.py

@@ -1,9 +1,13 @@
 from a2a.server.agent_execution import AgentExecutor, RequestContext
 from a2a.server.events import EventQueue
-from a2a.types import (TaskArtifactUpdateEvent, TaskState, TaskStatus,
-                       TaskStatusUpdateEvent)
+from a2a.types import (
+    TaskArtifactUpdateEvent,
+    TaskState,
+    TaskStatus,
+    TaskStatusUpdateEvent,
+)
 from a2a.utils import new_agent_text_message, new_task, new_text_artifact
-from typing_extensions import override
+from typing import override
 
 from src.no_llm_framework.server.agent import Agent
 

examples/no-llm-framework/src/no_llm_framework/server/constant.py

@@ -1,3 +1,3 @@
 import os
 
-GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')