Spaces:
Paused
Paused
Update llm_client.py
Browse files- 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
|
| 51 |
-
|
| 52 |
-
If the input is in Vietnamese, respond in Vietnamese.
|
| 53 |
|
| 54 |
Context:
|
| 55 |
{context}
|
| 56 |
|
| 57 |
-
Input
|
| 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,
|