code interpreter + post process answer

app.py

@@ -9,6 +9,7 @@ from llama_index.core.agent.workflow import ReActAgent, AgentStream, ToolCallRes
 from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI # customized to support different provider
 from llama_index.tools.wikipedia import WikipediaToolSpec
 from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
+from llama_index.tools.code_interpreter import CodeInterpreterToolSpec
 
 from prompt import custom_react_system_header_str
 from config import HF_MODEL_NAME, HF_PROVIDER
@@ -44,6 +45,7 @@ class LLamaIndexAgent:
         tool_spec_list = []
         # tool_spec_list += WikipediaToolSpec().to_tool_list()
         tool_spec_list += DuckDuckGoSearchToolSpec().to_tool_list()
+        tool_spec_list += CodeInterpreterToolSpec().to_tool_list()
 
         # agent definition
         self.agent = ReActAgent(llm=llm, tools=tool_spec_list)
@@ -67,7 +69,9 @@ class LLamaIndexAgent:
                 print(v.template)
 
     async def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        print("\n\n"+"*"*50)
+        print(f"Agent received question: {question}")
+        print("*"*50)
 
         handler = self.agent.run(question, ctx=self.ctx)
         async for ev in handler.stream_events():
@@ -77,6 +81,13 @@ class LLamaIndexAgent:
                 print(f"{ev.delta}", end="", flush=True)
 
         response = await handler
+
+        # post-process the response (cast AgentOutput to str and keep only what's after "FINAL ANSWER:" for the exact match)
+        response = str(response)
+        try:
+            response = response.split("FINAL ANSWER:")[-1].strip()
+        except:
+            print('Could not split response on "FINAL ANSWER:"')
         print("\n\n"+"-"*50)
         print(f"Agent returning with answer: {response}")
         return response
@@ -144,9 +155,7 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            # submitted_answer = agent(question_text)
             submitted_answer = await agent(question_text)
-            submitted_answer = str(submitted_answer) # cast AgentOutput to str
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
             agent.ctx.clear() # clear context for next question
@@ -240,6 +249,13 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     print("\n" + "-"*30 + " App Starting " + "-"*30)
+
+    # set hard-coded values to run locally
+    if not os.getenv("SPACE_HOST"):
+        os.environ["SPACE_HOST"] = "guillaumefrd-agents-final-assignment.hf.space"
+    if not os.getenv("SPACE_ID"):
+        os.environ["SPACE_ID"] = "guillaumefrd/agents_final_assignment"
+
     # Check for SPACE_HOST and SPACE_ID at startup for information
     space_host_startup = os.getenv("SPACE_HOST")
     space_id_startup = os.getenv("SPACE_ID") # Get SPACE_ID at startup

config.py

@@ -1,2 +1,2 @@
-HF_MODEL_NAME = "google/gemma-3-27b-it"
+HF_MODEL_NAME = "google/gemma-3-27b-it" # multimodal (need to handle image input)
 HF_PROVIDER = "nebius"

prompt.py

@@ -3,7 +3,7 @@
 custom_react_system_header_str = """\
 
 You are a general AI assistant.
-I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
+A human will ask you a question. Report your Thoughts, Actions, Observations as described in ## Output Format, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER HERE]
 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.
@@ -44,12 +44,12 @@ You should keep repeating the above format till you have enough information to a
 
 ```
 Thought: I can answer without using any more tools. I'll use the user's language to answer
-Answer: [your answer here (In the same language as the user's question)]
+FINAL ANSWER: [YOUR FINAL ANSWER HERE (In the same language as the user's question)]
 ```
 
 ```
 Thought: I cannot answer the question with the provided tools.
-Answer: [your answer here (In the same language as the user's question)]
+FINAL ANSWER: [YOUR FINAL ANSWER HERE (In the same language as the user's question)]
 ```
 
 ## Current Conversation

requirements.txt

@@ -3,4 +3,5 @@ requests
 llama-index
 llama-index-llms-huggingface-api @ git+https://github.com/guillaumefrd/llama_index.git@add-provider-HF-API#subdirectory=llama-index-integrations/llms/llama-index-llms-huggingface-api
 llama_index.tools.wikipedia
-llama_index.tools.duckduckgo
+llama_index.tools.duckduckgo
+llama_index.tools.code_interpreter