Update app.py

app.py

@@ -41,8 +41,8 @@ system_prompt = SystemMessage(content=system_prompt)
 # --- Basic Agent Definition ---
 # ----- THIS IS WHERE YOU CAN BUILD WHAT YOU WANT ------
 
-class BasicAgent:
-    """A LangGraph agent"""
+#class BasicAgent:
+"""A LangGraph agent
     def __init__(self):
         print("BasicAgent initialized.")
         #self.graph = build_graph("huggingface")
@@ -58,8 +58,9 @@ class BasicAgent:
         #messages = self.graph.invoke({"messages": messages})
         #answer = messages['messages'][-1].content
         #return answer[14:]
+"""
 
-class AssignmentAgent:
+class BasicAgent:
     def __init__(self, system_prompt):
         #if "ANTHROPIC_API_KEY" not in os.environ:
         #  raise Exception("No Anthropic API key in environment")