Modified files

agent.py

@@ -22,10 +22,10 @@ def initialize_agent():
     load_dotenv()
 
     # 1. Load the model
-    # Make sure to set HF_TOKEN in your Hugging Face Space secrets
+    # Make sure to set TOGETHER_API_KEY in your environment variables
     model_name = "mistralai/Mixtral-8x7B-Instruct-v0.1"
     try:
-        model = InferenceClientModel(model_id=model_name, token=os.environ.get("HF_TOKEN"), provider="together")
+        model = InferenceClientModel(model_id=model_name, token=os.environ.get("TOGETHER_API_KEY"), provider="together")
     except Exception as e:
         print(f"Error loading model: {e}")
         model = None
@@ -161,7 +161,7 @@ def initialize_agent():
             tools=[],
             model=model,
             managed_agents=[web_agent],
-            additional_authorized_imports=["time", "numpy", "pandas", "requests", "serpapi", "llama_index", "beautifulsoup4", "markdownify", "lxml", "json", "urllib.parse", "youtube_transcript_api"],
+            additional_authorized_imports=["time", "numpy", "pandas", "requests", "serpapi", "llama_index", "beautifulsoup4", "markdownify", "lxml", "json", "urllib.parse", "youtube_transcript_api", "together"],
             instructions='''You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the a new line and 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 above rules depending of whether the element to be put in the list is a number or a string.
 
 To achieve the best results, follow these steps:
@@ -174,4 +174,4 @@ To achieve the best results, follow these steps:
         )
         return manager_agent
     else:
-        return None
+        return None

requirements.txt

@@ -8,4 +8,5 @@ duckduckgo-search
 wikipedia
 serpapi
 llama-index
-youtube-transcript-api
+youtube-transcript-api
+together