From afa4dd864b245e238ba00eff46ceea30350ab961 Mon Sep 17 00:00:00 2001
From: lhpqaq <657407891@qq.com>
Date: Wed, 5 Feb 2025 00:54:15 +0800
Subject: [PATCH] add RAG module

---
 .../bigtop-manager-ai-rag/pom.xml             | 44 +++++++++++++++++++
 .../org/apache/bigtop/manager/ai/rag/RAG.java | 27 ++++++++++++
 bigtop-manager-ai/pom.xml                     |  1 +
 bigtop-manager-bom/pom.xml                    |  5 +++
 4 files changed, 77 insertions(+)
 create mode 100644 bigtop-manager-ai/bigtop-manager-ai-rag/pom.xml
 create mode 100644 bigtop-manager-ai/bigtop-manager-ai-rag/src/main/java/org/apache/bigtop/manager/ai/rag/RAG.java

diff --git a/bigtop-manager-ai/bigtop-manager-ai-rag/pom.xml b/bigtop-manager-ai/bigtop-manager-ai-rag/pom.xml
new file mode 100644
index 00000000..f9009aea
--- /dev/null
+++ b/bigtop-manager-ai/bigtop-manager-ai-rag/pom.xml
@@ -0,0 +1,44 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!--
+  ~ Licensed to the Apache Software Foundation (ASF) under one
+  ~ or more contributor license agreements.  See the NOTICE file
+  ~ distributed with this work for additional information
+  ~ regarding copyright ownership.  The ASF licenses this file
+  ~ to you under the Apache License, Version 2.0 (the
+  ~ "License"); you may not use this file except in compliance
+  ~ with the License.  You may obtain a copy of the License at
+  ~
+  ~    https://www.apache.org/licenses/LICENSE-2.0
+  ~
+  ~ Unless required by applicable law or agreed to in writing,
+  ~ software distributed under the License is distributed on an
+  ~ "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+  ~ KIND, either express or implied.  See the License for the
+  ~ specific language governing permissions and limitations
+  ~ under the License.
+-->
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+    <parent>
+        <groupId>org.apache.bigtop</groupId>
+        <artifactId>bigtop-manager-ai</artifactId>
+        <version>${revision}</version>
+    </parent>
+
+    <artifactId>bigtop-manager-ai-rag</artifactId>
+    <name>${project.artifactId}</name>
+    <description>Bigtop Manager AI RAG</description>
+
+    <dependencies>
+        <dependency>
+            <groupId>org.apache.bigtop</groupId>
+            <artifactId>bigtop-manager-ai-core</artifactId>
+            <version>${revision}</version>
+        </dependency>
+
+        <dependency>
+            <groupId>dev.langchain4j</groupId>
+            <artifactId>langchain4j-chroma</artifactId>
+        </dependency>
+    </dependencies>
+</project>
diff --git a/bigtop-manager-ai/bigtop-manager-ai-rag/src/main/java/org/apache/bigtop/manager/ai/rag/RAG.java b/bigtop-manager-ai/bigtop-manager-ai-rag/src/main/java/org/apache/bigtop/manager/ai/rag/RAG.java
new file mode 100644
index 00000000..1253e219
--- /dev/null
+++ b/bigtop-manager-ai/bigtop-manager-ai-rag/src/main/java/org/apache/bigtop/manager/ai/rag/RAG.java
@@ -0,0 +1,27 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.apache.bigtop.manager.ai.rag;
+
+import dev.langchain4j.model.embedding.EmbeddingModel;
+import dev.langchain4j.store.embedding.EmbeddingStore;
+import dev.langchain4j.store.embedding.chroma.ChromaEmbeddingStore;
+
+public class RAG {
+ChromaEmbeddingStore a = ChromaEmbeddingStore.builder().baseUrl("http://localhost:8080").build();
+}
diff --git a/bigtop-manager-ai/pom.xml b/bigtop-manager-ai/pom.xml
index 971bdf68..47b7b47e 100644
--- a/bigtop-manager-ai/pom.xml
+++ b/bigtop-manager-ai/pom.xml
@@ -37,6 +37,7 @@
         <module>bigtop-manager-ai-deepseek</module>
         <module>bigtop-manager-ai-core</module>
         <module>bigtop-manager-ai-assistant</module>
+        <module>bigtop-manager-ai-rag</module>
     </modules>
 
     <dependencyManagement>
diff --git a/bigtop-manager-bom/pom.xml b/bigtop-manager-bom/pom.xml
index ce8f88b8..dabaa346 100644
--- a/bigtop-manager-bom/pom.xml
+++ b/bigtop-manager-bom/pom.xml
@@ -278,6 +278,11 @@
                 <artifactId>langchain4j-reactor</artifactId>
                 <version>${langchain4j.version}</version>
             </dependency>
+            <dependency>
+                <groupId>dev.langchain4j</groupId>
+                <artifactId>langchain4j-chroma</artifactId>
+                <version>${langchain4j.version}</version>
+            </dependency>
         </dependencies>
     </dependencyManagement>
 </project>