IamRulo commited on
Commit
6fddcbb
·
verified ·
1 Parent(s): f2ed73d

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +14 -13
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
- #builder.add_edge("tools", "assistant")
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()