Skip to content

Commit 422711e

Browse files
authored
fix(ci): make only Google LLM providers optional (#70)
1 parent a2d0ab3 commit 422711e

5 files changed

Lines changed: 41 additions & 11 deletions

File tree

.github/workflows/scan-skills.yml

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,19 @@ jobs:
103103
python-version: ${{ inputs.python_version }}
104104

105105
- name: Install skill-scanner
106-
run: pip install cisco-ai-skill-scanner
106+
env:
107+
INPUT_LLM_MODEL: ${{ inputs.llm_model }}
108+
run: |
109+
EXTRAS=""
110+
case "$INPUT_LLM_MODEL" in
111+
gemini/*|*gemini*)
112+
EXTRAS="[google]"
113+
;;
114+
vertex_ai/*|vertex/*|*vertex*)
115+
EXTRAS="[vertex]"
116+
;;
117+
esac
118+
pip install "cisco-ai-skill-scanner${EXTRAS}"
107119
108120
- name: Validate inputs
109121
env:

README.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,9 @@ pip install cisco-ai-skill-scanner
8080
# AWS Bedrock support
8181
pip install cisco-ai-skill-scanner[bedrock]
8282

83+
# Google AI Studio / Gemini support
84+
pip install cisco-ai-skill-scanner[google]
85+
8386
# Google Vertex AI support
8487
pip install cisco-ai-skill-scanner[vertex]
8588

docs/reference/dependencies-and-llm-providers.md

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ All versions from [`pyproject.toml`](https://github.com/cisco-ai-defense/skill-s
4848
| `anthropic` | >= 0.40.0 | Anthropic Claude SDK |
4949
| `openai` | >= 1.0.0 | OpenAI SDK |
5050
| `litellm` | >= 1.77.0 | Multi-provider LLM routing |
51-
| `google-genai` | >= 0.2.0 | Google AI SDK |
52-
| `google-generativeai` | >= 0.8.0 | Google Generative AI SDK |
51+
| `google-genai` | optional via `[google]` | Google AI Studio / Gemini SDK |
52+
| `google-generativeai` | optional via `[google]` | Legacy Google Generative AI SDK compatibility |
5353

5454
## Optional Provider Extras
5555

@@ -59,6 +59,9 @@ Install only what you need:
5959
# AWS Bedrock
6060
pip install "cisco-ai-skill-scanner[bedrock]"
6161

62+
# Google AI Studio / Gemini
63+
pip install "cisco-ai-skill-scanner[google]"
64+
6265
# Google Vertex AI
6366
pip install "cisco-ai-skill-scanner[vertex]"
6467

@@ -71,6 +74,7 @@ pip install "cisco-ai-skill-scanner[all]"
7174

7275
| Extra | Package | Version | Purpose |
7376
|-------|---------|---------|---------|
77+
| `google` | `google-genai`, `google-generativeai` | varies | Google AI Studio / Gemini SDK support |
7478
| `bedrock` | `boto3` | >= 1.28.57 | AWS Bedrock IAM credential support |
7579
| `vertex` | `google-cloud-aiplatform` | >= 1.38.0 | Google Vertex AI support |
7680
| `azure` | `azure-identity` | >= 1.15.0 | Azure managed identity auth |
@@ -88,7 +92,7 @@ Set `SKILL_SCANNER_LLM_MODEL` using the provider prefix convention:
8892
| OpenAI | `openai/gpt-4o` | |
8993
| AWS Bedrock | `bedrock/anthropic.claude-sonnet-4-20250514-v1:0` | Requires `[bedrock]` extra or API key |
9094
| Google Vertex AI | `vertex_ai/gemini-2.5-pro` | Requires `[vertex]` extra |
91-
| Google AI Studio | `gemini/gemini-2.5-flash` | Via LiteLLM |
95+
| Google AI Studio | `gemini/gemini-2.5-flash` | Requires `[google]` extra |
9296
| Azure OpenAI | `azure/my-deployment-name` | Requires `[azure]` extra |
9397
| Ollama (local) | `ollama/llama3` | No API key needed |
9498

pyproject.toml

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,11 +71,14 @@ dependencies = [
7171
"anthropic==0.76.0",
7272
"openai==2.15.0",
7373
"litellm==1.82.3",
74-
"google-genai==1.60.0",
75-
"google-generativeai==0.8.6",
7674
]
7775

7876
[project.optional-dependencies]
77+
# Google AI Studio / Gemini SDKs
78+
google = [
79+
"google-genai==1.60.0",
80+
"google-generativeai==0.8.6",
81+
]
7982
# AWS Bedrock support (requires boto3 for IAM credentials)
8083
bedrock = [
8184
"boto3==1.42.37",
@@ -94,6 +97,8 @@ azure = [
9497
# supply-chain = ["guarddog==2.0.0"]
9598
# All optional provider extras
9699
all = [
100+
"google-genai==1.60.0",
101+
"google-generativeai==0.8.6",
97102
"boto3==1.42.37",
98103
"google-cloud-aiplatform==1.135.0",
99104
"azure-identity==1.25.1",

uv.lock

Lines changed: 11 additions & 5 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)