chore: Minor fixes.

Author: anirbanbasu

src/dqa/client/a2a_mixin.py

@@ -34,9 +34,7 @@ async def obtain_a2a_client(
         final_agent_card_to_use = _public_card
 
         client = ClientFactory(
-            config=ClientConfig(
-                streaming=True, polling=False, httpx_client=httpx_client
-            )
+            config=ClientConfig(streaming=True, polling=True, httpx_client=httpx_client)
         ).create(card=final_agent_card_to_use)
         logger.info("A2A client initialised.")
         return client, final_agent_card_to_use

src/dqa/executor/mhqa.py

@@ -19,7 +19,7 @@ def __init__(self):
         self._actor_mhqa = MHQAActor.__name__
         self._factory = ActorProxyFactory(retry_policy=RetryPolicy(max_attempts=1))
 
-    async def do_mhqa_response(self, data: MHQAInput):
+    async def do_mhqa_respond(self, data: MHQAInput):
         proxy = ActorProxy.create(
             actor_type=self._actor_mhqa,
             actor_id=ActorId(actor_id=data.thread_id),
@@ -37,9 +37,8 @@ async def do_mhqa_response(self, data: MHQAInput):
             raw_body=data.model_dump_json().encode(),
         )
         return result.decode().strip("\"'")
-        # Start listening to the pub-sub topic for responses (non-blocking)
 
-    async def do_get_history(self, data: MHQAHistoryInput) -> str:
+    async def do_mhqa_get_history(self, data: MHQAHistoryInput) -> str:
         proxy = ActorProxy.create(
             actor_type=self._actor_mhqa,
             actor_id=ActorId(actor_id=data.thread_id),
@@ -49,7 +48,7 @@ async def do_get_history(self, data: MHQAHistoryInput) -> str:
         result = await proxy.invoke_method(method=MHQAActorMethods.GetChatHistory)
         return result.decode().strip("\"'")
 
-    async def do_delete_history(self, data: MHQADeleteHistoryInput) -> str:
+    async def do_mhqa_delete_history(self, data: MHQADeleteHistoryInput) -> str:
         proxy = ActorProxy.create(
             actor_type=self._actor_mhqa,
             actor_id=ActorId(actor_id=data.thread_id),
@@ -75,11 +74,11 @@ async def execute(self, context: RequestContext, event_queue: EventQueue):
         response = None
         match message_payload.skill:
             case MHQAAgentSkills.Respond:
-                response = await self.do_mhqa_response(data=message_payload.data)
+                response = await self.do_mhqa_respond(data=message_payload.data)
             case MHQAAgentSkills.GetChatHistory:
-                response = await self.do_get_history(data=message_payload.data)
+                response = await self.do_mhqa_get_history(data=message_payload.data)
             case MHQAAgentSkills.ResetChatHistory:
-                response = await self.do_delete_history(data=message_payload.data)
+                response = await self.do_mhqa_delete_history(data=message_payload.data)
             case _:
                 raise ValueError(f"Unknown skill '{message_payload.skill}' requested!")
         if response:

src/dqa/web/gradio.py

@@ -370,14 +370,17 @@ async def btn_echo_clicked(
                                 ),
                             )
 
-                            chat_history.extend(
+                            waiting_messages = (
                                 self.convert_mhqa_response_to_chat_messages(
                                     MHQAResponse(
-                                        thread_id=selected_chat_id, user_input=txt_input
+                                        thread_id=selected_chat_id,
+                                        user_input=txt_input,
+                                        agent_output="...",
                                     )
                                 )
                             )
-                            last_added_messages = 1
+                            chat_history.extend(waiting_messages)
+                            last_added_messages = len(waiting_messages)
 
                             yield (
                                 None,
@@ -420,16 +423,16 @@ async def btn_echo_clicked(
                                     browser_state_chat_histories[selected_chat_id] = (
                                         chat_history
                                     )
+                                    yield (
+                                        None,
+                                        browser_state_chat_histories,
+                                        selected_chat_id,
+                                        chat_history,
+                                    )
                     else:
                         gr.Warning(
                             f"No input message was provided for chat ID {selected_chat_id}."
                         )
-                    yield (
-                        None,
-                        browser_state_chat_histories,
-                        selected_chat_id,
-                        chat_history,
-                    )
                 except Exception as e:
                     yield (
                         gr.update(value="", interactive=True),