Katya Beresneva commited on
Commit ·
7cdf11d
1
Parent(s): 94cd5f4
fix
Browse files
agent.py
CHANGED
|
@@ -22,6 +22,7 @@ from tools import search_wikipedia
|
|
| 22 |
from tools import SmolagentToolWrapper
|
| 23 |
from tools import tavily_extract_tool
|
| 24 |
from utils import get_llm
|
|
|
|
| 25 |
|
| 26 |
|
| 27 |
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY", "")
|
|
|
|
| 22 |
from tools import SmolagentToolWrapper
|
| 23 |
from tools import tavily_extract_tool
|
| 24 |
from utils import get_llm
|
| 25 |
+
from config import GOOGLE_API_KEY, AGENT_MODEL_NAME
|
| 26 |
|
| 27 |
|
| 28 |
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY", "")
|
tools.py
CHANGED
|
@@ -1,5 +1,4 @@
|
|
| 1 |
import os
|
| 2 |
-
import dotenv
|
| 3 |
import pandas as pd
|
| 4 |
import tempfile
|
| 5 |
import typing
|
|
@@ -24,13 +23,7 @@ from pydantic import Field
|
|
| 24 |
from typing_extensions import Annotated
|
| 25 |
|
| 26 |
from utils import get_llm
|
| 27 |
-
|
| 28 |
-
dotenv.load_dotenv()
|
| 29 |
-
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY", "")
|
| 30 |
-
if not GOOGLE_API_KEY:
|
| 31 |
-
raise ValueError("GOOGLE_API_KEY environment variable is not set.")
|
| 32 |
-
|
| 33 |
-
AGENT_MODEL_NAME = os.getenv("AGENT_MODEL_NAME", "gemini-2.0-flash")
|
| 34 |
|
| 35 |
MULTIMODAL_FILE_ANALYZER_PROMPT = """
|
| 36 |
You are a specialized file analysis AI assistant focused on extracting information from various file formats including images, videos, audio, and structured data.
|
|
|
|
| 1 |
import os
|
|
|
|
| 2 |
import pandas as pd
|
| 3 |
import tempfile
|
| 4 |
import typing
|
|
|
|
| 23 |
from typing_extensions import Annotated
|
| 24 |
|
| 25 |
from utils import get_llm
|
| 26 |
+
from config import GOOGLE_API_KEY, AGENT_MODEL_NAME
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
|
| 28 |
MULTIMODAL_FILE_ANALYZER_PROMPT = """
|
| 29 |
You are a specialized file analysis AI assistant focused on extracting information from various file formats including images, videos, audio, and structured data.
|