DocUA commited on
Commit
028f33a
·
1 Parent(s): e575bd8

feat: Allow default analysis model in PrecedentAnalysisWorkflow if not specified

Browse files
Files changed (1) hide show
  1. main.py +5 -1
main.py CHANGED
@@ -38,6 +38,7 @@ from config import (
38
  DEBUG_PROMPTS,
39
  ModelProvider,
40
  AnalysisModelName,
 
41
  DEEPSEEK_API_KEY,
42
  validate_environment
43
  )
@@ -618,13 +619,16 @@ class PrecedentAnalysisWorkflow(Workflow):
618
  """Workflow for analyzing legal precedents."""
619
 
620
  def __init__(self, provider: Any = ModelProvider.OPENAI,
621
- model_name: Any = AnalysisModelName.GPT4o_MINI,
622
  temperature: float = GENERATION_TEMPERATURE,
623
  max_tokens: Optional[int] = None,
624
  thinking_enabled: bool = False,
625
  thinking_level: str = "medium",
626
  openai_verbosity: str = "medium"):
627
  super().__init__()
 
 
 
628
  self.analyzer = LLMAnalyzer(provider, model_name, temperature, max_tokens,
629
  thinking_enabled, thinking_level, openai_verbosity)
630
 
 
38
  DEBUG_PROMPTS,
39
  ModelProvider,
40
  AnalysisModelName,
41
+ DEFAULT_ANALYSIS_MODEL,
42
  DEEPSEEK_API_KEY,
43
  validate_environment
44
  )
 
619
  """Workflow for analyzing legal precedents."""
620
 
621
  def __init__(self, provider: Any = ModelProvider.OPENAI,
622
+ model_name: Any = None,
623
  temperature: float = GENERATION_TEMPERATURE,
624
  max_tokens: Optional[int] = None,
625
  thinking_enabled: bool = False,
626
  thinking_level: str = "medium",
627
  openai_verbosity: str = "medium"):
628
  super().__init__()
629
+ # Use default analysis model if not specified
630
+ if model_name is None:
631
+ model_name = DEFAULT_ANALYSIS_MODEL or AnalysisModelName.GPT5_2
632
  self.analyzer = LLMAnalyzer(provider, model_name, temperature, max_tokens,
633
  thinking_enabled, thinking_level, openai_verbosity)
634