Upload folder using huggingface_hub

core/a2a_helpers.py

@@ -25,11 +25,36 @@ def parse_a2a_response(result: Any, output_class: type[BaseModel]) -> Optional[B
     """
     # Handle string responses
     if isinstance(result, str):
-        try:
-            result = json.loads(result)
-        except Exception as e:
-            print(f"Failed to parse string result as JSON: {e}")
-            return None
+        # First check if it's the Status/Message/TaskId format
+        if "Status:" in result and "Message:" in result:
+            try:
+                # Extract the JSON from the Message: line
+                lines = result.strip().split('\n')
+                for i, line in enumerate(lines):
+                    if line.startswith("Message:"):
+                        # Get the JSON part after "Message: "
+                        json_str = line[8:].strip()
+                        # If the JSON continues on multiple lines, collect them
+                        if i + 1 < len(lines) and not lines[i + 1].startswith(("TaskId:", "Timestamp:")):
+                            # Continue collecting lines until we hit TaskId or Timestamp
+                            json_lines = [json_str]
+                            for j in range(i + 1, len(lines)):
+                                if lines[j].startswith(("TaskId:", "Timestamp:")):
+                                    break
+                                json_lines.append(lines[j])
+                            json_str = '\n'.join(json_lines)
+                        result = json.loads(json_str)
+                        break
+            except Exception as e:
+                # Silently fail - the debug output in get_pick will show the issue
+                return None
+        else:
+            # Try parsing as plain JSON
+            try:
+                result = json.loads(result)
+            except Exception as e:
+                # Not JSON format
+                return None
     
     # Handle direct output class instance
     if isinstance(result, output_class):
@@ -42,14 +67,13 @@ def parse_a2a_response(result: Any, output_class: type[BaseModel]) -> Optional[B
             try:
                 return output_class(**result)
             except Exception as e:
-                print(f"Failed to convert dict to {output_class.__name__}: {e}")
-                print(f"Dict keys: {list(result.keys())}")
+                # Conversion failed
+                pass
         
         # Handle A2A wrapper format
         if 'status' in result and 'message' in result.get('status', {}):
             return _extract_from_wrapper(result, output_class)
     
-    print(f"Unable to parse result of type {type(result)}")
     return None
 
 
@@ -68,8 +92,19 @@ def _extract_from_wrapper(wrapped_result: Dict, output_class: type[BaseModel]) -
 
 def extract_task_id(result: Any) -> Optional[str]:
     """Extract task_id from A2A response if present."""
+    # Handle string format with TaskId: line
+    if isinstance(result, str) and "TaskId:" in result:
+        lines = result.strip().split('\n')
+        for line in lines:
+            if line.startswith("TaskId:"):
+                task_id = line[7:].strip()
+                if task_id and task_id != "None":
+                    return task_id
+    
+    # Handle dict format
     if isinstance(result, dict) and 'task_id' in result:
         return result['task_id']
+    
     return None
 
 

core/dynamic_a2a_manager.py

@@ -273,7 +273,6 @@ BE LOUD! BE PROUD! BE UNFORGETTABLE! 🎯""",
                       previous_picks: List[str], round_num: int = 1) -> Optional[A2AOutput]:
         """Get a pick from an A2A agent."""
         if team_num not in self.agent_tools:
-            print(f"❌ Team {team_num} not found in agent_tools: {list(self.agent_tools.keys())}")
             return None
         
         # Build the prompt with essential info
@@ -292,22 +291,22 @@ Remember your ENEMIES and CRUSH their dreams! Use emojis to emphasize your DOMIN
         try:
             # Use task_id if we have one for this agent
             task_id = self.task_ids.get(team_num)
-            print(f"🔍 Calling A2A agent for Team {team_num} (task_id: {task_id})")
             result = await self.agent_tools[team_num](prompt, task_id=task_id)
-            print(f"📥 Raw result type: {type(result)}")
             
             # Extract and store task_id
-            task_id = extract_task_id(result)
-            if task_id:
-                self.task_ids[team_num] = task_id
+            new_task_id = extract_task_id(result)
+            if new_task_id:
+                self.task_ids[team_num] = new_task_id
             
             # Parse the response
             output = parse_a2a_response(result, A2AOutput)
             if output:
-                print(f"✅ Parsed output - type: {output.type}, player: {output.player_name}")
+                print(f"✅ Team {team_num} pick: {output.player_name}")
             else:
                 print(f"❌ Failed to parse response from Team {team_num}")
-                print(f"   Raw result: {result}")
+                if isinstance(result, str) and len(result) > 100:
+                    # Show compact preview for debugging
+                    print(f"   Response format issue - check a2a_helpers.py parsing")
             return output
                 
         except Exception as e: