Refactors Python code for better readability

This commit refactors several Python functions to improve readability and maintainability by adjusting line breaks and spacing.

Specifically, it:
- Improves the formatting of conditional checks in the `_should_generate_vc` method of the `Agent` class.
- Enhances the readability of function signatures in `_replace_module_reference` and `_apply_vc_metadata`.
- Adjusts the formatting of `session_id` retrieval in `_execute_with_tracking`.
- Corrects the formatting for checking DID execution context and attributes in `_execute_with_tracking`.

Additionally, minor formatting adjustments were made in Go files related to CLI flags and response structures for consistency.

Author: santoshkumarradha

control-plane/internal/cli/root.go

@@ -14,16 +14,16 @@ import (
 )
 
 var (
-	cfgFile           string
-	verbose           bool
-	openBrowserFlag   bool
-	uiDevFlag         bool
-	backendOnlyFlag   bool
-	portFlag          int
-	noVCExecution     bool
-	forceVCExecution  bool
-	storageModeFlag   string
-	postgresURLFlag   string
+	cfgFile          string
+	verbose          bool
+	openBrowserFlag  bool
+	uiDevFlag        bool
+	backendOnlyFlag  bool
+	portFlag         int
+	noVCExecution    bool
+	forceVCExecution bool
+	storageModeFlag  string
+	postgresURLFlag  string
 )
 
 // NewRootCommand creates and returns the root Cobra command for the AgentField CLI.

control-plane/internal/handlers/workflow_dag.go

@@ -77,15 +77,15 @@ type WorkflowDAGLightweightNode struct {
 }
 
 type WorkflowDAGLightweightResponse struct {
-	RootWorkflowID string                        `json:"root_workflow_id"`
-	WorkflowStatus string                        `json:"workflow_status"`
-	WorkflowName   string                        `json:"workflow_name"`
-	SessionID      *string                       `json:"session_id,omitempty"`
-	ActorID        *string                       `json:"actor_id,omitempty"`
-	TotalNodes     int                           `json:"total_nodes"`
-	MaxDepth       int                           `json:"max_depth"`
-	Timeline       []WorkflowDAGLightweightNode  `json:"timeline"`
-	Mode           string                        `json:"mode"`
+	RootWorkflowID string                       `json:"root_workflow_id"`
+	WorkflowStatus string                       `json:"workflow_status"`
+	WorkflowName   string                       `json:"workflow_name"`
+	SessionID      *string                      `json:"session_id,omitempty"`
+	ActorID        *string                      `json:"actor_id,omitempty"`
+	TotalNodes     int                          `json:"total_nodes"`
+	MaxDepth       int                          `json:"max_depth"`
+	Timeline       []WorkflowDAGLightweightNode `json:"timeline"`
+	Mode           string                       `json:"mode"`
 }
 
 func GetWorkflowDAGHandler(storageProvider storage.StorageProvider) gin.HandlerFunc {

sdk/go/ai/response.go

@@ -40,9 +40,9 @@ type StreamChunk struct {
 
 // StreamDelta represents a delta in a streaming response.
 type StreamDelta struct {
-	Index        int           `json:"index"`
-	Delta        MessageDelta  `json:"delta"`
-	FinishReason *string       `json:"finish_reason"`
+	Index        int          `json:"index"`
+	Delta        MessageDelta `json:"delta"`
+	FinishReason *string      `json:"finish_reason"`
 }
 
 // MessageDelta represents the incremental message content.

sdk/python/agentfield/agent.py

@@ -886,8 +886,14 @@ def _effective_component_vc_setting(
             return overrides[component_id]
         return self._agent_vc_default()
 
-    def _should_generate_vc(self, component_id: str, overrides: Dict[str, bool]) -> bool:
-        if not self.did_enabled or not self.vc_generator or not self.vc_generator.is_enabled():
+    def _should_generate_vc(
+        self, component_id: str, overrides: Dict[str, bool]
+    ) -> bool:
+        if (
+            not self.did_enabled
+            or not self.vc_generator
+            or not self.vc_generator.is_enabled()
+        ):
             return False
         return self._effective_component_vc_setting(component_id, overrides)
 
@@ -1172,7 +1178,6 @@ async def run_reasoner() -> Any:
 
                 return await run_reasoner()
 
-
             # 🔥 ENHANCED: Comprehensive function replacement for unified tracking
             original_func = func
 
@@ -1773,7 +1778,9 @@ def include_router(
             router._attach_agent(self)
             normalized_prefix = prefix.rstrip("/") if prefix else ""
 
-            def _replace_module_reference(original_func: Callable, tracked_func: Callable) -> None:
+            def _replace_module_reference(
+                original_func: Callable, tracked_func: Callable
+            ) -> None:
                 module_name = getattr(original_func, "__module__", None)
                 attr_name = getattr(original_func, "__name__", None)
                 if not module_name or not attr_name:

sdk/python/agentfield/client.py

@@ -326,7 +326,9 @@ def get_nodes(self) -> Dict[str, Any]:
         response.raise_for_status()  # Raise an exception for bad status codes
         return response.json()
 
-    def _apply_vc_metadata(self, registration_data: Dict[str, Any], vc_metadata: Optional[Dict[str, Any]]) -> None:
+    def _apply_vc_metadata(
+        self, registration_data: Dict[str, Any], vc_metadata: Optional[Dict[str, Any]]
+    ) -> None:
         """Attach VC metadata to the registration payload if supplied."""
         if not vc_metadata:
             return

sdk/python/agentfield/decorators.py

@@ -177,19 +177,16 @@ async def _execute_with_tracking(func: Callable, *args, **kwargs) -> Any:
     )
     if agent_has_did:
         try:
-            session_id = (
-                execution_context.session_id or execution_context.workflow_id
-            )
+            session_id = execution_context.session_id or execution_context.workflow_id
             did_execution_context = agent_instance.did_manager.create_execution_context(
                 execution_context.execution_id,
                 execution_context.workflow_id,
                 session_id,
                 "agent",
                 reasoner_name,
             )
-            if (
-                did_execution_context
-                and hasattr(agent_instance, "_populate_execution_context_with_did")
+            if did_execution_context and hasattr(
+                agent_instance, "_populate_execution_context_with_did"
             ):
                 agent_instance._populate_execution_context_with_did(
                     execution_context, did_execution_context

sdk/python/tests/test_agent_networking.py

@@ -212,12 +212,8 @@ async def critical_reasoner(text: str) -> dict:
     def bulk_skill(amount: int) -> int:
         return amount
 
-    assert agent._should_generate_vc(
-        "critical", agent._reasoner_vc_overrides
-    ) is True
-    assert agent._should_generate_vc(
-        "fallback", agent._reasoner_vc_overrides
-    ) is False
+    assert agent._should_generate_vc("critical", agent._reasoner_vc_overrides) is True
+    assert agent._should_generate_vc("fallback", agent._reasoner_vc_overrides) is False
     assert agent._should_generate_vc("bulk", agent._skill_vc_overrides) is True
 
     metadata = agent._build_vc_metadata()

sdk/python/tests/test_async_execution.py

@@ -8,7 +8,9 @@
 
 @pytest.mark.asyncio
 async def test_reasoner_async_mode_sends_status(monkeypatch):
-    agent = Agent(node_id="test-agent", agentfield_server="http://control", auto_register=False)
+    agent = Agent(
+        node_id="test-agent", agentfield_server="http://control", auto_register=False
+    )
 
     @agent.reasoner()
     async def echo(value: int) -> dict:
@@ -51,7 +53,9 @@ async def fake_request(self, method, url, **kwargs):
 
 @pytest.mark.asyncio
 async def test_post_execution_status_retries(monkeypatch):
-    agent = Agent(node_id="test-agent", agentfield_server="http://control", auto_register=False)
+    agent = Agent(
+        node_id="test-agent", agentfield_server="http://control", auto_register=False
+    )
 
     attempts = {"count": 0}
 

sdk/python/tests/test_client.py

@@ -366,5 +366,6 @@ def on_request(method, url, **kwargs):
     assert posted[0][1].endswith("/nodes/register")
     body = posted[0][2]
     assert (
-        body["metadata"]["custom"]["vc_generation"]["reasoner_overrides"]["foo"] is False
+        body["metadata"]["custom"]["vc_generation"]["reasoner_overrides"]["foo"]
+        is False
     )