Method process_chunks

methods/mirix_agent.py:386–544

()

        chunk_responses = []

        async def process_chunks():
            nonlocal memory_entries, all_passages, chunk_responses

            for i, chunk in enumerate(chunks):
                chunk_start = time.time()
                try:
                    # Send message to MIRIX for memory extraction
                    # MIRIX's MetaAgent will automatically route content to appropriate
                    # memory components (Episodic, Semantic, Procedural, etc.)
                    # NOTE: We don't pass user_id here - MIRIX uses the LocalClient's
                    # default user which was created during initialization.
                    response = await self._client.send_message(
                        agent_id=self._meta_agent.id,
                        role="user",
                        message=chunk,
                    )

                    chunk_latency = time.time() - chunk_start

                    # Extract usage statistics from MirixResponse
                    if response and hasattr(response, 'usage') and response.usage:
                        self._record_mirix_usage(response.usage, chunk_latency)

                    # Record passage info for logging
                    passage_info = {
                        "chunk_index": i,
                        "chunk_tokens": self.count_tokens(chunk),
                        "content_preview": chunk[:500] + "..." if len(chunk) > 500 else chunk,
                        "latency": round(chunk_latency, 3),
                    }

                    # Extract response messages for logging
                    # MIRIX MirixResponse has messages as List[Union[ToolCallMessage, ToolReturnMessage, AssistantMessage, ...]]
                    # Each message has 'message_type' field to identify the type
                    # - ToolCallMessage: has tool_call.name, tool_call.arguments (JSON string)
                    # - ToolReturnMessage: has tool_return (string)
                    # - AssistantMessage: has content (string or list)
                    response_messages = []
                    if response and hasattr(response, 'messages'):
                        logger.debug(f"[MIRIXAgent] Response has {len(response.messages)} messages")
                        for msg in response.messages:
                            msg_type = getattr(msg, 'message_type', None)
                            if hasattr(msg_type, 'value'):
                                msg_type = msg_type.value

                            msg_content = ""
                            func_name = ""
                            func_args = {}

                            if msg_type == 'tool_call_message':
                                # ToolCallMessage has tool_call with name and arguments
                                tool_call = getattr(msg, 'tool_call', None)
                                if tool_call:
                                    func_name = getattr(tool_call, 'name', 'unknown')
                                    args_str = getattr(tool_call, 'arguments', '{}')
                                    try:
                                        if isinstance(args_str, str):
                                            func_args = json.loads(args_str) if args_str else {}