spring ai 해보기

gradle / groovy  springbooot 3.5.8

plugins {
    id 'java'
    id 'org.springframework.boot' version '3.5.8' // 혹시 에러나면 3.4.0 권장
    id 'io.spring.dependency-management' version '1.1.7'
}

group = 'com.ai'
version = '0.0.1-SNAPSHOT'

java {
    toolchain {
        languageVersion = JavaLanguageVersion.of(17)
    }
}

repositories {
    mavenCentral()

    maven { url 'https://repo.spring.io/snapshot' }
}

ext {
    set('springAiVersion', "1.1.0-M3")
}

dependencies {
    implementation 'org.springframework.boot:spring-boot-starter-web'


    implementation 'org.springframework.ai:spring-ai-starter-model-vertex-ai-gemini'

    compileOnly 'org.projectlombok:lombok'
    annotationProcessor 'org.projectlombok:lombok'
    testImplementation 'org.springframework.boot:spring-boot-starter-test'
    testRuntimeOnly 'org.junit.platform:junit-platform-launcher'
}

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

tasks.named('test') {
    useJUnitPlatform()
}

package com.ai.springai.controller;

import org.springframework.ai.vertexai.gemini.VertexAiGeminiChatModel;

import org.springframework.web.bind.annotation.*;

import java.util.HashMap;
import java.util.Map;

@RequestMapping("/api")
@RestController
public class ChatController {


  private final VertexAiGeminiChatModel vertexAiGeminiChatModel;

    public ChatController( VertexAiGeminiChatModel vertexAiGeminiChatModel) {

        this.vertexAiGeminiChatModel = vertexAiGeminiChatModel;
    }
    @GetMapping("/chat")
    public Map<String, String> chat(@RequestBody String message) {
        Map<String, String> responses = new HashMap<>();


        String vertexAiGeminiResponse = vertexAiGeminiChatModel.call(message);
        responses.put("vertexai(gemini) 응답", vertexAiGeminiResponse);
        return responses;

    }


    }

spring:

  ai:
    vertex:
      ai:
        gemini:
          project-id: //프로젝트 아이디  적어주세요 
          location: us-central1

 

환경 설정 컨트롤러 하나 만들었습니다

postman으로테스트해보면 

 

참고한 블로그 

https://velog.io/@hanni/Spring-Boot3-Spring-AI%EB%A5%BC-%ED%99%9C%EC%9A%A9%ED%95%9C-%EC%B4%88%EA%B0%84%EB%8B%A8-Chat-Model-API-%EA%B5%AC%ED%98%84%ED%95%98%EA%B8%B0-Gemini-ChatGPT

[Spring Boot3] Spring AI를 활용한 초간단 Chat Model API 🗣️구현하기 (Gemini, ChatGPT)

 

https://github.com/cookiboii/springai.git

GitHub - cookiboii/springai

제 소스 코드입니다restful 방식으로 코드를 짰고

https://github.com/cookiboii/springai-with-nextjs.git

GitHub - cookiboii/springai-with-nextjs

프론트는 사실 제미나이 cli써서 짰습니다 참고하세요 껍데기만 만들었습니다