Luong Huu Thanh commited on
Commit
9ff46a6
·
1 Parent(s): 88bc423

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +19 -28
agent.py CHANGED
@@ -692,7 +692,7 @@ embeddings = HuggingFaceEmbeddings(
692
  model_name="sentence-transformers/all-mpnet-base-v2"
693
  ) # dim=768
694
  supabase: Client = create_client(
695
- os.environ.get("SUPABASE_URL"), os.environ.get("SUPABASE_SERVICE_KEY")
696
  )
697
  vector_store = SupabaseVectorStore(
698
  client=supabase,
@@ -707,40 +707,31 @@ create_retriever_tool = create_retriever_tool(
707
  )
708
 
709
 
710
- # tools = [
711
- # web_search,
712
- # wiki_search,
713
- # arxiv_search,
714
- # multiply,
715
- # add,
716
- # subtract,
717
- # divide,
718
- # modulus,
719
- # power,
720
- # square_root,
721
- # save_and_read_file,
722
- # download_file_from_url,
723
- # extract_text_from_image,
724
- # analyze_csv_file,
725
- # analyze_excel_file,
726
- # execute_code_multilang,
727
- # analyze_image,
728
- # transform_image,
729
- # draw_on_image,
730
- # generate_simple_image,
731
- # combine_images,
732
- # ]
733
  tools = [
 
 
 
734
  multiply,
735
  add,
736
  subtract,
737
  divide,
738
  modulus,
739
- wiki_search,
740
- web_search,
741
- arxiv_search,
 
 
 
 
 
 
 
 
 
 
742
  ]
743
 
 
744
  # Build graph function
745
  def build_graph(provider: str = "groq"):
746
  """Build the graph"""
@@ -802,7 +793,7 @@ def build_graph(provider: str = "groq"):
802
 
803
  # test
804
  if __name__ == "__main__":
805
- question = ".rewsna eht sa ""tfel"" drow eht fo etisoppo eht etirw ,ecnetnes siht dnatsrednu uoy fI"
806
  graph = build_graph(provider="groq")
807
  messages = [HumanMessage(content=question)]
808
  messages = graph.invoke({"messages": messages})
 
692
  model_name="sentence-transformers/all-mpnet-base-v2"
693
  ) # dim=768
694
  supabase: Client = create_client(
695
+ os.environ.get("SUPABASE_URL"), os.environ.get("SUPABASE_SERVICE_ROLE_KEY")
696
  )
697
  vector_store = SupabaseVectorStore(
698
  client=supabase,
 
707
  )
708
 
709
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
710
  tools = [
711
+ web_search,
712
+ wiki_search,
713
+ arxiv_search,
714
  multiply,
715
  add,
716
  subtract,
717
  divide,
718
  modulus,
719
+ power,
720
+ square_root,
721
+ save_and_read_file,
722
+ download_file_from_url,
723
+ extract_text_from_image,
724
+ analyze_csv_file,
725
+ analyze_excel_file,
726
+ execute_code_multilang,
727
+ analyze_image,
728
+ transform_image,
729
+ draw_on_image,
730
+ generate_simple_image,
731
+ combine_images,
732
  ]
733
 
734
+
735
  # Build graph function
736
  def build_graph(provider: str = "groq"):
737
  """Build the graph"""
 
793
 
794
  # test
795
  if __name__ == "__main__":
796
+ question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"
797
  graph = build_graph(provider="groq")
798
  messages = [HumanMessage(content=question)]
799
  messages = graph.invoke({"messages": messages})