bibibi12345 commited on
Commit
6838d8b
·
1 Parent(s): afcace3

added safety score. added proxy

Browse files
Files changed (1) hide show
  1. app/vertex_ai_init.py +2 -1
app/vertex_ai_init.py CHANGED
@@ -1,8 +1,9 @@
1
  import json
 
2
  from google import genai
3
  from credentials_manager import CredentialManager, parse_multiple_json_credentials
4
  import config as app_config
5
- from google.generativeai.client import types
6
  from model_loader import refresh_models_config_cache # Import new model loader function
7
 
8
  # VERTEX_EXPRESS_MODELS list is now dynamically loaded via model_loader
 
1
  import json
2
+ from typing import Optional
3
  from google import genai
4
  from credentials_manager import CredentialManager, parse_multiple_json_credentials
5
  import config as app_config
6
+ from google.genai import types
7
  from model_loader import refresh_models_config_cache # Import new model loader function
8
 
9
  # VERTEX_EXPRESS_MODELS list is now dynamically loaded via model_loader