Update agent.py

agent.py

@@ -1059,29 +1059,33 @@ model_config = {
     "huggingfacehub_api_token": os.getenv("HF_TOKEN")
 }
 
-    # Get LLM
-    def get_llm(provider: str, config: dict):
-        if provider == "huggingface":
-            from langchain_huggingface import HuggingFaceEndpoint
-            return HuggingFaceEndpoint(
-                repo_id=config["repo_id"],
-                task=config["task"],
-                huggingfacehub_api_token=config["huggingfacehub_api_token"],
-                temperature=config["temperature"],
-                max_new_tokens=config["max_new_tokens"]
-            )
-        else:
-            raise ValueError(f"Unsupported provider: {provider}")
+# Get LLM
+def get_llm(provider: str, config: dict):
+    if provider == "huggingface":
+        from langchain_huggingface import HuggingFaceEndpoint
+        return HuggingFaceEndpoint(
+            repo_id=config["repo_id"],
+            task=config["task"],
+            huggingfacehub_api_token=config["huggingfacehub_api_token"],
+            temperature=config["temperature"],
+            max_new_tokens=config["max_new_tokens"]
+        )
+    else:
+        raise ValueError(f"Unsupported provider: {provider}")
 
 
-    def assistant(state: dict):
-        return {"messages": [llm_with_tools.invoke(state["messages"])]}
+def assistant(state: dict):
+    return {
+        "messages": [llm_with_tools.invoke(state["messages"])]
+    }
+
+
+def tools_condition(state: dict) -> str:
+    if "use tool" in state["messages"][-1].content.lower():
+        return "tools"
+    else:
+        return "END"
 
-    def tools_condition(state: dict) -> str:
-        if "use tool" in state["messages"][-1].content.lower():
-            return "tools"
-        else:
-            return "END" 
             
 
 from langgraph.graph import StateGraph