dungeon29 commited on
Commit
c7555aa
·
verified ·
1 Parent(s): 47c8ad8

Update llm_client.py

Browse files
Files changed (1) hide show
  1. llm_client.py +4 -5
llm_client.py CHANGED
@@ -47,17 +47,16 @@ class LLMClient:
47
  return "❌ Vector Store not initialized."
48
 
49
  # Custom Prompt Template
50
- template = """You are a cybersecurity expert specializing in phishing detection.
51
- Use the following pieces of context to analyze the input.
52
- If the input is in Vietnamese, respond in Vietnamese.
53
 
54
  Context:
55
  {context}
56
 
57
- Input to Analyze:
58
  {question}
59
 
60
- Analysis:"""
61
 
62
  PROMPT = PromptTemplate(
63
  template=template,
 
47
  return "❌ Vector Store not initialized."
48
 
49
  # Custom Prompt Template
50
+ template = """You are a cybersecurity expert. Please provide a brief and concise analysis of the input below based on the Context.
51
+ Answer base on input language. Focus on: Is it phishing? Why?
 
52
 
53
  Context:
54
  {context}
55
 
56
+ Input:
57
  {question}
58
 
59
+ Short Analysis:"""
60
 
61
  PROMPT = PromptTemplate(
62
  template=template,