Skip to content

Commit

Permalink
Rename embeddings index() function to load()
Browse files Browse the repository at this point in the history
  • Loading branch information
Justin Reese committed Mar 7, 2024
1 parent 1578efc commit 31cf525
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/curate_gpt/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -1718,15 +1718,15 @@ def load_embeddings(file_path, embedding_format=None):
return df.to_dict(orient='records')


@embeddings.command(name="index")
@embeddings.command(name="load")
@path_option
@collection_option
@model_option
@append_option
@click.option("--embedding-format", "-f",
type=click.Choice(['parquet', 'csv']), help="Format of the input file")
@click.argument("file_or_url")
def index(path, collection, append, embedding_format, model, file_or_url):
def load_embeddings(path, collection, append, embedding_format, model, file_or_url):
"""
Index embeddings from a local file or URL into a ChromaDB collection.
"""
Expand Down

0 comments on commit 31cf525

Please sign in to comment.