Merge branch 'main' of https://github.com/LangMem/lang-memory

Author: Aias00

QUICK_START.md

@@ -6,7 +6,7 @@
 
 ```bash
 # 克隆仓库
-git clone https://github.com/Mem4j/mem4j.git
+git clone https://github.com/mem4j/mem4j.git
 cd mem4j
 
 # 安装到本地Maven仓库

pom.xml

@@ -12,7 +12,7 @@
 
     <name>Mem4j</name>
     <description>Long-term memory for AI Agents - Java Implementation</description>
-    <url>https://github.com/Mem4j/mem4j</url>
+    <url>https://github.com/mem4j/mem4j</url>
 
     <licenses>
         <license>
@@ -23,18 +23,18 @@
     </licenses>
 
     <scm>
-        <connection>scm:git:https://github.com/Mem4j/mem4j.git</connection>
-        <developerConnection>scm:git:https://github.com/Mem4j/mem4j.git</developerConnection>
-        <url>https://github.com/Mem4j/mem4j</url>
+        <connection>scm:git:https://github.com/mem4j/mem4j.git</connection>
+        <developerConnection>scm:git:https://github.com/mem4j/mem4j.git</developerConnection>
+        <url>https://github.com/mem4j/mem4j</url>
     </scm>
 
     <issueManagement>
         <system>Github Issues</system>
-        <url>https://github.com/Mem4j/mem4j/issues</url>
+        <url>https://github.com/mem4j/mem4j/issues</url>
     </issueManagement>
     <ciManagement>
         <system>Github Actions</system>
-        <url>https://github.com/Mem4j/mem4j/actions</url>
+        <url>https://github.com/mem4j/mem4j/actions</url>
     </ciManagement>
 
     <developers>

src/main/java/com/langmem/mem4j/Mem4jApplication.java

@@ -22,6 +22,7 @@
 /**
  * Main Spring Boot application class for Mem4j
  */
+
 @SpringBootApplication
 public class Mem4jApplication {
 

src/main/java/com/langmem/mem4j/embeddings/DashScopeEmbeddingService.java

@@ -21,6 +21,7 @@
 import com.langmem.mem4j.configs.MemoryConfig;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.context.annotation.Primary;
 import org.springframework.http.HttpEntity;
 import org.springframework.http.HttpHeaders;
 import org.springframework.http.MediaType;
@@ -39,6 +40,7 @@
  */
 
 @Service
+@Primary
 public class DashScopeEmbeddingService implements EmbeddingService {
 
 	private static final Logger logger = LoggerFactory.getLogger(DashScopeEmbeddingService.class);
@@ -64,7 +66,7 @@ public DashScopeEmbeddingService(MemoryConfig config) {
 	}
 
 	@Override
-	public double[] embed(String text) {
+	public Double[] embed(String text) {
 		try {
 			Map<String, Object> requestBody = new HashMap<>();
 			requestBody.put("model", model);
@@ -88,7 +90,7 @@ public double[] embed(String text) {
 				JsonNode firstEmbedding = embeddings.get(0);
 				JsonNode embeddingArray = firstEmbedding.path("embedding");
 
-				double[] result = new double[embeddingArray.size()];
+				Double[] result = new Double[embeddingArray.size()];
 				for (int i = 0; i < embeddingArray.size(); i++) {
 					result[i] = embeddingArray.get(i).asDouble();
 				}
@@ -104,7 +106,7 @@ public double[] embed(String text) {
 	}
 
 	@Override
-	public double[][] embed(String[] texts) {
+	public Double[][] embed(String[] texts) {
 		try {
 			Map<String, Object> requestBody = new HashMap<>();
 			requestBody.put("model", model);
@@ -124,10 +126,10 @@ public double[][] embed(String[] texts) {
 			JsonNode responseJson = objectMapper.readTree(response.getBody());
 			JsonNode embeddings = responseJson.path("output").path("embeddings");
 
-			double[][] result = new double[embeddings.size()][];
+            Double[][] result = new Double[embeddings.size()][];
 			for (int i = 0; i < embeddings.size(); i++) {
 				JsonNode embeddingArray = embeddings.get(i).path("embedding");
-				result[i] = new double[embeddingArray.size()];
+				result[i] = new Double[embeddingArray.size()];
 				for (int j = 0; j < embeddingArray.size(); j++) {
 					result[i][j] = embeddingArray.get(j).asDouble();
 				}
@@ -142,12 +144,12 @@ public double[][] embed(String[] texts) {
 	}
 
 	@Override
-	public int getDimension() {
+	public Integer getDimension() {
 		return dimension;
 	}
 
 	@Override
-	public boolean isAvailable() {
+	public Boolean isAvailable() {
 		try {
 			// Simple test to check if service is available
 			embed("test");

src/main/java/com/langmem/mem4j/embeddings/EmbeddingService.java

@@ -23,23 +23,23 @@
 public interface EmbeddingService {
 
 	/**
-	 * Generate embedding for a text
-	 */
-	double[] embed(String text);
+     * Generate embedding for a text
+     */
+    Double[] embed(String text);
 
 	/**
-	 * Generate embeddings for multiple texts
-	 */
-	double[][] embed(String[] texts);
+     * Generate embeddings for multiple texts
+     */
+    Double[][] embed(String[] texts);
 
 	/**
 	 * Get the dimension of embeddings
 	 */
-	int getDimension();
+	Integer getDimension();
 
 	/**
 	 * Check if the service is available
 	 */
-	boolean isAvailable();
+	Boolean isAvailable();
 
 }

src/main/java/com/langmem/mem4j/embeddings/OpenAIEmbeddingService.java

@@ -54,16 +54,16 @@ public OpenAIEmbeddingService(MemoryConfig config) {
 	}
 
 	@Override
-	public double[] embed(String text) {
+	public Double[] embed(String text) {
 
 		try {
 			EmbeddingRequest request = EmbeddingRequest.builder().model(model).input(List.of(text)).build();
 
 			EmbeddingResult result = openAiService.createEmbeddings(request);
 
 			if (!result.getData().isEmpty()) {
-				List<Double> embedding = result.getData().get(0).getEmbedding();
-				return embedding.stream().mapToDouble(Double::doubleValue).toArray();
+                List<Double> embedding = result.getData().get(0).getEmbedding();
+                return embedding.toArray(new Double[0]);
 			}
 
 			throw new RuntimeException("No embedding generated");
@@ -76,7 +76,7 @@ public double[] embed(String text) {
 	}
 
 	@Override
-	public double[][] embed(String[] texts) {
+	public Double[][] embed(String[] texts) {
 
 		try {
 			EmbeddingRequest request = EmbeddingRequest.builder().model(model).input(Arrays.asList(texts)).build();
@@ -86,7 +86,7 @@ public double[][] embed(String[] texts) {
 			return result.getData()
 				.stream()
 				.map(data -> data.getEmbedding().stream().mapToDouble(Double::doubleValue).toArray())
-				.toArray(double[][]::new);
+				.toArray(Double[][]::new);
 		}
 		catch (Exception e) {
 
@@ -96,12 +96,12 @@ public double[][] embed(String[] texts) {
 	}
 
 	@Override
-	public int getDimension() {
+	public Integer getDimension() {
 		return dimension;
 	}
 
 	@Override
-	public boolean isAvailable() {
+	public Boolean isAvailable() {
 
 		try {
 			// Simple test to check if service is available

src/main/java/com/langmem/mem4j/llms/DashScopeLLMService.java

@@ -22,6 +22,7 @@
 import com.langmem.mem4j.memory.Message;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.context.annotation.Primary;
 import org.springframework.http.HttpEntity;
 import org.springframework.http.HttpHeaders;
 import org.springframework.http.MediaType;
@@ -40,6 +41,7 @@
  */
 
 @Service
+@Primary
 public class DashScopeLLMService implements LLMService {
 
 	private static final Logger logger = LoggerFactory.getLogger(DashScopeLLMService.class);

src/main/java/com/langmem/mem4j/memory/Memory.java

@@ -78,7 +78,7 @@ public void add(List<Message> messages, String userId, Map<String, Object> metad
 
 			// Generate embeddings and store
 			for (MemoryItem item : memoryItems) {
-				double[] embedding = embeddingService.embed(item.getContent());
+                Double[] embedding = embeddingService.embed(item.getContent());
 				item.setEmbedding(embedding);
 				vectorStoreService.add(item);
 			}
@@ -107,7 +107,7 @@ public List<MemoryItem> search(String query, String userId, Map<String, Object>
 
 		try {
 			// Generate embedding for query
-			double[] queryEmbedding = embeddingService.embed(query);
+            Double[] queryEmbedding = embeddingService.embed(query);
 
 			// Build search filters
 			Map<String, Object> searchFilters = buildSearchFilters(userId, filters);
@@ -155,7 +155,7 @@ public void update(String memoryId, Map<String, Object> data) {
 				if (data.containsKey("content")) {
 					item.setContent((String) data.get("content"));
 					// Re-embed if content changed
-					double[] embedding = embeddingService.embed(item.getContent());
+                    Double[] embedding = embeddingService.embed(item.getContent());
 					item.setEmbedding(embedding);
 				}
 				if (data.containsKey("metadata")) {

src/main/java/com/langmem/mem4j/memory/MemoryItem.java

@@ -56,7 +56,7 @@ public class MemoryItem {
 	private Instant updatedAt;
 
 	@JsonProperty("embedding")
-	private double[] embedding;
+	private Double[] embedding;
 
 	public MemoryItem() {
 		this.createdAt = Instant.now();
@@ -159,11 +159,11 @@ public void setUpdatedAt(Instant updatedAt) {
 		this.updatedAt = updatedAt;
 	}
 
-	public double[] getEmbedding() {
+	public Double[] getEmbedding() {
 		return embedding;
 	}
 
-	public void setEmbedding(double[] embedding) {
+	public void setEmbedding(Double[] embedding) {
 		this.embedding = embedding;
 	}
 

src/main/java/com/langmem/mem4j/vectorstores/InMemoryVectorStoreService.java

@@ -37,7 +37,7 @@ public class InMemoryVectorStoreService implements VectorStoreService {
 
 	private final Map<String, MemoryItem> memoryStore = new ConcurrentHashMap<>();
 
-	private final Map<String, double[]> embeddings = new ConcurrentHashMap<>();
+	private final Map<String, Double[]> embeddings = new ConcurrentHashMap<>();
 
 	@Override
 	public void add(MemoryItem item) {
@@ -58,7 +58,7 @@ public void add(MemoryItem item) {
 	}
 
 	@Override
-	public List<MemoryItem> search(double[] queryEmbedding, Map<String, Object> filters, int limit, double threshold) {
+	public List<MemoryItem> search(Double[] queryEmbedding, Map<String, Object> filters, Integer limit, Double threshold) {
 
 		try {
 			return memoryStore.values().stream().filter(item -> matchesFilters(item, filters)).map(item -> {
@@ -73,7 +73,7 @@ public List<MemoryItem> search(double[] queryEmbedding, Map<String, Object> filt
 				result.setUpdatedAt(item.getUpdatedAt());
 
 				// Calculate similarity score
-				double[] itemEmbedding = embeddings.get(item.getId());
+                Double[] itemEmbedding = embeddings.get(item.getId());
 				if (itemEmbedding != null) {
 					double similarity = cosineSimilarity(queryEmbedding, itemEmbedding);
 					result.setScore(similarity);
@@ -94,7 +94,7 @@ public List<MemoryItem> search(double[] queryEmbedding, Map<String, Object> filt
 	}
 
 	@Override
-	public List<MemoryItem> getAll(Map<String, Object> filters, int limit) {
+	public List<MemoryItem> getAll(Map<String, Object> filters, Integer limit) {
 
 		try {
 			return memoryStore.values()
@@ -216,7 +216,7 @@ private boolean matchesFilters(MemoryItem item, Map<String, Object> filters) {
 		});
 	}
 
-	private double cosineSimilarity(double[] a, double[] b) {
+	private Double cosineSimilarity(Double[] a, Double[] b) {
 
 		if (a.length != b.length) {
 			return 0.0;