AbuAlone09 commited on
Commit
4e4d4a3
·
verified ·
1 Parent(s): 5e54cca

Update backend_models.py

Browse files
Files changed (1) hide show
  1. backend_models.py +7 -6
backend_models.py CHANGED
@@ -1,17 +1,18 @@
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"
 
 
 
 
 
 
 
 
1
  import os
2
  import google.generativeai as genai
3
 
 
4
  genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
5
 
6
  def get_inference_client(model_id: str, provider: str = "auto"):
 
 
 
 
7
  model = genai.GenerativeModel('gemini-1.5-flash')
8
  return model
9
 
10
  def get_real_model_id(model_id: str) -> str:
 
11
  return "gemini-1.5-flash"
12
+
13
+ # Thêm lại các hàm này để backend_api không bị lỗi import
14
+ def is_native_sdk_model(model_id: str) -> bool:
15
+ return True
16
+
17
+ def is_mistral_model(model_id: str) -> bool:
18
+ return False