helloguddn (PC helloguddn) commited on
Commit
9b49a5b
·
1 Parent(s): f4bf832

fix: 모델 수정

Browse files
agents/main_agent.py CHANGED
@@ -94,7 +94,7 @@ async def run_agent(request: ChatRequest) -> ChatResponse:
94
  system = _SYSTEM_PROMPT.replace("{LANGUAGE}", lang)
95
 
96
  model = genai.GenerativeModel(
97
- model_name="gemini-2.0-flash",
98
  tools=[search_labor_law, find_nearby_centers, get_center_info],
99
  system_instruction=system,
100
  )
 
94
  system = _SYSTEM_PROMPT.replace("{LANGUAGE}", lang)
95
 
96
  model = genai.GenerativeModel(
97
+ model_name="gemini-3.1-pro",
98
  tools=[search_labor_law, find_nearby_centers, get_center_info],
99
  system_instruction=system,
100
  )
data/labor_law_chunks.json CHANGED
@@ -3,7 +3,7 @@
3
  "id": "ll_001",
4
  "topic": "최저임금",
5
  "law": "최저임금법 제6조·제28조",
6
- "content": "2024년 최저임금은 시간당 9,860원입니다. 사용자는 반드시 최저임금 이상을 지급해야 하며, 이를 위반하면 3년 이하 징역 또는 2천만 원 이하 벌금에 처합니다. 식비·교통비 등 복리후생 명목 금품은 최저임금 산입에서 제외됩니다. 수습 기간(3개월 이내)에는 최저임금의 90%까지 지급 가능하나, 단순노무 직종에는 적용되지 않습니다.",
7
  "keywords": ["최저임금", "시급", "임금", "급여", "월급", "시간당"],
8
  "violation_patterns": ["최저임금 미달", "시급 미달", "수습 90% 적용 오남용"]
9
  },
 
3
  "id": "ll_001",
4
  "topic": "최저임금",
5
  "law": "최저임금법 제6조·제28조",
6
+ "content": "2025년 최저임금은 시간당 10,030원입니다. 사용자는 반드시 최저임금 이상을 지급해야 하며, 이를 위반하면 3년 이하 징역 또는 2천만 원 이하 벌금에 처합니다. 식비·교통비 등 복리후생 명목 금품은 최저임금 산입에서 제외됩니다. 수습 기간(3개월 이내)에는 최저임금의 90%까지 지급 가능하나, 단순노무 직종에는 적용되지 않습니다.",
7
  "keywords": ["최저임금", "시급", "임금", "급여", "월급", "시간당"],
8
  "violation_patterns": ["최저임금 미달", "시급 미달", "수습 90% 적용 오남용"]
9
  },
services/gemini_service.py CHANGED
@@ -9,5 +9,5 @@ def configure() -> None:
9
  genai.configure(api_key=api_key)
10
 
11
 
12
- def get_model(model_name: str = "gemini-2.0-flash", **kwargs) -> genai.GenerativeModel:
13
  return genai.GenerativeModel(model_name=model_name, **kwargs)
 
9
  genai.configure(api_key=api_key)
10
 
11
 
12
+ def get_model(model_name: str = "gemini-3.1-pro", **kwargs) -> genai.GenerativeModel:
13
  return genai.GenerativeModel(model_name=model_name, **kwargs)
tools/contract_scanner.py CHANGED
@@ -12,7 +12,7 @@ from services import gemini_service
12
  _SCAN_PROMPT = """당신은 한국 노동법 전문가입니다. 첨부된 근로계약서 이미지를 면밀히 분석하세요.
13
 
14
  아래 항목을 반드시 검토하세요:
15
- 1. 임금 - 2024년 최저임금(시급 9,860원) 이상인지, 식대·교통비 별도 지급 여부
16
  2. 근로시간 - 1일 8시간·주 40시간 초과 여부
17
  3. 초과근무 수당 - 통상임금 50% 가산 명시 여부
18
  4. 연차유급휴가 - 법정 연차 보장 여부
@@ -51,7 +51,7 @@ async def scan_contract_image(image_base64: str, language: str = "ko") -> dict:
51
  lang_name = _LANG_MAP.get(language, "한국어")
52
  prompt = _SCAN_PROMPT.replace("{{LANGUAGE}}", lang_name)
53
 
54
- model = genai.GenerativeModel("gemini-2.0-flash")
55
 
56
  import asyncio
57
  response = await asyncio.to_thread(model.generate_content, [image, prompt])
 
12
  _SCAN_PROMPT = """당신은 한국 노동법 전문가입니다. 첨부된 근로계약서 이미지를 면밀히 분석하세요.
13
 
14
  아래 항목을 반드시 검토하세요:
15
+ 1. 임금 - 2025년 최저임금(시급 10,030원) 이상인지, 식대·교통비 별도 지급 여부
16
  2. 근로시간 - 1일 8시간·주 40시간 초과 여부
17
  3. 초과근무 수당 - 통상임금 50% 가산 명시 여부
18
  4. 연차유급휴가 - 법정 연차 보장 여부
 
51
  lang_name = _LANG_MAP.get(language, "한국어")
52
  prompt = _SCAN_PROMPT.replace("{{LANGUAGE}}", lang_name)
53
 
54
+ model = genai.GenerativeModel("gemini-3.1-pro")
55
 
56
  import asyncio
57
  response = await asyncio.to_thread(model.generate_content, [image, prompt])
tools/translation_tool.py CHANGED
@@ -26,7 +26,7 @@ async def translate_text(text: str, target_language: str = "vi") -> str:
26
  gemini_service.configure()
27
  target = _LANG_NAMES.get(target_language, "Vietnamese")
28
 
29
- model = genai.GenerativeModel("gemini-2.0-flash")
30
  prompt = (
31
  f"Translate the following text to {target}. "
32
  "Output ONLY the translation, no explanations:\n\n"
 
26
  gemini_service.configure()
27
  target = _LANG_NAMES.get(target_language, "Vietnamese")
28
 
29
+ model = genai.GenerativeModel("gemini-3.1-pro")
30
  prompt = (
31
  f"Translate the following text to {target}. "
32
  "Output ONLY the translation, no explanations:\n\n"