fix sys prompt

change default agent to huggingface

agent.py

@@ -127,10 +127,9 @@ def arvix_search(query: str) -> str:
     return {'arvix_results': formatted_search_docs}
 
 
-system_prompt = """
-
-"""
-
+# load the system prompt from the file
+with open("system_prompt.txt", "r", encoding="utf-8") as f:
+    system_prompt = f.read()
 
 
 # System message
@@ -172,7 +171,7 @@ tools = [
 ]
 
 # build graph function
-def build_graph(tag: str='google'):
+def build_graph(tag: str='huggingface'):
     """Build the graph"""
 
     if tag == 'local':
@@ -180,14 +179,14 @@ def build_graph(tag: str='google'):
     elif tag == 'google':
         # Google Gemini
         llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
-    else:
-        url='https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf'
-        token = os.getenv('HF_TOKEN')
-        end_point = HuggingFaceEndpoint(
-            endpoint_url=url,
+    elif tag == "huggingface":
+        llm = ChatHuggingFace(
+            llm=HuggingFaceEndpoint(
+                endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf"),
             temperature=0,
-            huggingfacehub_api_token=token)
-        llm = ChatHuggingFace(llm=end_point)
+        )
+    else:
+        raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
     # bind tools to llm
     llm_with_tools = llm.bind_tools(tools)
 
@@ -203,15 +202,19 @@ def build_graph(tag: str='google'):
 
     builder = StateGraph(MessagesState)
     builder.add_node('retriever', retriever)
-    builder.add_node('assistant', assistant)
-    builder.add_node('tools', ToolNode(tools))
-    builder.add_edge(START, 'retriever')
-    builder.add_edge('retriever', 'assistant')
-    builder.add_conditional_edges(
-        'assistant',
-        tools_condition
-    )
-    builder.add_edge('tools', 'assistant')
+    # builder.add_node('assistant', assistant)
+    # builder.add_node('tools', ToolNode(tools))
+    # builder.add_edge(START, 'retriever')
+    # builder.add_edge('retriever', 'assistant')
+    # builder.add_conditional_edges(
+    #     'assistant',
+    #     tools_condition
+    # )
+    # builder.add_edge('tools', 'assistant')
+
+    builder.set_entry_point("retriever")
+    builder.set_finish_point("retriever")
+
     return builder.compile()
 
 

requirements.txt

@@ -10,4 +10,4 @@ faiss-cpu
 langchain-ollama
 langgraph
 wikipedia
-python-dotenv
+python-dotenv

system_prompt.txt

@@ -0,0 +1,5 @@
+You are a helpful assistant tasked with answering questions using a set of tools.
+Now, I will ask you a question. Report your thoughts, and finish your answer with the following template:
+FINAL ANSWER: [YOUR FINAL ANSWER].
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, Apply the rules above for each element (number or string), ensure there is exactly one space after each comma.
+Your answer should only start with "FINAL ANSWER: ", then follows with the answer.