AbuAlone09 commited on
Commit
ec7c4de
·
verified ·
1 Parent(s): e883ec5

Update backend_models.py

Browse files
Files changed (1) hide show
  1. backend_models.py +14 -9
backend_models.py CHANGED
@@ -1,21 +1,26 @@
1
  import os
2
  import google.generativeai as genai
3
 
4
- # Cấu hình Gemini làm "xương sống" cho toàn hệ thống
5
- genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
 
 
 
 
6
 
7
- # Mapping: Bất kỳ model nào trên giao diện cũng sẽ trỏ về 1 trong 2 model của Google
8
  def get_inference_client(model_id: str, provider: str = "auto"):
9
- # Gemini 1.5 Flash là model cực nhanh, Flash-8B cực tiết kiệm
10
- model_name = 'gemini-1.5-flash'
11
- return genai.GenerativeModel(model_name)
 
 
12
 
13
  def get_real_model_id(model_id: str) -> str:
14
- # Ở đây ta "lừa" hệ thống: frontend yêu cầu Kimi hay Qwen,
15
- # ta vẫn trả về id của Gemini để backend không bị lỗi 404
16
  return "gemini-1.5-flash"
17
 
18
- # Giữ nguyên các hàm này để backend_api không báo lỗi thiếu thư viện
19
  def is_native_sdk_model(model_id: str) -> bool:
20
  return True
21
 
 
1
  import os
2
  import google.generativeai as genai
3
 
4
+ # Cấu hình API Key
5
+ api_key = os.getenv("GEMINI_API_KEY")
6
+ genai.configure(api_key=api_key)
7
+
8
+ # Khởi tạo model một lần duy nhất tại đây
9
+ # Chúng ta bỏ qua bước 'models/' để tránh lỗi đường dẫn
10
+ model = genai.GenerativeModel('gemini-1.5-flash')
11
 
 
12
  def get_inference_client(model_id: str, provider: str = "auto"):
13
+ """
14
+ Luôn trả về instance model đã khởi tạo,
15
+ không gọi bất kỳ API kiểm tra model nào của Hugging Face.
16
+ """
17
+ return model
18
 
19
  def get_real_model_id(model_id: str) -> str:
20
+ """Luôn trả về giá trị cố định để backend không bị lạc"""
 
21
  return "gemini-1.5-flash"
22
 
23
+ # Các hàm phụ trợ để tránh lỗi Import trong backend_api.py
24
  def is_native_sdk_model(model_id: str) -> bool:
25
  return True
26