Update tools.py

tools.py

@@ -133,24 +133,14 @@ def build_graph(provider: str = "groq", csv_file_path: str = "embeddings.csv"):
     # System message
     sys_msg = SystemMessage(content=system_prompt)
     
-    # Initialize LLM based on provider
-    if provider == "google":
-        llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
-
-    elif provider == "huggingface":
-        llm = ChatHuggingFace(
-            llm=HuggingFaceEndpoint(
-                url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
-                temperature=0,
-            ),
-        )
-
-    elif provider == "groq":
-        from langchain_groq import ChatGroq
-        llm = ChatGroq(model="mixtral-8x7b-32768", temperature=0)
-
-    else:
-        raise ValueError("Invalid provider. Choose 'google', 'huggingface', or 'groq'.")
+    from langchain_community.llms import HuggingFaceEndpoint
+
+    llm = HuggingFaceEndpoint(
+        endpoint_url="https://api.endpoints.huggingface.co/v1/completions",
+        huggingfacehub_api_token="your_huggingface_token",
+        model_kwargs={"max_tokens": 512}
+    )
+
 
     # Bind tools to LLM
     llm_with_tools = llm.bind_tools(tools)