Update agent.py
Browse files
agent.py
CHANGED
@@ -668,18 +668,19 @@ tools = [
|
|
668 |
wiki_search,
|
669 |
web_search,
|
670 |
arvix_search,
|
671 |
-
#save_and_read_file,
|
672 |
-
#download_file_from_url,
|
673 |
-
#extract_text_from_image,
|
674 |
-
#analyze_csv_file,
|
675 |
-
#analyze_excel_file,
|
676 |
-
#execute_code_multilang,
|
677 |
-
#analyze_image,
|
678 |
-
#transform_image,
|
679 |
-
#draw_on_image,
|
680 |
-
#generate_simple_image,
|
681 |
-
#combine_images,
|
682 |
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
683 |
|
684 |
# Build graph function
|
685 |
def build_graph(provider: str = "huggingface"):
|
@@ -754,8 +755,8 @@ def build_graph(provider: str = "huggingface"):
|
|
754 |
"assistant",
|
755 |
tools_condition,
|
756 |
)
|
757 |
-
builder.add_edge("tools", "retriever")
|
758 |
-
|
759 |
|
760 |
# Compile graph
|
761 |
return builder.compile()
|
|
|
668 |
wiki_search,
|
669 |
web_search,
|
670 |
arvix_search,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
671 |
]
|
672 |
+
#save_and_read_file,
|
673 |
+
#download_file_from_url,
|
674 |
+
#extract_text_from_image,
|
675 |
+
#analyze_csv_file,
|
676 |
+
#analyze_excel_file,
|
677 |
+
#execute_code_multilang,
|
678 |
+
#analyze_image,
|
679 |
+
#transform_image,
|
680 |
+
#draw_on_image,
|
681 |
+
#generate_simple_image,
|
682 |
+
#combine_images,
|
683 |
+
|
684 |
|
685 |
# Build graph function
|
686 |
def build_graph(provider: str = "huggingface"):
|
|
|
755 |
"assistant",
|
756 |
tools_condition,
|
757 |
)
|
758 |
+
#builder.add_edge("tools", "retriever")
|
759 |
+
builder.add_edge("tools", "assistant")
|
760 |
|
761 |
# Compile graph
|
762 |
return builder.compile()
|