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])
|