-
Notifications
You must be signed in to change notification settings - Fork 199
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
[Draft]: Vertex AI model garden with OpenAI Compatibility #2336
Conversation
Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA). View this failed invocation of the CLA check for more information. For the most up to date status, view the checks section at the bottom of the pull request. |
@zarinn3pal added you to corp CLA. Can you please retrigger the checks? |
py/plugins/vertex-ai/src/genkit/plugins/vertex_ai/model_garden/__init__.py
Show resolved
Hide resolved
py/plugins/vertex-ai/src/genkit/plugins/vertex_ai/model_garden/model_garden.py
Show resolved
Hide resolved
...lugins/vertex-ai/src/genkit/plugins/vertex_ai/model_garden/openai_compatiblility/__init__.py
Show resolved
Hide resolved
...x-ai/src/genkit/plugins/vertex_ai/model_garden/openai_compatiblility/openai_compatibility.py
Show resolved
Hide resolved
py/plugins/vertex-ai/src/genkit/plugins/vertex_ai/model_garden/modelgarden_plugin.py
Show resolved
Hide resolved
py/plugins/vertex-ai/src/genkit/plugins/vertex_ai/model_garden/modelgarden_plugin.py
Show resolved
Hide resolved
def initialize(self, ai: GenkitRegistry) -> None: | ||
"""Handles actions for various openaicompatible models.""" | ||
for model in self.plugin_options.models: | ||
openai_model = next( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can you just use SUPPORTED_OPENAI_FORMAT_MODELS.get(model)
?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
you are right. modified it
...x-ai/src/genkit/plugins/vertex_ai/model_garden/openai_compatiblility/openai_compatibility.py
Show resolved
Hide resolved
676f346
to
d3630a2
Compare
Created a new PR for this , rebasing from recent main,addressing the PR suggestions New PR: #2568 |
Description here... Help the reviewer by:
Added Model garden to vertexAI and implemented openai_compatibility support
linking to an issue that includes more details
if it's a new feature include samples of how to use the new feature
(optional if issue link is provided) if you fixed a bug include basic bug details
Checklist (if applicable):