Skip to content

Add support to embeddings from VertexAI models#673

Merged
zainhoda merged 2 commits intovanna-ai:mainfrom
gquental:bigquery-vector
Oct 23, 2024
Merged

Add support to embeddings from VertexAI models#673
zainhoda merged 2 commits intovanna-ai:mainfrom
gquental:bigquery-vector

Conversation

@gquental
Copy link
Contributor

Changed the logic to support both Google AI Models and Vertex AI models. The logic is based on the api_key parameter being sent.

Created get_embeddings wrapper to abstract model call.

@gquental
Copy link
Contributor Author

Hello @zainhoda, have you had the time to review this PR? Thank you!

@zainhoda zainhoda merged commit 6145610 into vanna-ai:main Oct 23, 2024
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