AbuAlone09 commited on
Commit
e35c4e0
·
verified ·
1 Parent(s): 22b7b4c

Update backend_models.py

Browse files
Files changed (1) hide show
  1. backend_models.py +4 -10
backend_models.py CHANGED
@@ -1,23 +1,17 @@
1
  import os
2
  import google.generativeai as genai
3
 
4
- # Cấu hình API Key của bạn
5
- # Bạn nên lưu key này trong phần 'Secrets' của Hugging Face với tên GEMINI_API_KEY
6
  genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
7
 
8
  def get_inference_client(model_id: str, provider: str = "auto"):
9
  """
10
- Trả về model Gemini để xử lý yêu cầu.
11
  """
12
- # Gemini 1.5 Flash rất nhanh và thông minh cho tác vụ code
13
  model = genai.GenerativeModel('gemini-1.5-flash')
14
  return model
15
 
16
  def get_real_model_id(model_id: str) -> str:
 
17
  return "gemini-1.5-flash"
18
-
19
- def is_native_sdk_model(model_id: str) -> bool:
20
- return True
21
-
22
- def is_mistral_model(model_id: str) -> bool:
23
- return False
 
1
  import os
2
  import google.generativeai as genai
3
 
4
+ # Cấu hình API Key
 
5
  genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
6
 
7
  def get_inference_client(model_id: str, provider: str = "auto"):
8
  """
9
+ Sử dụng model 'gemini-1.5-flash' chuẩn xác.
10
  """
11
+ # Khởi tạo model đúng cách
12
  model = genai.GenerativeModel('gemini-1.5-flash')
13
  return model
14
 
15
  def get_real_model_id(model_id: str) -> str:
16
+ # Trả về tên model chính xác mà Google AI Studio yêu cầu
17
  return "gemini-1.5-flash"