pvanand commited on
Commit
e36d1a0
·
verified ·
1 Parent(s): 5594ef5

update clarification function

Browse files
Files changed (1) hide show
  1. main.py +18 -0
main.py CHANGED
@@ -424,6 +424,24 @@ def parse_followup_response(response):
424
 
425
  return response_content, clarification
426
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
427
  @app.post("/followup-agent")
428
  async def followup_agent(query: FollowupQueryModel, background_tasks: BackgroundTasks, api_key: str = Depends(verify_api_key)):
429
  """
 
424
 
425
  return response_content, clarification
426
 
427
+
428
+ def parse_clarification(text):
429
+ questions = []
430
+ current_question = None
431
+
432
+ for line in text.split('\n'):
433
+ line = line.strip()
434
+ if line.startswith('- text:'):
435
+ if current_question:
436
+ questions.append(current_question)
437
+ current_question = {'text': line[7:].strip(), 'options': []}
438
+ elif line.startswith('- ') and current_question:
439
+ current_question['options'].append(line[2:].strip())
440
+
441
+ if current_question:
442
+ questions.append(current_question)
443
+
444
+
445
  @app.post("/followup-agent")
446
  async def followup_agent(query: FollowupQueryModel, background_tasks: BackgroundTasks, api_key: str = Depends(verify_api_key)):
447
  """