bstraehle commited on
Commit
67a608a
·
verified ·
1 Parent(s): c0e1900

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +3 -3
crew.py CHANGED
@@ -11,7 +11,7 @@ from google import genai
11
  from google.genai import types
12
  from openinference.instrumentation.crewai import CrewAIInstrumentor
13
  from phoenix.otel import register
14
- from util import read_file, get_final_answer
15
 
16
  ## LLMs
17
 
@@ -25,7 +25,7 @@ IMAGE_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
25
  AUDIO_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
26
  VIDEO_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
27
  YOUTUBE_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
28
- DOCUMENT_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
29
  CODE_GENERATION_MODEL = "gemini-2.5-flash-preview-04-17"
30
  CODE_EXECUTION_MODEL = "gemini-2.5-flash-preview-04-17"
31
 
@@ -41,7 +41,7 @@ tracer_provider = register(
41
  project_name="gaia"
42
  )
43
 
44
- CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
45
 
46
  def run_crew(question, file_path):
47
  # Tools
 
11
  from google.genai import types
12
  from openinference.instrumentation.crewai import CrewAIInstrumentor
13
  from phoenix.otel import register
14
+ from util import read_file
15
 
16
  ## LLMs
17
 
 
25
  AUDIO_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
26
  VIDEO_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
27
  YOUTUBE_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
28
+ DOCUMENT_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
29
  CODE_GENERATION_MODEL = "gemini-2.5-flash-preview-04-17"
30
  CODE_EXECUTION_MODEL = "gemini-2.5-flash-preview-04-17"
31
 
 
41
  project_name="gaia"
42
  )
43
 
44
+ #CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
45
 
46
  def run_crew(question, file_path):
47
  # Tools