Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,6 @@ llm = HuggingFaceEndpoint(
|
|
23 |
)
|
24 |
chat_model = ChatHuggingFace(llm=llm)
|
25 |
|
26 |
-
global messages
|
27 |
messages = [
|
28 |
SystemMessage(content="You are a helpful assistant."),
|
29 |
HumanMessage(content="Hi AI, how are you today?"),
|
@@ -39,6 +38,7 @@ def handle_query(query: str, mode: str):
|
|
39 |
return "Invalid mode selected."
|
40 |
|
41 |
def chat_mode(query: str):
|
|
|
42 |
prompt = HumanMessage(content=query)
|
43 |
messages.append(prompt)
|
44 |
response = chat_model.invoke(messages)
|
@@ -48,6 +48,7 @@ def chat_mode(query: str):
|
|
48 |
return f"You: {query}\nIT-Assistant: {response.content}"
|
49 |
|
50 |
def web_search(query: str):
|
|
|
51 |
similar_docs = db.similarity_search(query, k=3)
|
52 |
if similar_docs:
|
53 |
source_knowledge = "\n".join([x.page_content for x in similar_docs])
|
|
|
23 |
)
|
24 |
chat_model = ChatHuggingFace(llm=llm)
|
25 |
|
|
|
26 |
messages = [
|
27 |
SystemMessage(content="You are a helpful assistant."),
|
28 |
HumanMessage(content="Hi AI, how are you today?"),
|
|
|
38 |
return "Invalid mode selected."
|
39 |
|
40 |
def chat_mode(query: str):
|
41 |
+
global messages
|
42 |
prompt = HumanMessage(content=query)
|
43 |
messages.append(prompt)
|
44 |
response = chat_model.invoke(messages)
|
|
|
48 |
return f"You: {query}\nIT-Assistant: {response.content}"
|
49 |
|
50 |
def web_search(query: str):
|
51 |
+
global messages
|
52 |
similar_docs = db.similarity_search(query, k=3)
|
53 |
if similar_docs:
|
54 |
source_knowledge = "\n".join([x.page_content for x in similar_docs])
|