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.

Sign up for GitHub

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

Update for Gen AI SDK #562

Draft
wants to merge 3 commits into
base: master
Choose a base branch
from
Draft

Update for Gen AI SDK #562

Show file tree
Hide file tree
Changes from 2 commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
288916e
Consolidated dependencies and add google-genai
takumiohym Dec 16, 2024
aa11110
Move saliency to the main requirements.txt
takumiohym Dec 17, 2024
f8dc6e6
update google-genai to 0.3
takumiohym Dec 18, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 0 additions & 3 deletions requirements-without-deps.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,3 @@ huggingface-hub==0.17.2
multiprocess==0.70.15
responses==0.18.0
xxhash==3.3.0

# XAI
saliency==0.2.1
44 changes: 38 additions & 6 deletions requirements.txt
View file
Open in desktop
Copy link
Collaborator Author

@takumiohym takumiohym Dec 17, 2024 •
edited
Loading

Choose a reason for hiding this comment

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

Notes on requirements.txt update

  • Merged 3 kernels and categorized/reordered
  • Added google-genai with the latest version as of 2025/12/16. Will update it later depending on the timing to merge the PR.
  • Added some underlying dependency versions explicitly (e.g. kubernetes, jax, google-api-core, google-cloud-storage) to speed up the installation process by removing the necessity to install multiple versions and check compatibilities.
  • Moved saliency to from the requirements-without-deps.txt to requirements.txt, as its dependencies don't crush with other libraries.

Original file line number Diff line number Diff line change
@@ -1,12 +1,15 @@
# Requirements for asl-ml-immersion repository
google-cloud-aiplatform==1.43.0
google-cloud-pipeline-components==2.8.0

# Google Cloud
google-genai==0.2.2 # tentative version
google-api-core==2.24.0
google-cloud-storage==2.19.0
google-cloud-aiplatform==1.74.0
google-cloud-bigquery==3.21.0
pyyaml==5.3.1
kfp==2.4.0
# tfx==1.13.0
cloudml-hypertune
apache-beam==2.46.0

# Machine Learning Frameworks
tensorflow==2.12.0
tensorflow-io==0.32.0
tensorflow-serving-api==2.12.1
Expand All @@ -17,8 +20,37 @@ tensorflow-model-remediation==0.1.7.1
tensorflow-recommenders==0.7.2
keras_cv==0.5.1
keras_nlp==0.5.1
fire==0.6.0
jax==0.4.30

# MLOps
kubernetes==17.17.0
kfp==2.4.0
google-cloud-pipeline-components==2.8.0
# tfx==1.13.0

# Explainable AI
saliency==0.2.1

# Langchain dependencies
langchain==0.2.12
langsmith==0.1.129
langchain-chroma==0.1.2
chroma-hnswlib==0.7.3
chromadb==0.4.14
langchain-community==0.2.11
langchain-core==0.2.29
langchain-google-vertexai==1.0.8
langchain-text-splitters==0.2.2

# Utilities
pyyaml==5.3.1
pydantic==2.8.2
pypdf==4.3.1
faiss-cpu==1.7.4
unstructured==0.14.4
unstructured-client==0.22.0
wikipedia==1.4.0
fire==0.6.0

# For development work
pre-commit
Expand Down
Loading