-
Notifications
You must be signed in to change notification settings - Fork 135
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
test: Add tests for VertexAIChatGeminiGenerator and migrate from preview #1042
Conversation
@julian-risch worst case scenario, merging changes with testing failing only on windows is an option? |
You should also add a serialization/deserialization test of a pipeline containing a VertexAIChatGeminiGenerator |
@Amnah199 here's an example of the type of test I mentioned above: |
The QdrantDocumentStore did not properly implement the DocumentStore protocol.
* fix: Add NONE option to EmbeddingTruncateMode * refactor: Validate input with _missing_ method * test: Add EmbeddingTruncateMode test * refactor: Revert "refactor: Validate input with _missing_ method" This reverts commit 8334a50. * Update integrations/nvidia/src/haystack_integrations/components/embedders/nvidia/truncate.py --------- Co-authored-by: Madeesh Kannan <[email protected]>
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.
Left one comment about tool definition
assert gemini._model_name == "gemini-1.5-flash" | ||
assert gemini._project_id == "TestID123" | ||
assert gemini._safety_settings == {HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_ONLY_HIGH} | ||
# assert repr(gemini._tools) == repr([Tool(function_declarations=[GET_CURRENT_WEATHER_FUNC])]) |
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.
Why comment out?
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.
Was by mistake. Just fixed it 👍
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.
LGTM
…iew package in vertexai (#1042) * Add tests for chat generator and migrate from preview package to a stable version of vertexai generative_model
Related Issues
Proposed Changes:
preview
package to stable versionVertexAIChatGeminiGenerator
How did you test it?
Notes for the reviewer
Checklist
fix:
,feat:
,build:
,chore:
,ci:
,docs:
,style:
,refactor:
,perf:
,test:
.