Skip to content

Fix DeepSeek4HyperHead instantiation and usage in NNXDecoder - #4782

Closed
olufiyin19 wants to merge 2 commits into
test-pipeline-ckpt-validationfrom
fix-validation-pipeline-deepseek4-284b-dag_verify_checkpoint_shape-manual__dbb8774c-checkpoint_shape_validation
Closed

Fix DeepSeek4HyperHead instantiation and usage in NNXDecoder#4782
olufiyin19 wants to merge 2 commits into
test-pipeline-ckpt-validationfrom
fix-validation-pipeline-deepseek4-284b-dag_verify_checkpoint_shape-manual__dbb8774c-checkpoint_shape_validation

Conversation

@olufiyin19

Copy link
Copy Markdown
Collaborator

Fix DeepSeek4HyperHead instantiation and usage in NNXDecoder

@google-cla

google-cla Bot commented Aug 7, 2026

Copy link
Copy Markdown

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.

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

Code Review

This pull request adds the Google Cloud CLI to the agent sidecar Dockerfile, increases the retry count and sleep duration for Gemini API calls, adds a mock failure log, and removes DeepSeek4-specific hyper-head logic from the NNX decoders. Feedback recommends using HTTPS and curl -fsSL for downloading the Google Cloud SDK keyring to ensure secure and reliable package installation, and updating the docstring in adk_agent.py to match the new 60-second default sleep time.

Comment on lines +10 to +12
echo "deb [signed-by=/usr/share/keyrings/cloud.google.gpg] http://packages.cloud.google.com/apt cloud-sdk main" | tee -a /etc/apt/sources.list.d/google-cloud-sdk.list && \
curl https://packages.cloud.google.com/apt/doc/apt-key.gpg | gpg --dearmor -o /usr/share/keyrings/cloud.google.gpg && \
apt-get update && apt-get install -y gh google-cloud-cli && \

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

security-medium medium

Use https instead of http for the Google Cloud SDK apt repository to ensure secure package metadata transport. Additionally, use curl -fsSL to ensure that any download failures are propagated and do not silently result in an invalid or empty keyring file.

    echo "deb [signed-by=/usr/share/keyrings/cloud.google.gpg] https://packages.cloud.google.com/apt cloud-sdk main" | tee -a /etc/apt/sources.list.d/google-cloud-sdk.list && \
    curl -fsSL https://packages.cloud.google.com/apt/doc/apt-key.gpg | gpg --dearmor -o /usr/share/keyrings/cloud.google.gpg && \
    apt-get update && apt-get install -y gh google-cloud-cli && \



def _send_message_with_retry(chat, prompt, max_retries=3, sleep_seconds=30):
def _send_message_with_retry(chat, prompt, max_retries=5, sleep_seconds=60):

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

medium

The default value of sleep_seconds has been updated from 30 to 60, but the docstring on the following line still states: Sends a message to Gemini with retry and a 30-second sleep.... Please update the docstring to reflect the new 60-second default sleep time to keep the documentation accurate.

@olufiyin19 olufiyin19 closed this Aug 8, 2026
@olufiyin19
olufiyin19 deleted the fix-validation-pipeline-deepseek4-284b-dag_verify_checkpoint_shape-manual__dbb8774c-checkpoint_shape_validation branch August 8, 2026 00:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant