add openai

Author: alexcheng1982

.gitignore

@@ -32,4 +32,6 @@ build/
 .vscode/
 
 ### Mac OS ###
-.DS_Store
+.DS_Store
+
+.pdf-imported

compose.yaml

@@ -0,0 +1,9 @@
+services:
+  chroma:
+    image: chromadb/chroma:0.6.1
+    volumes:
+      - chroma-data:/chroma/chroma
+    ports:
+      - "8000:8000"
+volumes:
+  chroma-data:

pom.xml

@@ -7,7 +7,7 @@
   <parent>
     <groupId>org.springframework.boot</groupId>
     <artifactId>spring-boot-starter-parent</artifactId>
-    <version>3.4.1</version>
+    <version>3.3.7</version>
     <relativePath/>
   </parent>
 
@@ -43,6 +43,10 @@
       <groupId>org.springframework.ai</groupId>
       <artifactId>spring-ai-ollama-spring-boot-starter</artifactId>
     </dependency>
+    <dependency>
+      <groupId>org.springframework.ai</groupId>
+      <artifactId>spring-ai-openai-spring-boot-starter</artifactId>
+    </dependency>
     <dependency>
       <groupId>org.springframework.ai</groupId>
       <artifactId>spring-ai-pdf-document-reader</artifactId>
@@ -51,6 +55,11 @@
       <groupId>org.springframework.boot</groupId>
       <artifactId>spring-boot-starter-web</artifactId>
     </dependency>
+    <dependency>
+      <groupId>org.springdoc</groupId>
+      <artifactId>springdoc-openapi-starter-webmvc-ui</artifactId>
+      <version>2.5.0</version>
+    </dependency>
   </dependencies>
 
   <repositories>

src/main/java/com/javaaidev/pdfqa/PDFContentLoader.java

@@ -32,6 +32,7 @@ public void load(Path pdfFilePath) {
   public void run(String... args) throws Exception {
     var markerFile = Path.of(".", ".pdf-imported");
     if (Files.exists(markerFile)) {
+      LOGGER.info("Marker file {} exists, skip. Delete this file to re-import.", markerFile);
       return;
     }
     load(Path.of(".", "content", "Understanding_Climate_Change.pdf"));

src/main/resources/application-openai.yaml

@@ -0,0 +1,12 @@
+spring:
+  ai:
+    ollama:
+      chat:
+        enabled: false
+      embedding:
+        enabled: false
+    openai:
+      chat:
+        enabled: true
+      embedding:
+        enabled: true

src/main/resources/application.yaml

@@ -7,12 +7,25 @@ spring:
   ai:
     ollama:
       chat:
+        enabled: true
         options:
           model: "phi3"
           temperature: 0
       embedding:
+        enabled: true
         options:
           model: "bge-large"
+    openai:
+      api-key: ${OPENAI_API_KEY:}
+      chat:
+        enabled: false
+        options:
+          model: gpt-3.5-turbo
+          temperature: 0.0
+      embedding:
+        enabled: false
+        options:
+          model: text-embedding-3-small
     vectorstore:
       chroma:
         collectionName: pdf-qa