Spaces:
Running
Running
feat: Allow default analysis model in PrecedentAnalysisWorkflow if not specified
Browse files
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 =
|
| 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 |
|