assdf

agent.py

@@ -1,5 +1,6 @@
 """LangGraph Agent"""
 import os
+import pandas as pd
 from dotenv import load_dotenv
 from langgraph.graph import START, StateGraph, MessagesState
 from langgraph.prebuilt import tools_condition
@@ -16,7 +17,7 @@ from langchain_core.tools import tool
 from langchain.tools.retriever import create_retriever_tool
 from supabase.client import Client, create_client
 from pydantic import BaseModel, Field
-import pandas as pd
+
 
 from typing import List, Set, Any
 
@@ -335,12 +336,12 @@ def build_graph(provider: str = "huggingface"):
         # Groq https://console.groq.com/docs/models
         llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
     elif provider == "huggingface":
-        repo_id = "togethercomputer/evo-1-131k-base"
-        repo_id="HuggingFaceH4/zephyr-7b-beta",
+        # repo_id = "togethercomputer/evo-1-131k-base"
+        # repo_id="HuggingFaceH4/zephyr-7b-beta",
         if not hf_token:
             raise ValueError("HF_TOKEN environment variable not set. It's required for Hugging Face provider.")
         llm = HuggingFaceEndpoint(
-            repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+            repo_id="togethercomputer/LLaMA-2-7B-32K",
             provider="auto",
             task="text-generation",
             max_new_tokens=1000,