Ing commited on
Commit
7b5c91e
·
1 Parent(s): d905cf7

name observation process

Browse files
Files changed (1) hide show
  1. chat_3.py +3 -4
chat_3.py CHANGED
@@ -163,7 +163,7 @@ class Chat:
163
  return detect_language(user_input)
164
 
165
  # If the main chat method itself should be a trace, uncomment @observe() below
166
- @observe("MainChatFlow")
167
  def chat(self, user_input: str) -> str:
168
  # print(f"\n\n-- USER INPUT: {user_input} --")
169
  try:
@@ -209,8 +209,7 @@ class Chat:
209
  return ai_response_content
210
 
211
 
212
-
213
- @observe("Rag-Flow")
214
  def call_rag_v2(self, user_input: str, input_lang: str, history_for_rewrite: list) -> str:
215
  try:
216
  retriever, selected_rag_prompt = self.get_retriever_and_prompt(input_lang)
@@ -251,7 +250,7 @@ class Chat:
251
  print(f"Error during RAG LLM call: {e}")
252
  return "Sorry, I encountered an error while generating the response."
253
 
254
- @observe("RewriteQuery")
255
  def _rewrite_query_if_needed_v2(self, user_input: str, history_list: list) -> str:
256
  if not history_list:
257
  # self.langfuse_handler.trace(name="rewrite_query_skipped_no_history", input={"user_input": user_input}, output=user_input)
 
163
  return detect_language(user_input)
164
 
165
  # If the main chat method itself should be a trace, uncomment @observe() below
166
+ @observe(name="MainChatFlow")
167
  def chat(self, user_input: str) -> str:
168
  # print(f"\n\n-- USER INPUT: {user_input} --")
169
  try:
 
209
  return ai_response_content
210
 
211
 
212
+ @observe(name="Rag-Flow")
 
213
  def call_rag_v2(self, user_input: str, input_lang: str, history_for_rewrite: list) -> str:
214
  try:
215
  retriever, selected_rag_prompt = self.get_retriever_and_prompt(input_lang)
 
250
  print(f"Error during RAG LLM call: {e}")
251
  return "Sorry, I encountered an error while generating the response."
252
 
253
+ @observe(name="RewriteQuery")
254
  def _rewrite_query_if_needed_v2(self, user_input: str, history_list: list) -> str:
255
  if not history_list:
256
  # self.langfuse_handler.trace(name="rewrite_query_skipped_no_history", input={"user_input": user_input}, output=user_input)