Update agent_v2.py

agent_v2.py

@@ -707,14 +707,14 @@ tools = [
 
 
 # Build graph function
-def build_graph(provider: str = "groq"):
+def build_graph(provider: str = "huggingface"):
     """Build the graph"""
     # Load environment variables from .env file
     if provider == "groq":
         # Groq https://console.groq.com/docs/models
         llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
     elif provider == "openai":
-        model_openai = "gpt4o"
+        model_openai = "gpt-4o"
         llm = ChatOpenAI(
             model=model_openai,
             temperature=0,
@@ -762,8 +762,9 @@ def build_graph(provider: str = "groq"):
     #builder.add_node("retriever", retriever)
     builder.add_node("assistant", assistant)
     builder.add_node("tools", ToolNode(tools))
-    builder.add_edge(START, "retriever")
-    builder.add_edge("retriever", "assistant")
+    #builder.add_edge(START, "retriever")
+    builder.add_edge(START, "assistant")
+    #builder.add_edge("retriever", "assistant")
     builder.add_conditional_edges(
         "assistant",
         tools_condition,