import org.jetbrains.kotlin.gradle.tasks.KotlinCompile

plugins {
	id("org.springframework.boot") version "3.2.3"
	id("io.spring.dependency-management") version "1.1.4"
	id("org.graalvm.buildtools.native") version "0.9.28"
	kotlin("jvm") version "1.9.22"
	kotlin("plugin.spring") version "1.9.22"
}

group = "com.xavierbouclet"
version = "0.0.1-SNAPSHOT"

java {
	sourceCompatibility = JavaVersion.VERSION_17
}

repositories {
	mavenCentral()
	maven { url = uri("https://repo.spring.io/milestone") }
}

extra["springAiVersion"] = "0.8.0"

dependencies {
	implementation("org.springframework.boot:spring-boot-starter-web")
	implementation("com.fasterxml.jackson.module:jackson-module-kotlin")
	implementation("org.jetbrains.kotlin:kotlin-reflect")
	implementation("org.springframework.ai:spring-ai-openai-spring-boot-starter")
	testImplementation("org.springframework.boot:spring-boot-starter-test")
}

dependencyManagement {
	imports {
		mavenBom("org.springframework.ai:spring-ai-bom:${property("springAiVersion")}")
	}
}

tasks.withType<KotlinCompile> {
	kotlinOptions {
		freeCompilerArgs += "-Xjsr305=strict"
		jvmTarget = "17"
	}
}

tasks.withType<Test> {
	useJUnitPlatform()
}

import org.springframework.ai.chat.messages.UserMessage
import org.springframework.ai.chat.prompt.Prompt
import org.springframework.ai.openai.OpenAiChatClient
import org.springframework.context.annotation.Bean
import org.springframework.context.annotation.Configuration
import org.springframework.web.servlet.function.ServerResponse
import org.springframework.web.servlet.function.router

@Configuration(proxyBeanMethods = false)
class RouterConfiguration {

    @Bean
    fun aiRouter(chatClient: OpenAiChatClient) = router {
        GET("/api/ai/generate") { request ->
            ServerResponse
                .ok()
                .body(
                    chatClient.call(
                        request
                            .param("message")
                            .orElse("Tell me a chuck norris fact")
                    )
                )
        }
        GET("/api/ai/generateStream") { request ->
            ServerResponse
                .ok()
                .body(chatClient.stream(
                    Prompt(
                        UserMessage(
                            request
                                .param("message")
                                .orElse("Tell me a chuck norris fact")
                        )
                    )
                ).mapNotNull { chatResp -> chatResp?.result?.output?.content }
                    .toStream()
                    .toList()
                )
        }
    }
}

spring:
  ai:
    openai:
      api-key: ${OPEN_AI_KEY}
      chat:
        options:
          model: gpt-3.5-turbo
          temperature: 0.7

export OPEN_AI_KEY=your-open-ai-key

./gradlew bootJar

java -jar build/libs/spring-ai-0.0.1-SNAPSHOT.jar

$ curl localhost:8080/api/ai/generate

Chuck Norris doesn't do push-ups. He pushes the Earth down.%

curl "localhost:8080/api/ai/generate?message=explain%20me%20spring%20boot"

curl "localhost:8080/api/ai/generateStream?message=explain%20me%20spring%20boot"

["","Spring"," Boot"," is"," an"," open","-source"," Java","-based"," framework"," that"," is"," used"," to"," create"," stand","-alone",","," production","-grade"," Spring","-based"," applications","."," It"," provides"," a"," pre","-config","ured"," set"," of"," tools"," and"," libraries"," to"," simplify"," the"," process"," of"," building"," and"," deploying"," applications","."," \n\n","Spring"," Boot"," aims"," to"," streamline"," the"," development"," process"," by"," providing"," a"," convention","-over","-","configuration"," approach",","," meaning"," that"," developers"," can"," quickly"," set"," up"," a"," project"," without"," having"," to"," spend"," time"," on"," configuring"," various"," components","."," It"," also"," includes"," embedded"," servers",","," such"," as"," Tom","cat",","," Jet","ty",","," or"," Undert","ow",","," allowing"," applications"," to"," be"," run"," as"," standalone"," J","AR"," files"," without"," the"," need"," for"," a"," separate"," server"," installation",".\n\n","Additionally",","," Spring"," Boot"," offers"," built","-in"," support"," for"," aspects"," such"," as"," security",","," logging",","," monitoring",","," and"," testing",","," making"," it"," easier"," for"," developers"," to"," create"," robust"," and"," scalable"," applications","."," It"," also"," integrates"," seamlessly"," with"," other"," Spring"," projects",","," such"," as"," Spring"," Framework",","," Spring"," Data",","," and"," Spring"," Security",","," allowing"," developers"," to"," leverage"," the"," full"," power"," of"," the"," Spring"," ecosystem",".\n\n","Overall",","," Spring"," Boot"," is"," a"," powerful"," framework"," that"," simpl","ifies"," the"," development"," process"," and"," helps"," developers"," create"," high","-quality",","," production","-ready"," applications"," quickly"," and"," efficiently","."]%

./gradlew bootBuildImage

tasks.withType<BootBuildImage> {
	val osName = System.getProperty("os.name").lowercase()
	val arch = System.getProperty("os.arch")

	val runningOnM1Mac = "mac" in osName && arch == "aarch64"
	if (runningOnM1Mac) {
		builder.set("dashaun/builder:tiny")
		environment.set(mapOf("BP_NATIVE_IMAGE" to "true"))
	}
}

Deprecated Gradle features were used in this build, making it incompatible with Gradle 9.0.

You can use '--warning-mode all' to show the individual deprecation warnings and determine if they come from your own scripts or plugins.

For more on this, please refer to https://docs.gradle.org/8.5/userguide/command_line_interface.html#sec:command_line_warnings in the Gradle documentation.

BUILD SUCCESSFUL in 6m 13s
10 actionable tasks: 2 executed, 8 up-to-date

$ docker images | grep spring-ai
spring-ai                                           0.0.1-SNAPSHOT         d66dd3be8bb9   44 years ago    130MB

docker run -e OPEN_AI_KEY=$OPEN_AI_KEY -p 8080:8080 spring-ai:0.0.1-SNAPSHOT

  .   ____          _            __ _ _
 /\\ / ___'_ __ _ _(_)_ __  __ _ \ \ \ \
( ( )\___ | '_ | '_| | '_ \/ _` | \ \ \ \
 \\/  ___)| |_)| | | | | || (_| |  ) ) ) )
  '  |____| .__|_| |_|_| |_\__, | / / / /
 =========|_|==============|___/=/_/_/_/
 :: Spring Boot ::                (v3.2.3)

2024-03-05T01:51:52.610Z  INFO 1 --- [           main] c.x.springai.SpringAiApplicationKt       : Starting AOT-processed SpringAiApplicationKt using Java 17.0.10 with PID 1 (/workspace/com.xavierbouclet.springai.SpringAiApplicationKt started by cnb in /workspace)
2024-03-05T01:51:52.610Z  INFO 1 --- [           main] c.x.springai.SpringAiApplicationKt       : No active profile set, falling back to 1 default profile: "default"
2024-03-05T01:51:52.617Z  INFO 1 --- [           main] o.s.b.w.embedded.tomcat.TomcatWebServer  : Tomcat initialized with port 8080 (http)
2024-03-05T01:51:52.618Z  INFO 1 --- [           main] o.apache.catalina.core.StandardService   : Starting service [Tomcat]
2024-03-05T01:51:52.618Z  INFO 1 --- [           main] o.apache.catalina.core.StandardEngine    : Starting Servlet engine: [Apache Tomcat/10.1.19]
2024-03-05T01:51:52.622Z  INFO 1 --- [           main] o.a.c.c.C.[Tomcat].[localhost].[/]       : Initializing Spring embedded WebApplicationContext
2024-03-05T01:51:52.622Z  INFO 1 --- [           main] w.s.c.ServletWebServerApplicationContext : Root WebApplicationContext: initialization completed in 12 ms
2024-03-05T01:51:52.646Z  INFO 1 --- [           main] o.s.b.w.embedded.tomcat.TomcatWebServer  : Tomcat started on port 8080 (http) with context path ''
2024-03-05T01:51:52.661Z  INFO 1 --- [           main] c.x.springai.SpringAiApplicationKt       : Started SpringAiApplicationKt in 0.057 seconds (process running for 0.059)