Katya Beresneva commited on
Commit
7cdf11d
·
1 Parent(s): 94cd5f4
Files changed (2) hide show
  1. agent.py +1 -0
  2. tools.py +1 -8
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.