Skip to main content

Get embeddings for code snippets in code

io.moderne.ai.research.GetCodeEmbedding

This recipe calls an AI model to get an embedding for either classes or methods which can then be used for downstream tasks.

Recipe source

This recipe is only available to users of Moderne.

This recipe is available under the Moderne Proprietary License.

Options

TypeNameDescriptionExample
StringcodeSnippetTypeChoose whether you want to get an embedding for the classes or methods. Valid options: methods, classesmethods

Usage

This recipe has required configuration parameters and can only be run by users of Moderne. To run this recipe, you will need to provide the Moderne CLI run command with the required options. Or, if you'd like to create a declarative recipe, please see the below example of a rewrite.yml file:

rewrite.yml
---
type: specs.openrewrite.org/v1beta/recipe
name: com.yourorg.GetCodeEmbeddingExample
displayName: Get embeddings for code snippets in code example
recipeList:
- io.moderne.ai.research.GetCodeEmbedding:
codeSnippetType: methods

You will need to have configured the Moderne CLI on your machine before you can run the following command.

shell
mod run . --recipe GetCodeEmbedding --recipe-option "codeSnippetType=methods"

If the recipe is not available locally, then you can install it using:

mod config recipes jar install org.openrewrite.recipe:rewrite-ai-search:0.27.0

See how this recipe works across multiple open-source repositories

Run this recipe on OSS repos at scale with the Moderne SaaS.

The community edition of the Moderne platform enables you to easily run recipes across thousands of open-source repositories.

Please contact Moderne for more information about safely running the recipes on your own codebase in a private SaaS.

Data Tables

Embeddings

io.moderne.ai.table.Embeddings

Collects the embeddings for either each classes or methods.

Column NameDescription
SourceSource
NameName of the class or method
EmbeddingThe embedding for the code snippet

Contributors

juju, Tim te Beek