Can function callback call the reactive method of the reactor used?

[PeatBoy]

Hi，I have a scenario for calling the reactor method in function calling, as shown below：

// This is a reactor method that calls chatModel.call and uses withFunctionCallbacks to call dataQueryService.
  public Mono<String> chat(SessionUser sessionUser, boolean preview,
      String question, Mode mode,
      DataSourceType dbEngine, Mono<Struct> structMono) {

    return structMono.map(struct -> {
      String systemPrompt = this.assistantPromptTemplate.createSystemMessage(
          dbEngine, this.processStruct(struct));
      SystemMessage systemMessage = new SystemMessage(systemPrompt);
      UserMessage userMessage = new UserMessage(question);
      return chatModel.call(
              new Prompt(List.of(systemMessage, userMessage), OpenAiChatOptions.builder()
                  .withModel(mode.getModel())
                  .withFunctionCallbacks(List.of(FunctionCallbackWrapper.builder(this.dataQueryService)
                      .withName("dataQueryFunction")
                      .withDescription("Use query statements to get data")
                      .withObjectMapper(this.objectMapper)
                      .build()))
                  .withToolContext(
                      Map.of("tenantId", sessionUser.getTenantId(), "preview", preview, "dbEngine",
                          dbEngine.name()))
                  .build()
              ))
          .getResult()
          .getOutput()
          .getContent();
    });
  }

// The data query service application method will call the dataQueryRepository.query method, which is a reactor method.
public class DataQueryService implements
    BiFunction<Request, ToolContext, Response> {

  @Resource
  private DataQueryRepository dataQueryRepository;

  @Override
  public Response apply(Request request, ToolContext toolContext) {
    Assert.notNull(request.sql, "sql must not be null");

    UUID tenantId = (UUID) toolContext.getContext()
        .get("tenantId");
    boolean preview = (boolean) toolContext.getContext()
        .get("preview");
    String dbEngine = (String) toolContext.getContext()
        .get("dbEngine");

    ArrayList<Object> res = new ArrayList<>();
    this.dataQueryRepository.query(tenantId, preview, request.sql)
        .onErrorResume(e -> {
          log.error("Assistant execute query error!", e);
          return Flux.empty();
        })
        .collectList()
        .subscribe(res::add);

    log.info("Query result: {}", res);

    return new Response(res);
  }

With the above method, chatModel can't get the data query result normally because it is asynchronous. I tried an asynchronous thread and blocked the acquisition of data (as shown below), which can be successfully obtained, but sometimes it will cause the flow in the asynchronous thread to be broken, resulting in the main thread of DataQueryService being blocked, and the request is returned to 504. Excuse me, is there any way to realize function callback calling reactor method?

List<Map<String, Object>> data = CompletableFuture.supplyAsync(
            () -> this.dataQueryRepository.query(tenantId, preview, request.sql)
                .onErrorResume(e -> {
                  log.error("Assistant execute query error!", e);
                  return Flux.empty();
                })
                .collectList()
                .block())
        .join();

    log.info("Query result: {}", data);

[tzolov]

Lets follow up this discussion in the related issue: #1778