Update app.py

app.py

@@ -1,7 +1,10 @@
 import argparse
 import json
 import os
+import re
 import threading
+import mimetypes
+import shutil
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from datetime import datetime
 from pathlib import Path
@@ -43,7 +46,6 @@ from smolagents import (
 from smolagents.agent_types import AgentText, AgentImage, AgentAudio
 from smolagents.gradio_ui import pull_messages_from_step, handle_agent_output_types
 
-
 AUTHORIZED_IMPORTS = [
     "requests",
     "zipfile",
@@ -86,13 +88,11 @@ _ = """
 eval_ds = datasets.load_dataset("gaia-benchmark/GAIA", "2023_all")[SET]
 eval_ds = eval_ds.rename_columns({"Question": "question", "Final answer": "true_answer", "Level": "task"})
 
-
 def preprocess_file_paths(row):
     if len(row["file_name"]) > 0:
         row["file_name"] = f"data/gaia/{SET}/" + row["file_name"]
     return row
 
-
 eval_ds = eval_ds.map(preprocess_file_paths)
 eval_df = pd.DataFrame(eval_ds)
 print("Loaded evaluation dataset:")
@@ -113,13 +113,19 @@ BROWSER_CONFIG = {
 
 os.makedirs(f"./{BROWSER_CONFIG['downloads_folder']}", exist_ok=True)
 
+# --- 수정된 부분: OPENAI_API_BASE의 잘못된 엔드포인트 제거 ---
+openai_api_base = os.getenv("OPENAI_API_BASE", "https://api.openai.com/v1")
+# 만약 환경변수에 불필요한 엔드포인트가 포함되어 있다면 제거합니다.
+if openai_api_base.endswith("/chat/completions"):
+    openai_api_base = openai_api_base.rsplit("/chat/completions", 1)[0]
+
 model = LiteLLMModel(
-    # "gpt-4o",
     os.getenv("MODEL_ID", "gpt-4o-mini"),
     custom_role_conversions=custom_role_conversions,
-    api_base=os.getenv("OPENAI_API_BASE"),
+    api_base=openai_api_base,
     api_key=os.getenv("OPENAI_API_KEY"),
 )
+# --- 수정 끝 ---
 
 text_limit = 20000
 ti_tool = TextInspectorTool(model, text_limit)
@@ -149,23 +155,6 @@ agent = CodeAgent(
 document_inspection_tool = TextInspectorTool(model, 20000)
 
 
-# augmented_question = """You have one question to answer. It is paramount that you provide a correct answer.
-# Give it all you can: I know for a fact that you have access to all the relevant tools to solve it and find the correct answer (the answer does exist). Failure or 'I cannot answer' or 'None found' will not be tolerated, success will be rewarded.
-# Run verification steps if that's needed, you must make sure you find the correct answer!
-# Here is the task:
-# """ + example["question"]
-
-# if example["file_name"]:
-#     prompt_use_files = "\n\nTo solve the task above, you will have to use this attached file:"
-#     prompt_use_files += get_single_file_description(
-#         example["file_name"], example["question"], visual_inspection_tool, document_inspection_tool
-#     )
-#     augmented_question += prompt_use_files
-
-
-# final_result = agent.run(augmented_question)
-
-
 def stream_to_gradio(
     agent,
     task: str,
@@ -174,9 +163,7 @@ def stream_to_gradio(
 ):
     """Runs an agent with the given task and streams the messages from the agent as gradio ChatMessages."""
     for step_log in agent.run(task, stream=True, reset=reset_agent_memory, additional_args=additional_args):
-        for message in pull_messages_from_step(
-            step_log,
-        ):
+        for message in pull_messages_from_step(step_log):
             yield message
 
     final_answer = step_log  # Last log is the run's final_answer
@@ -245,9 +232,7 @@ class GradioUI:
 
         # Sanitize file name
         original_name = os.path.basename(file.name)
-        sanitized_name = re.sub(
-            r"[^\w\-.]", "_", original_name
-        )  # Replace any non-alphanumeric, non-dash, or non-dot characters with underscores
+        sanitized_name = re.sub(r"[^\w\-.]", "_", original_name)  # Replace any non-alphanumeric, non-dash, or non-dot characters with underscores
 
         type_to_ext = {}
         for ext, t in mimetypes.types_map.items():
@@ -315,4 +300,4 @@ You can try a simplified version below. 👇""")
 
         demo.launch(debug=True, share=True, **kwargs)
 
-GradioUI(agent).launch()
+GradioUI(agent).launch()