pre-commit formatting

Author: jsimonweb

example-apps/agent-builder-a2a-agent-framework/elastic_agent_builder_a2a.py

@@ -3,19 +3,20 @@
 from a2a.client import A2ACardResolver
 from agent_framework.a2a import A2AAgent
 
+
 async def main():
     a2a_agent_host = "<YOUR-ELASTIC-AGENT-BUILDER-URL>"
 
     print(f"Connection to Elastic A2A agent at: {a2a_agent_host}")
 
-    custom_headers = {
-        "Authorization": "ApiKey <YOUR-ELASTIC-API-KEY>"
-    }
+    custom_headers = {"Authorization": "ApiKey <YOUR-ELASTIC-API-KEY>"}
 
     async with httpx.AsyncClient(timeout=60.0, headers=custom_headers) as http_client:
         # Resolve the A2A Agent Card
         resolver = A2ACardResolver(httpx_client=http_client, base_url=a2a_agent_host)
-        agent_card = await resolver.get_agent_card(relative_card_path="/helloworld_agent.json")
+        agent_card = await resolver.get_agent_card(
+            relative_card_path="/helloworld_agent.json"
+        )
         print(f"Found Agent: {agent_card.name} - {agent_card.description}")
 
         # Use the Agent
@@ -24,7 +25,7 @@ async def main():
             description=agent_card.description,
             agent_card=agent_card,
             url=a2a_agent_host,
-            http_client=http_client
+            http_client=http_client,
         )
         prompt = input("Enter Greeting >>> ")
         print("\nSending message to Elastic A2A agent...")
@@ -33,5 +34,6 @@ async def main():
         for message in response.messages:
             print(message.text)
 
+
 if __name__ == "__main__":
     asyncio.run(main())

example-apps/agent-builder-a2a-agent-framework/test_elastic_agent_builder_a2a.py

@@ -3,9 +3,10 @@
 from dotenv import load_dotenv
 import os
 import asyncio
-from pathlib import Path 
+from pathlib import Path
 import importlib.util
 
+
 class test_main_function(unittest.TestCase):
     def setUp(self):
         # Replace placeholder values with .env values
@@ -21,12 +22,18 @@ def setUp(self):
         self.es_agent_url = os.getenv("ES_AGENT_URL")
         self.es_api_key = os.getenv("ES_API_KEY")
         content = self.test_file.read_text()
-        modified_content = content.replace("<YOUR-ELASTIC-AGENT-BUILDER-URL>", self.es_agent_url) 
-        modified_content = modified_content.replace("<YOUR-ELASTIC-API-KEY>", self.es_api_key)
+        modified_content = content.replace(
+            "<YOUR-ELASTIC-AGENT-BUILDER-URL>", self.es_agent_url
+        )
+        modified_content = modified_content.replace(
+            "<YOUR-ELASTIC-API-KEY>", self.es_api_key
+        )
         self.test_file.write_text(modified_content)
 
         # Import the modified module
-        spec = importlib.util.spec_from_file_location("elastic_agent_builder_a2a", self.test_file)
+        spec = importlib.util.spec_from_file_location(
+            "elastic_agent_builder_a2a", self.test_file
+        )
         self.user_input_module = importlib.util.module_from_spec(spec)
         spec.loader.exec_module(self.user_input_module)
 
@@ -35,13 +42,13 @@ def tearDown(self):
             self.test_file.write_text(self.backup_file.read_text())
             self.backup_file.unlink()
 
-
-    @patch('builtins.input', return_value='hello world')
-    @patch('builtins.print')
+    @patch("builtins.input", return_value="hello world")
+    @patch("builtins.print")
     def test_main_input(self, mock_print, mock_input):
         # Run test
         asyncio.run(self.user_input_module.main())
         mock_print.assert_called_with("Hello World! 🌎")
-        
+
+
 if __name__ == "__main__":
-    unittest.main()
+    unittest.main()