Skip to content

Support for Fine-tuning APIs #448

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

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

krishnaagrawal7508
Copy link

What was changed?

  • Implemented fine-tuning support in the Node.js SDK for the Gemini API.
  • Added new methods for fine-tuning inside the GenerativeModel class.
  • Added method to listTunedModels(): Lists all tuned models.
  • Added method to createTunedModel(): Creates a new tuned model using a dataset.
  • Added method to checkTuningStatus(): Checks the status of an ongoing fine-tuning job.
  • Added method to deleteTunedModel(): Deletes a tuned model.
  • Introduced a new fine-tuning helper module in methods/fine-tuning.ts that handles API requests for fine-tuning.
  • Updated unit tests in generative-model.test.ts to cover fine-tuning methods using sinon stubs.

How was this tested?

Added unit tests for fine-tuning API calls using sinon stubs to mock API responses.

Screenshot 2025-03-30 101353

@wanlin31 wanlin31 added p3 status:triaged Issue/PR triaged to the corresponding sub-team labels Mar 31, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
p3 status:triaged Issue/PR triaged to the corresponding sub-team
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants