add youtube transcription

agents/search_agent.py

@@ -7,14 +7,14 @@ langfuse_handler = CallbackHandler(host="https://cloud.langfuse.com")
 
 class SearchAgent:
     def __init__(self):
-        print("SearchAgent initialized.")
+        """
+        Initialize the SearchAgent with the workflow.
+        """
+        self.workflow = build_workflow()
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        workflow = build_workflow()
-        
         messages =  [HumanMessage(content=question)]
 
-        state = workflow.invoke({
+        state = self.workflow.invoke({
             "messages":messages,
             "question": question,
         }, config={"callbacks": [langfuse_handler]})

graphs/evaluation.py

@@ -32,7 +32,6 @@ def call_node(state: EvaluationState):
     """
     This node call the model with the question and the tools
     """
-    print(state["question"])
     response = bound_model_llama.invoke(state["messages"])
     
     state["messages"].append(response)
@@ -66,7 +65,7 @@ def map_answer(state: EvaluationState):
     """
     Map the answer to the final answer
     """
-    answer = anthropic_model.invoke("Map the answer, I want only the number, string or list. ANSWER:"+ state["answer"])
+    answer = anthropic_model.invoke("Map the answer, I want only the number, string or list. Remove quotes. ANSWER:"+ state["answer"])
     
     return {
         "answer": answer.content
@@ -85,7 +84,6 @@ def validator(state: EvaluationState):
     
     # Check if we've hit the validator recursion limit
     if state["step_counter"]["validator"] >= 3:  # Smaller limit for validator recursion
-        print("Validator recursion limit reached. Accepting current answer format.")
         state["is_valid_answer"] = True
         return state