Skip to content

Commit e969354

Browse files
committed
style: standardize code formatting and indentation across multiple files
1 parent a478d2e commit e969354

File tree

6 files changed

+21
-19
lines changed

6 files changed

+21
-19
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ public Double[][] embed(String[] texts) {
126126
JsonNode responseJson = objectMapper.readTree(response.getBody());
127127
JsonNode embeddings = responseJson.path("output").path("embeddings");
128128

129-
Double[][] result = new Double[embeddings.size()][];
129+
Double[][] result = new Double[embeddings.size()][];
130130
for (int i = 0; i < embeddings.size(); i++) {
131131
JsonNode embeddingArray = embeddings.get(i).path("embedding");
132132
result[i] = new Double[embeddingArray.size()];

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

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -23,14 +23,14 @@
2323
public interface EmbeddingService {
2424

2525
/**
26-
* Generate embedding for a text
27-
*/
28-
Double[] embed(String text);
26+
* Generate embedding for a text
27+
*/
28+
Double[] embed(String text);
2929

3030
/**
31-
* Generate embeddings for multiple texts
32-
*/
33-
Double[][] embed(String[] texts);
31+
* Generate embeddings for multiple texts
32+
*/
33+
Double[][] embed(String[] texts);
3434

3535
/**
3636
* Get the dimension of embeddings

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,8 +62,8 @@ public Double[] embed(String text) {
6262
EmbeddingResult result = openAiService.createEmbeddings(request);
6363

6464
if (!result.getData().isEmpty()) {
65-
List<Double> embedding = result.getData().get(0).getEmbedding();
66-
return embedding.toArray(new Double[0]);
65+
List<Double> embedding = result.getData().get(0).getEmbedding();
66+
return embedding.toArray(new Double[0]);
6767
}
6868

6969
throw new RuntimeException("No embedding generated");

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ public void add(List<Message> messages, String userId, Map<String, Object> metad
7878

7979
// Generate embeddings and store
8080
for (MemoryItem item : memoryItems) {
81-
Double[] embedding = embeddingService.embed(item.getContent());
81+
Double[] embedding = embeddingService.embed(item.getContent());
8282
item.setEmbedding(embedding);
8383
vectorStoreService.add(item);
8484
}
@@ -107,7 +107,7 @@ public List<MemoryItem> search(String query, String userId, Map<String, Object>
107107

108108
try {
109109
// Generate embedding for query
110-
Double[] queryEmbedding = embeddingService.embed(query);
110+
Double[] queryEmbedding = embeddingService.embed(query);
111111

112112
// Build search filters
113113
Map<String, Object> searchFilters = buildSearchFilters(userId, filters);
@@ -155,7 +155,7 @@ public void update(String memoryId, Map<String, Object> data) {
155155
if (data.containsKey("content")) {
156156
item.setContent((String) data.get("content"));
157157
// Re-embed if content changed
158-
Double[] embedding = embeddingService.embed(item.getContent());
158+
Double[] embedding = embeddingService.embed(item.getContent());
159159
item.setEmbedding(embedding);
160160
}
161161
if (data.containsKey("metadata")) {

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,8 @@ public void add(MemoryItem item) {
5858
}
5959

6060
@Override
61-
public List<MemoryItem> search(Double[] queryEmbedding, Map<String, Object> filters, Integer limit, Double threshold) {
61+
public List<MemoryItem> search(Double[] queryEmbedding, Map<String, Object> filters, Integer limit,
62+
Double threshold) {
6263

6364
try {
6465
return memoryStore.values().stream().filter(item -> matchesFilters(item, filters)).map(item -> {
@@ -73,7 +74,7 @@ public List<MemoryItem> search(Double[] queryEmbedding, Map<String, Object> filt
7374
result.setUpdatedAt(item.getUpdatedAt());
7475

7576
// Calculate similarity score
76-
Double[] itemEmbedding = embeddings.get(item.getId());
77+
Double[] itemEmbedding = embeddings.get(item.getId());
7778
if (itemEmbedding != null) {
7879
double similarity = cosineSimilarity(queryEmbedding, itemEmbedding);
7980
result.setScore(similarity);

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

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -110,15 +110,16 @@ public void add(MemoryItem item) {
110110
}
111111

112112
@Override
113-
public List<MemoryItem> search(Double[] queryEmbedding, Map<String, Object> filters, Integer limit, Double threshold) {
113+
public List<MemoryItem> search(Double[] queryEmbedding, Map<String, Object> filters, Integer limit,
114+
Double threshold) {
114115
try {
115116
// Simple similarity search using in-memory storage
116117
List<MemoryItem> results = memoryStore.values()
117118
.stream()
118119
.filter(item -> matchesFilters(item, filters))
119120
.map(item -> {
120121
// Calculate simple cosine similarity
121-
Double similarity = calculateCosineSimilarity(queryEmbedding, item.getEmbedding());
122+
Double similarity = calculateCosineSimilarity(queryEmbedding, item.getEmbedding());
122123
// Store similarity for sorting
123124
return new AbstractMap.SimpleEntry<>(similarity, item);
124125
})
@@ -298,9 +299,9 @@ private Double calculateCosineSimilarity(Double[] vec1, Double[] vec2) {
298299
return 0.0;
299300
}
300301

301-
double dotProduct = 0.0;
302-
double norm1 = 0.0;
303-
double norm2 = 0.0;
302+
double dotProduct = 0.0;
303+
double norm1 = 0.0;
304+
double norm2 = 0.0;
304305

305306
for (int i = 0; i < vec1.length; i++) {
306307
dotProduct += vec1[i] * vec2[i];

0 commit comments

Comments
 (0)