}

    /**
     * Regenerates a search query when the previous query failed.
     *
     * @param userMessage the user's original message
     * @param failedQuery the query that failed
     * @param failureReason the reason for failure
     * @param history the conversation history
     * @return a new query string
     * @throws LlmException if LLM is not available
     */

// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//

            }

            @Override
            public String getSystemProperty(final String key) {
                return systemPropMap.get(key);
            }
        };

        // Test that new key takes precedence over legacy key
        systemPropMap.put("entraid.permission.fields", "newField");
        systemPropMap.put("aad.permission.fields", "legacyField");

    assertThat(first.requestLine).isEqualTo("GET / HTTP/1.1")
    val redirect = server.takeRequest()
    assertThat(redirect.requestLine).isEqualTo("GET /new-path HTTP/1.1")
  }

  /**
   * Test that MockWebServer blocks for a call to enqueue() if a request is made before a mock
   * response is ready.
   */
  @Test
  fun dispatchBlocksWaitingForEnqueue() {
    Thread {
      try {
        Thread.sleep(1000)

        @Override
        public int compare(Field left, Field right) {
          return left.getName().compareTo(right.getName());
        }
      };

  /**
   * Returns a new {@code MatchResult} that corresponds to a successful match. Apache Harmony (used
   * in Android) requires a successful match in order to generate a {@code MatchResult}:

        }

        /** The return value of the operation */
        public int retval;
        /** The handle to the alias */
        public rpc.policy_handle alias_handle;
        /** The array of SIDs that are members of the alias */
        public lsarpc.LsarSidArray sids;

        /**
         * Constructs a SamrGetMembersInAlias request.
         *
         * @param alias_handle The alias handle

        }

        /** The return value of the operation */
        public int retval;
        /** The handle to the alias */
        public rpc.policy_handle alias_handle;
        /** The array of SIDs that are members of the alias */
        public lsarpc.LsarSidArray sids;

        /**
         * Constructs a SamrGetMembersInAlias request.
         *
         * @param alias_handle The alias handle

    for (int i = 0; i < count; i++) {
      assertWithMessage("result(%s) didn't match expected", i)
          .that(resultArray.get(i))
          .isSameInstanceAs(result);
    }
  }

  /**
   * On a concurrent computation that returns null, all threads should get an
   * InvalidCacheLoadException, with the loader only called once. The result should not be cached (a

 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *

        assertEquals(0L, callback.getExecuteTime());

        // Store a document
        callback.store(new DataStoreParams(), new HashMap<>());

        // Execution time should be greater than 0
        assertTrue(callback.getExecuteTime() >= 10L);
    }

    @Test
    public void test_concurrent_stores() {
        // Test thread-safe implementation