AashitaK commited on
Commit
4ee3f67
·
verified ·
1 Parent(s): d7446f7

Update utils/chatbot.py

Browse files
Files changed (1) hide show
  1. utils/chatbot.py +3 -0
utils/chatbot.py CHANGED
@@ -52,6 +52,8 @@ def answer_query_with_context(
52
  query: str,
53
  df: pd.DataFrame,
54
  document_embeddings: dict[(str, str), np.array],
 
 
55
  ) -> str:
56
  """
57
  Generates a response to a user's query using the most relevant service description.
@@ -79,5 +81,6 @@ def answer_query_with_context(
79
  ]
80
 
81
  response = get_response(messages)
 
82
 
83
  return response + end_message
 
52
  query: str,
53
  df: pd.DataFrame,
54
  document_embeddings: dict[(str, str), np.array],
55
+ META_PROMPT=META_PROMPT,
56
+ end_message=end_message
57
  ) -> str:
58
  """
59
  Generates a response to a user's query using the most relevant service description.
 
81
  ]
82
 
83
  response = get_response(messages)
84
+ print(end_message)
85
 
86
  return response + end_message