fix: test

payments_py/ai_query_api.py

@@ -40,6 +40,8 @@ async def subscribe(self, callback: Any, did: Optional[str]=None, events: Option
         print('query-api:: Connected to the server')
         try: 
             pending_steps = self.get_steps(AgentExecutionStatus.Pending)
+            if callback is not None:
+                await callback(pending_steps.json()['steps'])
             await self._emit_events(pending_steps.json()['steps'])
         except Exception as e:
             print('query-api:: Unable to get pending events', e)

tests/protocol_test.py

@@ -67,9 +67,9 @@ async def eventsReceived(data):
                                                                     'output': 'success',
                                                                     'is_last': True
                                                                     })
-    response_data = data
-    response_event.set()
-    print(result.json())
+        response_data = data
+        response_event.set()
+        print(result.json())
 
 
 @pytest.mark.asyncio(loop_scope="module")
@@ -118,7 +118,7 @@ async def test_AIQueryApi_create_task(ai_query_api_build_fixture, ai_query_api_s
     task = subscriber.ai_protocol.create_task(agent.did, {'query': 'sample_query', 'name': 'sample_task', 'additional_params': {'param1': 'value1', 'param2': 'value2'}})
     print('Task created:', task.json())
 
-    await asyncio.wait_for(response_event.wait(), timeout=60)
+    await asyncio.wait_for(response_event.wait(), timeout=10)
 
     assert response_data is not None, "Builder did not receive the event from subscriber"
     print('Task received by builder:', response_data)