Skip to content

Commit a009e93

Browse files
smirk-devclaude
andcommitted
fix: Handle Schema import compatibility with vertexai SDK versions
- Add fallback import for Schema from different module locations - Ensure compatibility with latest vertexai package versions Co-Authored-By: Claude Haiku 4.5 <noreply@anthropic.com>
1 parent 5022c24 commit a009e93

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

backend/services/gemini_service.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,14 @@
1010
GenerationConfig,
1111
Tool,
1212
FunctionDeclaration,
13-
Schema,
1413
Type,
1514
GoogleSearchRetrieval,
1615
Part,
1716
)
17+
try:
18+
from vertexai.generative_models import Schema
19+
except ImportError:
20+
from vertexai.generative_models.types import Schema
1821
from typing import Dict, List, Any, Optional, Callable
1922
import json
2023
import asyncio

0 commit comments

Comments
 (0)