fix

Author: wenjin272

python/flink_agents/plan/actions/chat_model_action.py

@@ -251,7 +251,6 @@ async def _process_tool_response(event: ToolResponseEvent, ctx: RunnerContext) -
     initial_request_id = tool_request_event_context["initial_request_id"]
 
     # update tool call context, and get the entire chat messages.
-    print(f"RequestId: {request_id}, Initial request_id: {initial_request_id}")
     messages = _update_tool_call_context(
         sensory_memory,
         initial_request_id,
@@ -286,6 +285,8 @@ async def process_chat_request_or_tool_response(
     the complete chat flow including tool calls. It uses sensory memory to save
     the tool call context, which is a dict mapping request id to chat messages.
     """
+    # To avoid https://github.com/alibaba/pemja/issues/88, we log a message here.
+    logging.debug("Processing chat request asynchronously.")
     if isinstance(event, ChatRequestEvent):
         await _process_chat_request(event, ctx)
     elif isinstance(event, ToolResponseEvent):

python/flink_agents/plan/actions/context_retrieval_action.py

@@ -15,6 +15,8 @@
 #  See the License for the specific language governing permissions and
 # limitations under the License.
 #################################################################################
+import logging
+
 from flink_agents.api.core_options import AgentExecutionOptions
 from flink_agents.api.events.context_retrieval_event import (
     ContextRetrievalRequestEvent,
@@ -27,6 +29,7 @@
 from flink_agents.plan.actions.action import Action
 from flink_agents.plan.function import PythonFunction
 
+_logger = logging.getLogger(__name__)
 
 async def process_context_retrieval_request(event: Event, ctx: RunnerContext) -> None:
     """Built-in action for processing context retrieval requests."""
@@ -37,6 +40,9 @@ async def process_context_retrieval_request(event: Event, ctx: RunnerContext) ->
 
         rag_async = ctx.config.get(AgentExecutionOptions.RAG_ASYNC)
         if rag_async:
+            # To avoid https://github.com/alibaba/pemja/issues/88,
+            # we log a message here.
+            _logger.debug("Processing context retrieval asynchronously.")
             result = await ctx.execute_async(vector_store.query, query)
         else:
             result = vector_store.query(query)

python/flink_agents/plan/actions/tool_call_action.py

@@ -15,6 +15,7 @@
 #  See the License for the specific language governing permissions and
 # limitations under the License.
 #################################################################################
+import logging
 
 from flink_agents.api.core_options import AgentExecutionOptions
 from flink_agents.api.events.tool_event import ToolRequestEvent, ToolResponseEvent
@@ -23,11 +24,16 @@
 from flink_agents.plan.actions.action import Action
 from flink_agents.plan.function import PythonFunction
 
+_logger = logging.getLogger(__name__)
 
 async def process_tool_request(event: ToolRequestEvent, ctx: RunnerContext) -> None:
     """Built-in action for processing tool call requests."""
     tool_call_async = ctx.config.get(AgentExecutionOptions.TOOL_CALL_ASYNC)
 
+    if tool_call_async:
+        # To avoid https://github.com/alibaba/pemja/issues/88, we log a message here.
+        _logger.debug("Processing tool call asynchronously.")
+
     responses = {}
     external_ids = {}
     for tool_call in event.tool_calls: