ChillThrills commited on
Commit
d4d544b
·
1 Parent(s): dd839c8

gemini-2.5-flash-preview-05-20

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -106,7 +106,7 @@ AGENT_DEFAULT_TIMEOUT = 15
106
  MAX_CONTEXT_LENGTH_LLM = 30000
107
 
108
  MAX_FILE_SIZE = 5 * 1024 * 1024
109
- CSV_SAMPLE_ROWS = 3
110
  MAX_FILE_CONTEXT_LENGTH = 10000
111
 
112
  # Global instances for video analysis pipelines
@@ -762,7 +762,7 @@ class GaiaLevel1Agent:
762
  if genai and GOOGLE_GEMINI_API_KEY:
763
  try:
764
  genai.configure(api_key=GOOGLE_GEMINI_API_KEY)
765
- model_name = 'gemini-2.0-flash'
766
  self.llm_model = genai.GenerativeModel(model_name)
767
  gaia_logger.info(f"Gemini LLM ('{model_name}') initialized.")
768
  except Exception as e:
@@ -774,13 +774,13 @@ class GaiaLevel1Agent:
774
  gaia_logger.warning("LLM (Gemini) unavailable. Limited capabilities.")
775
 
776
  _get_video_object_detector()
777
- _get_video_vqa_pipeline() # Initialize VQA pipeline
778
 
779
  gaia_logger.info(f"GaiaLevel1Agent (RAG, FileProcessor, VideoAnalysis) initialized. API: {self.api_url}")
780
 
781
  @lru_cache(maxsize=32)
782
  def _fetch_and_process_file_content(self, task_id: str) -> Optional[str]:
783
- # ... (this method remains unchanged) ...
784
  file_url = f"{self.api_url}/files/{task_id}"
785
  for attempt in range(2):
786
  try:
 
106
  MAX_CONTEXT_LENGTH_LLM = 30000
107
 
108
  MAX_FILE_SIZE = 5 * 1024 * 1024
109
+ CSV_SAMPLE_ROWS = 10
110
  MAX_FILE_CONTEXT_LENGTH = 10000
111
 
112
  # Global instances for video analysis pipelines
 
762
  if genai and GOOGLE_GEMINI_API_KEY:
763
  try:
764
  genai.configure(api_key=GOOGLE_GEMINI_API_KEY)
765
+ model_name = 'gemini-2.5-flash-preview-05-20'
766
  self.llm_model = genai.GenerativeModel(model_name)
767
  gaia_logger.info(f"Gemini LLM ('{model_name}') initialized.")
768
  except Exception as e:
 
774
  gaia_logger.warning("LLM (Gemini) unavailable. Limited capabilities.")
775
 
776
  _get_video_object_detector()
777
+ _get_video_vqa_pipeline()
778
 
779
  gaia_logger.info(f"GaiaLevel1Agent (RAG, FileProcessor, VideoAnalysis) initialized. API: {self.api_url}")
780
 
781
  @lru_cache(maxsize=32)
782
  def _fetch_and_process_file_content(self, task_id: str) -> Optional[str]:
783
+
784
  file_url = f"{self.api_url}/files/{task_id}"
785
  for attempt in range(2):
786
  try: