j-js commited on
Commit
5eccd5d
·
verified ·
1 Parent(s): 4ec2d5e

Update conversation_logic.py

Browse files
Files changed (1) hide show
  1. conversation_logic.py +7 -8
conversation_logic.py CHANGED
@@ -6,8 +6,8 @@ from typing import Any, Dict, List, Optional
6
 
7
  from context_parser import mentions_choice_letter
8
  from formatting import format_reply
9
- from models import SolverResult
10
- from quant_solver import extract_choices, is_quant_question, solve_quant
11
 
12
 
13
  class ResponseContext:
@@ -376,7 +376,6 @@ def generate_response(
376
  question_difficulty: str = "",
377
  retrieval_context: str = "",
378
  ) -> SolverResult:
379
-
380
  ctx = ResponseContext(
381
  visible_user_text=raw_user_text,
382
  question_text=question_text,
@@ -384,14 +383,14 @@ def generate_response(
384
  question_category=question_category,
385
  question_difficulty=question_difficulty,
386
  chat_history=chat_history,
387
- )
388
 
389
- ctx.retrieval_context = retrieval_context
390
 
391
- solver_input = question_text.strip() if question_text.strip() else raw_user_text.strip()
392
 
393
- if is_quant_question(solver_input):
394
- return solve_quant(solver_input, help_mode)
395
 
396
  visible_user_text = ctx.visible_user_text
397
  question_block = ctx.combined_question_block
 
6
 
7
  from context_parser import mentions_choice_letter
8
  from formatting import format_reply
9
+ from models import SolverResult, ResponseContext
10
+ from quant_solver import solve_quant, is_quant_question, extract_choices
11
 
12
 
13
  class ResponseContext:
 
376
  question_difficulty: str = "",
377
  retrieval_context: str = "",
378
  ) -> SolverResult:
 
379
  ctx = ResponseContext(
380
  visible_user_text=raw_user_text,
381
  question_text=question_text,
 
383
  question_category=question_category,
384
  question_difficulty=question_difficulty,
385
  chat_history=chat_history,
386
+ )
387
 
388
+ ctx.retrieval_context = retrieval_context
389
 
390
+ solver_input = question_text.strip() if question_text.strip() else raw_user_text.strip()
391
 
392
+ if is_quant_question(solver_input):
393
+ return solve_quant(solver_input, help_mode)
394
 
395
  visible_user_text = ctx.visible_user_text
396
  question_block = ctx.combined_question_block