Update agents/agents_nodes.py

agents/agents_nodes.py

@@ -12,16 +12,24 @@ from tools.financial_tools import time_value_tool
 
 # LLL instantation
 
-text_generator = pipeline(
-    "text-generation",  # Task type
-    # model="deepseek-ai/DeepSeek-R1-0528-Qwen3-8B",
-    # model="google/gemma-3n-E2B-it",
-    # model="google/gemma-3n-e4b-it",
-    model="Qwen/Qwen3-Embedding-0.6B",
-    # device="cuda" if torch.cuda.is_available() else "cpu",
-    device= "cpu",
-    torch_dtype=torch.bfloat16,
-    max_new_tokens=700  # Limit output length
+# text_generator = pipeline(
+#     "text-generation",  # Task type
+#     # model="deepseek-ai/DeepSeek-R1-0528-Qwen3-8B",
+#     # model="google/gemma-3n-E2B-it",
+#     # model="google/gemma-3n-e4b-it",
+#     model="Qwen/Qwen3-Embedding-0.6B",
+#     # device="cuda" if torch.cuda.is_available() else "cpu",
+#     device= "cpu",
+#     torch_dtype=torch.bfloat16,
+#     max_new_tokens=700  # Limit output length
+# )
+
+
+text_generator = HuggingFaceEndpoint(
+    repo_id="Qwen/Qwen3-Embedding-0.6B",
+    task="text-generation",
+    max_new_tokens=512,
+    do_sample=False,
 )