feat: Adding autoCreateSession in Runner

PiperOrigin-RevId: 858855558

Author: google-genai-bot

core/src/main/java/com/google/adk/agents/RunConfig.java

@@ -70,6 +70,8 @@ public enum ToolExecutionMode {
 
   public abstract int maxLlmCalls();
 
+  public abstract boolean autoCreateSession();
+
   public abstract Builder toBuilder();
 
   public static Builder builder() {
@@ -78,7 +80,8 @@ public static Builder builder() {
         .setResponseModalities(ImmutableList.of())
         .setStreamingMode(StreamingMode.NONE)
         .setToolExecutionMode(ToolExecutionMode.NONE)
-        .setMaxLlmCalls(500);
+        .setMaxLlmCalls(500)
+        .setAutoCreateSession(false);
   }
 
   public static Builder builder(RunConfig runConfig) {
@@ -90,7 +93,8 @@ public static Builder builder(RunConfig runConfig) {
         .setResponseModalities(runConfig.responseModalities())
         .setSpeechConfig(runConfig.speechConfig())
         .setOutputAudioTranscription(runConfig.outputAudioTranscription())
-        .setInputAudioTranscription(runConfig.inputAudioTranscription());
+        .setInputAudioTranscription(runConfig.inputAudioTranscription())
+        .setAutoCreateSession(runConfig.autoCreateSession());
   }
 
   /** Builder for {@link RunConfig}. */
@@ -123,6 +127,9 @@ public abstract Builder setInputAudioTranscription(
     @CanIgnoreReturnValue
     public abstract Builder setMaxLlmCalls(int maxLlmCalls);
 
+    @CanIgnoreReturnValue
+    public abstract Builder setAutoCreateSession(boolean autoCreateSession);
+
     abstract RunConfig autoBuild();
 
     public RunConfig build() {

core/src/main/java/com/google/adk/runner/Runner.java

@@ -378,9 +378,15 @@ public Flowable<Event> runAsync(
         this.sessionService.getSession(appName, userId, sessionId, Optional.empty());
     return maybeSession
         .switchIfEmpty(
-            Single.error(
-                new IllegalArgumentException(
-                    String.format("Session not found: %s for user %s", sessionId, userId))))
+            Single.defer(
+                () -> {
+                  if (runConfig.autoCreateSession()) {
+                    return this.sessionService.createSession(appName, userId, null, sessionId);
+                  }
+                  return Single.error(
+                      new IllegalArgumentException(
+                          String.format("Session not found: %s for user %s", sessionId, userId)));
+                }))
         .flatMapPublisher(session -> this.runAsync(session, newMessage, runConfig, stateDelta));
   }
 
@@ -668,15 +674,17 @@ public Flowable<Event> runLive(
       String userId, String sessionId, LiveRequestQueue liveRequestQueue, RunConfig runConfig) {
     return this.sessionService
         .getSession(appName, userId, sessionId, Optional.empty())
-        .flatMapPublisher(
-            session -> {
-              if (session == null) {
-                return Flowable.error(
-                    new IllegalArgumentException(
-                        String.format("Session not found: %s for user %s", sessionId, userId)));
-              }
-              return this.runLive(session, liveRequestQueue, runConfig);
-            });
+        .switchIfEmpty(
+            Single.defer(
+                () -> {
+                  if (runConfig.autoCreateSession()) {
+                    return this.sessionService.createSession(appName, userId, null, sessionId);
+                  }
+                  return Single.error(
+                      new IllegalArgumentException(
+                          String.format("Session not found: %s for user %s", sessionId, userId)));
+                }))
+        .flatMapPublisher(session -> this.runLive(session, liveRequestQueue, runConfig));
   }
 
   /**

core/src/test/java/com/google/adk/runner/RunnerTest.java

@@ -832,4 +832,86 @@ public void resumabilityConfig_isNotResumable_isFalseInInvocationContext() {
         runner.runAsync("user", session.id(), createContent("from user")).toList().blockingGet();
     assertThat(contextCaptor.getValue().isResumable()).isFalse();
   }
+
+  @Test
+  public void runAsync_withoutSessionAndAutoCreateSessionTrue_createsSession() {
+    RunConfig runConfig = RunConfig.builder().setAutoCreateSession(true).build();
+    String newSessionId = "new-session-id";
+
+    // Make sure session doesn't exist.
+    assertThat(
+            runner
+                .sessionService()
+                .getSession("test", "user", newSessionId, Optional.empty())
+                .blockingGet())
+        .isNull();
+
+    var events =
+        runner
+            .runAsync("user", newSessionId, createContent("from user"), runConfig)
+            .toList()
+            .blockingGet();
+
+    assertThat(simplifyEvents(events)).containsExactly("test agent: from llm");
+    assertThat(
+            runner
+                .sessionService()
+                .getSession("test", "user", newSessionId, Optional.empty())
+                .blockingGet())
+        .isNotNull();
+  }
+
+  @Test
+  public void runAsync_withoutSessionAndAutoCreateSessionFalse_throwsException() {
+    RunConfig runConfig = RunConfig.builder().setAutoCreateSession(false).build();
+    String newSessionId = "new-session-id";
+
+    runner
+        .runAsync("user", newSessionId, createContent("from user"), runConfig)
+        .test()
+        .assertError(IllegalArgumentException.class);
+  }
+
+  @Test
+  public void runLive_withoutSessionAndAutoCreateSessionTrue_createsSession() throws Exception {
+    RunConfig runConfig = RunConfig.builder().setAutoCreateSession(true).build();
+    String newSessionId = "new-session-id";
+    LiveRequestQueue liveRequestQueue = new LiveRequestQueue();
+
+    // Make sure session doesn't exist.
+    assertThat(
+            runner
+                .sessionService()
+                .getSession("test", "user", newSessionId, Optional.empty())
+                .blockingGet())
+        .isNull();
+
+    TestSubscriber<Event> testSubscriber =
+        runner.runLive("user", newSessionId, liveRequestQueue, runConfig).test();
+
+    liveRequestQueue.content(createContent("from user"));
+    liveRequestQueue.close();
+
+    testSubscriber.await();
+    testSubscriber.assertComplete();
+    assertThat(simplifyEvents(testSubscriber.values())).containsExactly("test agent: from llm");
+    assertThat(
+            runner
+                .sessionService()
+                .getSession("test", "user", newSessionId, Optional.empty())
+                .blockingGet())
+        .isNotNull();
+  }
+
+  @Test
+  public void runLive_withoutSessionAndAutoCreateSessionFalse_throwsException() {
+    RunConfig runConfig = RunConfig.builder().setAutoCreateSession(false).build();
+    String newSessionId = "new-session-id";
+    LiveRequestQueue liveRequestQueue = new LiveRequestQueue();
+
+    runner
+        .runLive("user", newSessionId, liveRequestQueue, runConfig)
+        .test()
+        .assertError(IllegalArgumentException.class);
+  }
 }