Upload agent.py

agent.py

@@ -209,18 +209,18 @@ search_tool = DuckDuckGoSearchResults()
 #     temperature=0.2,
 #     max_tokens=4096,
 # )
-# llm = ChatHuggingFace(
-#     llm=HuggingFaceEndpoint(
-#         # repo_id="microsoft/Phi-3-mini-4k-instruct",
-#         repo_id="meta-llama/Llama-3-70B-Instruct",
-#         temperature=0,
-#         # huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API_TOKEN"],
-#     ),
-#     verbose=True,
-# )
-llm = ChatGoogleGenerativeAI(
-    model="gemini-2.0-flash-exp", google_api_key=os.environ["GOOGLE_API_KEY"]
+llm = ChatHuggingFace(
+    llm=HuggingFaceEndpoint(
+        repo_id="Qwen/Qwen3-30B-A3B",
+        # repo_id="meta-llama/Llama-3-70B-Instruct",
+        temperature=0,
+        huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API_TOKEN"],
+    ),
+    verbose=True,
 )
+# llm = ChatGoogleGenerativeAI(
+#     model="gemini-2.0-flash-exp", google_api_key=os.environ["GOOGLE_API_KEY"]
+# )
 tools = [
     analyze_csv_file,
     analyze_excel_file,
@@ -274,10 +274,16 @@ def build_agent_workflow():
                     """
 
         if question.startswith(".") or ".rewsna eht sa" in question:
+            print("Reversed text detected.")
+            print(state.get("messages")[::-1])
+
             context = f"""
-            This question appears to be in reversed text. Here's the reversed version:
-            {state['message'][::-1]}
-            Now answer the question above. Remember to format your answer exactly as requested.
+            This question appears to be in reversed text. your task to reverse the sentence. Here's the reversed example:
+            .rewsna eht sa "tfel" drow eht fo etisoppo eht etirw ,ecnetnes siht dnatsrednu uoy fI
+            and the answer is:
+            "If you understand this sentence, write the opposite of the word "left" as the answer."
+       
+            Now rewrite in to proper formate the {question}. Remember to format your answer exactly as requested.
             """
         system_prompt = SystemMessage(
             f"""{context}
@@ -304,12 +310,12 @@ def build_agent_workflow():
     return app
 
 
-# if __name__ == "__main__":
-#     question = "Who nominated the only Featured Article on English Wikipedia about a dinosaur that was promoted in November 2016?"
-#     # Build the graph
-#     graph = build_agent_workflow()
-#     # Run the graph
-#     messages = [HumanMessage(content=question)]
-#     messages = graph.invoke({"messages": messages, "input_file": None})
-#     for m in messages["messages"]:
-#         m.pretty_print()
+if __name__ == "__main__":
+    question = '.rewsna eht sa "tfel" drow eht fo etisoppo eht etirw ,ecnetnes siht dnatsrednu uoy fI'
+    # Build the graph
+    graph = build_agent_workflow()
+    # Run the graph
+    messages = [HumanMessage(content=question)]
+    messages = graph.invoke({"messages": messages, "input_file": None})
+    for m in messages["messages"]:
+        m.pretty_print()