Skip to content
Snippets Groups Projects
Unverified Commit 23af82e3 authored by Yasmine MHIRI's avatar Yasmine MHIRI Committed by GitHub
Browse files

fix: model_name validation error and nomic embed model name (#10421)

parent 6dbecd56
No related branches found
No related tags found
No related merge requests found
...@@ -32,7 +32,7 @@ class NomicEmbedding(BaseEmbedding): ...@@ -32,7 +32,7 @@ class NomicEmbedding(BaseEmbedding):
def __init__( def __init__(
self, self,
model_name: str = "nomic-text-embed-v1", model_name: str = "nomic-embed-text-v1",
embed_batch_size: int = 32, embed_batch_size: int = 32,
api_key: Optional[str] = None, api_key: Optional[str] = None,
callback_manager: Optional[CallbackManager] = None, callback_manager: Optional[CallbackManager] = None,
...@@ -57,6 +57,7 @@ class NomicEmbedding(BaseEmbedding): ...@@ -57,6 +57,7 @@ class NomicEmbedding(BaseEmbedding):
if api_key is not None: if api_key is not None:
nomic.cli.login(api_key) nomic.cli.login(api_key)
super().__init__( super().__init__(
model_name=model_name,
embed_batch_size=embed_batch_size, embed_batch_size=embed_batch_size,
callback_manager=callback_manager, callback_manager=callback_manager,
_model=embed, _model=embed,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment