LLM에게 정보를 강제로 주입하면 어떻게 되는지 봅니다. 문서가 짧을 때는 가장 확실하고 쉬운 방법입니다.
step2_context.pyfrom langchain_ollama import ChatOllama
llm = ChatOllama(model="deepseek-r1:8b", temperature=0)
# 1. 정보를 변수에 담습니다 (아직 DB 안 씀)
context_data = """
[테크컴퍼니 취업규칙]
1. 신입사원은 입사 후 3년 동안은 연차가 없다. (파격적인 규정)
2. 대신 매월 1회 '리프레시 데이'를 유급으로 제공한다.
3. 3년 근속 시 30일의 연차가 일시에 발생한다.
"""
question = "우리 회사(테크컴퍼니)의 신입사원 연차 발생 규정이 어떻게 돼?"
# 2. 프롬프트에 정보를 포함시킵니다.
prompt = f"""
아래 [참고 정보]를 보고 질문에 답해줘.
[참고 정보]
{context_data}
질문: {question}
"""
print(f"질문: {question}\\\\n")
response = llm.invoke(prompt)
print(f"답변:\\\\n{response.content}")
질문: 우리 회사(테크컴퍼니)의 신입사원 연차 발생 규정이 어떻게 돼?
답변:
[테크컴퍼니 취업규칙]에 따르면, **신입사원의 연차 발생 규정**은 다음과 같습니다:
1. **신입사원은 입사 후 3년 동안 연차가 없다.**
- 즉, 첫 3년간 쌓이는 연차는 없습니다.
2. **대신 매월 1회 '리프레시 데이'를 유급으로 제공한다.**
- 매월 1일 제공되는 유급 휴무로, 연차 대신 사용됩니다.
3. **3년 근속 시 30일의 연차가 일시에 발생한다.**
- 3년 차 이후부터 30일의 연차가 일시에 부여됩니다.
이러한 규정은 **신입사원에게 연차가 부족하지만, 대체 휴무와 3년 만에 일시적 연차**를 제공하는 파격적인 정책입니다.
이번 단계에서 AI가 거짓말을 하지 않고 정확히 답변한 이유입니다.