Skip to content
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

feat: add ml.llm.GeminiTextGenerator model #370

Merged
merged 5 commits into from
Feb 9, 2024

Conversation

GarrettWu
Copy link
Contributor

Thank you for opening a Pull Request! Before submitting your PR, there are a few things you can do to make sure it goes smoothly:

  • Make sure to open an issue as a bug/issue before writing your code! That way we can discuss the change, evaluate designs, and agree on the general idea
  • Ensure the tests and linter pass
  • Code coverage does not decrease (if any source code was changed)
  • Appropriate docs were updated (if necessary)

Fixes #<issue_number_goes_here> 🦕

@GarrettWu GarrettWu requested review from shobsi and ashleyxuu February 6, 2024 19:19
@GarrettWu GarrettWu requested review from a team as code owners February 6, 2024 19:19
@product-auto-label product-auto-label bot added size: m Pull request size is medium. api: bigquery Issues related to the googleapis/python-bigquery-dataframes API. labels Feb 6, 2024
Copy link
Contributor

@ashleyxuu ashleyxuu left a comment

Choose a reason for hiding this comment

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

Please add the GeminiTextGenerator model in the README.rst as well : https://github.com/googleapis/python-bigquery-dataframes/blob/main/README.rst?plain=1#L190

"top_k": top_k,
"top_p": top_p,
"flatten_json_output": True,
}
Copy link
Contributor

Choose a reason for hiding this comment

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

I believe we don't give Python users the option to specify this flatten_json_output as we return a DataFrame format? I guess it is only applicable in SQL interface?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

right. It isn't user-facing, but what we send to SQL. It is always True value.

@@ -107,6 +107,8 @@
uid: bigframes.ml.llm.PaLM2TextGenerator
- name: PaLM2TextEmbeddingGenerator
uid: bigframes.ml.llm.PaLM2TextEmbeddingGenerator
- name: GeminiTextGenerator
Copy link
Contributor

Choose a reason for hiding this comment

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

Should be sorted it in alphabetic order.

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Done

@GarrettWu
Copy link
Contributor Author

Please add the GeminiTextGenerator model in the README.rst as well : https://github.com/googleapis/python-bigquery-dataframes/blob/main/README.rst?plain=1#L190

Done

@GarrettWu GarrettWu requested a review from ashleyxuu February 8, 2024 21:48
@GarrettWu GarrettWu added the automerge Merge the pull request once unit tests and other checks pass. label Feb 9, 2024
@gcf-merge-on-green gcf-merge-on-green bot merged commit de1e0a4 into main Feb 9, 2024
14 of 15 checks passed
@gcf-merge-on-green gcf-merge-on-green bot deleted the garrettwu-gemini branch February 9, 2024 19:56
@gcf-merge-on-green gcf-merge-on-green bot removed the automerge Merge the pull request once unit tests and other checks pass. label Feb 9, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
api: bigquery Issues related to the googleapis/python-bigquery-dataframes API. size: m Pull request size is medium.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants