mou3az commited on
Commit
6fea818
·
verified ·
1 Parent(s): 52a6248

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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])