dnj0 commited on
Commit
e00eeca
·
verified ·
1 Parent(s): 4faa6dd

Update src/rag_system.py

Browse files
Files changed (1) hide show
  1. src/rag_system.py +2 -2
src/rag_system.py CHANGED
@@ -30,7 +30,7 @@ class VisualMultimodalRAG:
30
 
31
  # Use gpt-4o for vision capabilities
32
  self.llm = ChatOpenAI(
33
- model_name="gpt-4o", # CRITICAL: gpt-4o has vision
34
  api_key=api_key,
35
  temperature=TEMPERATURE,
36
  max_tokens=MAX_TOKENS,
@@ -403,7 +403,7 @@ class AnsweringRAG:
403
  self.debug = debug
404
 
405
  self.llm = ChatOpenAI(
406
- model_name="gpt-4o", # Use gpt-4o for better understanding
407
  api_key=api_key,
408
  temperature=TEMPERATURE,
409
  max_tokens=MAX_TOKENS,
 
30
 
31
  # Use gpt-4o for vision capabilities
32
  self.llm = ChatOpenAI(
33
+ model_name="gpt-4o-mini", # CRITICAL: gpt-4o has vision
34
  api_key=api_key,
35
  temperature=TEMPERATURE,
36
  max_tokens=MAX_TOKENS,
 
403
  self.debug = debug
404
 
405
  self.llm = ChatOpenAI(
406
+ model_name="gpt-4o-mini", # Use gpt-4o for better understanding
407
  api_key=api_key,
408
  temperature=TEMPERATURE,
409
  max_tokens=MAX_TOKENS,