akhaliq HF Staff commited on
Commit
ed64574
·
1 Parent(s): 73cedf5

add new gemini models

Browse files
Files changed (1) hide show
  1. app.py +23 -1
app.py CHANGED
@@ -1833,6 +1833,16 @@ AVAILABLE_MODELS = [
1833
  "id": "gemini-2.5-pro",
1834
  "description": "Google Gemini 2.5 Pro via OpenAI-compatible API"
1835
  },
 
 
 
 
 
 
 
 
 
 
1836
  {
1837
  "name": "GPT-OSS-120B",
1838
  "id": "openai/gpt-oss-120b",
@@ -1886,7 +1896,7 @@ AVAILABLE_MODELS = [
1886
  ]
1887
 
1888
  # Default model selection
1889
- DEFAULT_MODEL_NAME = "Qwen3-Max-2025-09-23"
1890
  DEFAULT_MODEL = None
1891
  for _m in AVAILABLE_MODELS:
1892
  if _m.get("name") == DEFAULT_MODEL_NAME:
@@ -2063,6 +2073,18 @@ def get_inference_client(model_id, provider="auto"):
2063
  api_key=os.getenv("GEMINI_API_KEY"),
2064
  base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
2065
  )
 
 
 
 
 
 
 
 
 
 
 
 
2066
  elif model_id == "kimi-k2-turbo-preview":
2067
  # Use Moonshot AI (OpenAI-compatible) client for Kimi K2 Turbo (Preview)
2068
  return OpenAI(
 
1833
  "id": "gemini-2.5-pro",
1834
  "description": "Google Gemini 2.5 Pro via OpenAI-compatible API"
1835
  },
1836
+ {
1837
+ "name": "Gemini Flash Latest",
1838
+ "id": "gemini-flash-latest",
1839
+ "description": "Google Gemini Flash Latest model via native Gemini API"
1840
+ },
1841
+ {
1842
+ "name": "Gemini Flash Lite Latest",
1843
+ "id": "gemini-flash-lite-latest",
1844
+ "description": "Google Gemini Flash Lite Latest model via OpenAI-compatible API"
1845
+ },
1846
  {
1847
  "name": "GPT-OSS-120B",
1848
  "id": "openai/gpt-oss-120b",
 
1896
  ]
1897
 
1898
  # Default model selection
1899
+ DEFAULT_MODEL_NAME = "Gemini Flash Latest"
1900
  DEFAULT_MODEL = None
1901
  for _m in AVAILABLE_MODELS:
1902
  if _m.get("name") == DEFAULT_MODEL_NAME:
 
2073
  api_key=os.getenv("GEMINI_API_KEY"),
2074
  base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
2075
  )
2076
+ elif model_id == "gemini-flash-latest":
2077
+ # Use Google Gemini Flash Latest (OpenAI-compatible) client
2078
+ return OpenAI(
2079
+ api_key=os.getenv("GEMINI_API_KEY"),
2080
+ base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
2081
+ )
2082
+ elif model_id == "gemini-flash-lite-latest":
2083
+ # Use Google Gemini Flash Lite Latest (OpenAI-compatible) client
2084
+ return OpenAI(
2085
+ api_key=os.getenv("GEMINI_API_KEY"),
2086
+ base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
2087
+ )
2088
  elif model_id == "kimi-k2-turbo-preview":
2089
  # Use Moonshot AI (OpenAI-compatible) client for Kimi K2 Turbo (Preview)
2090
  return OpenAI(