Skip to content

Commit

Permalink
Merge branch 'main' into get_delete_embeddings_space
Browse files Browse the repository at this point in the history
  • Loading branch information
juanjoman committed Aug 31, 2023
2 parents b0037b9 + 4968a61 commit 0575bbe
Show file tree
Hide file tree
Showing 5 changed files with 4 additions and 29 deletions.
26 changes: 1 addition & 25 deletions docs-site/docs/03_components/02_embeddings_space.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,41 +23,17 @@ EmbeddingsStore pineconeStore = new PineconeEmbeddingsStore("YOUR_PINECONE_URL",
EmbeddingsSpaceComponent embeddingsSpace = new EmbeddingsSpaceComponent(openAIModel, pineconeStore);
```

## `generate`

Generates an embedding from a text without persisting it.

- **Parameters**:
- `text`: The textual input for embedding.
- `additionalMetadata`: Supplementary metadata associated with the text.
- **Returns**: The generated embedding.

```java
String sampleText = "Hello, eLLMental!";
Map<String, String> additionalMetadata = new HashMap<>();
additionalMetadata.put("key", "value");

Embedding embedding = embeddingsSpace.generate(sampleText, additionalMetadata);
```

## `save`

Generates and persists an embedding for a given text.

- **Parameters**:
- `text`: Text to be embedded.
- `additionalMetadata`: (Optional) Additional metadata.
- **Returns**: The generated embedding.

```java
Map<String, String> additionalMetadata = new HashMap<>();
additionalMetadata.put("key", "value");

String sampleText = "Hello, eLLMental!";
Embedding embedding = embeddingsSpace.save(sampleText, additionalMetadata);

// Or just
Embedding embedding = embeddingSpace.save(sampleText);
Embedding embedding = embeddingsSpace.save(sampleText);
```

## `mostSimilarEmbeddings`
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
import java.util.List;

public class OpenAIEmbeddingsModelTest {
@Test
@Test
public void testGenerateEmbedding(){
OpenAIEmbeddingsModel openAI = new OpenAIEmbeddingsModel("fakeAPIKey");
Embedding embedding = openAI.generateEmbedding("The Agile Monkeys rule!");
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -41,5 +41,4 @@ public Embedding get(UUID uuid) {
public void delete(UUID uuid) {
embeddingsStore.delete(uuid);
}

}
Original file line number Diff line number Diff line change
Expand Up @@ -97,4 +97,4 @@ public void testDeleteEmbedding() {

verifyNoMoreInteractions(embeddingsGenerationModel, embeddingsStore);
}
}
}
2 changes: 1 addition & 1 deletion settings.gradle
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,6 @@ include(
"core",
"modules:embeddingsgeneration",
"modules:embeddingsstore",
"modules:embeddingsspace",
"examples:simplejava"
)

0 comments on commit 0575bbe

Please sign in to comment.