Skip to content

Prod 1785 enable adding any embedding in ai r not just embedding model #514

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Conversation

Muhammad-Elmallah
Copy link
Collaborator

  1. Removed private models from SDK
  2. Enabled calling aiR with any embedding model ID
  3. Add validation for the model (validate_embedding_model)

from urllib.parse import urljoin

def validate_embedding_model(model_id) -> bool:
resp = None
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Use ModelFactory instead. Please also use pre-commit to fix the indentation

@@ -21,13 +21,13 @@


class EmbeddingModel(str, Enum):
SNOWFLAKE_ARCTIC_EMBED_M_LONG = "6658d40729985c2cf72f42ec"
# SNOWFLAKE_ARCTIC_EMBED_M_LONG = "6658d40729985c2cf72f42ec"
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Remove comment

@Muhammad-Elmallah Muhammad-Elmallah force-pushed the PROD-1785-enable-adding-any-embedding-in-ai-r-not-just-embedding-model branch from 2e9ed57 to 8c95f35 Compare May 21, 2025 17:37
@Muhammad-Elmallah Muhammad-Elmallah merged commit ad4ddbf into development May 21, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants