name observation process
Browse files
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)
|