Spaces:
Sleeping
Sleeping
helloguddn (PC helloguddn) commited on
Commit ·
9b49a5b
1
Parent(s): f4bf832
fix: 모델 수정
Browse files- agents/main_agent.py +1 -1
- data/labor_law_chunks.json +1 -1
- services/gemini_service.py +1 -1
- tools/contract_scanner.py +2 -2
- tools/translation_tool.py +1 -1
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-
|
| 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": "
|
| 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-
|
| 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. 임금 -
|
| 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-
|
| 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-
|
| 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"
|